github.com/kaydxh/golang@v0.0.131/go/net/http/http_client.do.go (about) 1 /* 2 *Copyright (c) 2022, kaydxh 3 * 4 *Permission is hereby granted, free of charge, to any person obtaining a copy 5 *of this software and associated documentation files (the "Software"), to deal 6 *in the Software without restriction, including without limitation the rights 7 *to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 8 *copies of the Software, and to permit persons to whom the Software is 9 *furnished to do so, subject to the following conditions: 10 * 11 *The above copyright notice and this permission notice shall be included in all 12 *copies or substantial portions of the Software. 13 * 14 *THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 15 *IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 16 *FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 17 *AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 18 *LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 19 *OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 20 *SOFTWARE. 21 */ 22 package http 23 24 import ( 25 "context" 26 "io" 27 "net/http" 28 29 time_ "github.com/kaydxh/golang/go/time" 30 31 logs_ "github.com/kaydxh/golang/pkg/logs" 32 ) 33 34 func (c *Client) get(ctx context.Context, url string) (*http.Response, error) { 35 return c.HttpDo(ctx, http.MethodGet, url, "", nil, nil, nil) 36 } 37 38 func (c *Client) post(ctx context.Context, url string, contentType string, headers map[string]string, 39 auth func(r *http.Request) error, 40 body io.Reader, 41 ) (*http.Response, error) { 42 return c.HttpDo(ctx, http.MethodPost, url, contentType, headers, auth, body) 43 } 44 45 func (c *Client) put(ctx context.Context, url string, contentType string, headers map[string]string, 46 auth func(r *http.Request) error, 47 body io.Reader, 48 ) (*http.Response, error) { 49 return c.HttpDo(ctx, http.MethodPut, url, contentType, headers, auth, body) 50 } 51 52 func (c *Client) HttpDo(ctx context.Context, method string, url string, contentType string, headers map[string]string, 53 auth func(r *http.Request) error, 54 body io.Reader, 55 ) (*http.Response, error) { 56 req, err := http.NewRequest(method, url, body) 57 if err != nil { 58 return nil, err 59 } 60 61 if contentType != "" { 62 req.Header.Set("Content-Type", contentType) 63 } 64 for key, value := range headers { 65 req.Header.Set(key, value) 66 } 67 68 if auth != nil { 69 err = auth(req) 70 if err != nil { 71 return nil, err 72 } 73 } 74 75 return c.Do(ctx, req) 76 } 77 78 func (c *Client) Do(ctx context.Context, req *http.Request) (*http.Response, error) { 79 /* 80 err := RequestWithTargetHost(req, c.opts.targetHost) 81 if err != nil { 82 return nil, err 83 } 84 */ 85 86 tc := time_.New(true) 87 logger := logs_.GetLogger(ctx) 88 logger.WithField("target_addr", req.Host).Infof("http do %v", req.URL.Path) 89 summary := func() { 90 tc.Tick(req.Method) 91 logger.WithField("method", req.Method).Infof(tc.String()) 92 } 93 defer summary() 94 95 return c.Client.Do(req) 96 }