添加请求重试机制 (#117)
This commit is contained in:
parent
9b1bc9189b
commit
d2a2ffc119
13
client.go
13
client.go
@ -71,11 +71,22 @@ func (c *Client) AddHttpHook(hooks ...HttpHook) {
|
|||||||
c.HttpHooks = append(c.HttpHooks, hooks...)
|
c.HttpHooks = append(c.HttpHooks, hooks...)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const maxRetry = 2
|
||||||
|
|
||||||
func (c *Client) do(req *http.Request) (*http.Response, error) {
|
func (c *Client) do(req *http.Request) (*http.Response, error) {
|
||||||
for _, hook := range c.HttpHooks {
|
for _, hook := range c.HttpHooks {
|
||||||
hook.BeforeRequest(req)
|
hook.BeforeRequest(req)
|
||||||
}
|
}
|
||||||
resp, err := c.Client.Do(req)
|
var (
|
||||||
|
resp *http.Response
|
||||||
|
err error
|
||||||
|
)
|
||||||
|
for i := 0; i < maxRetry; i++ {
|
||||||
|
resp, err = c.Client.Do(req)
|
||||||
|
if err == nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
err = ErrorWrapper(NetworkErr, err.Error())
|
err = ErrorWrapper(NetworkErr, err.Error())
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user