bingyi/gf
1
0
Fork 0
mirror of https://gitee.com/johng/gf synced 2026-06-06 16:21:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

improve ghttp.Client

Browse Source
This commit is contained in:
john
2020-07-25 11:24:35 +08:00
parent e92fd05f9f
commit 245c6d24a1
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
net/ghttp/ghttp_client_request.go
View File

@ -227,6 +227,10 @@ func (c *Client) DoRequest(method, url string, data ...interface{}) (resp *Clien
req.Body = utils.NewReadCloser(reqBodyContent, false)
for {
if resp.Response, err = c.Do(req); err != nil {
// The response might not be nil when err != nil.
if resp.Response != nil {
resp.Response.Body.Close()
}
if c.retryCount > 0 {
c.retryCount--
time.Sleep(c.retryInterval)

3
net/ghttp/ghttp_client_response.go
View File

@ -63,6 +63,9 @@ func (r *ClientResponse) ReadAllString() string {
// Close closes the response when it will never be used.
func (r *ClientResponse) Close() error {
if r == nil || r.Response == nil || r.Response.Close {
return nil
}
r.Response.Close = true
return r.Response.Body.Close()
}