45 lines
767 B
Go
45 lines
767 B
Go
|
package httpclient
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"net/http"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
// DefaultRetryTimes 如果请求失败，最多重试3次
|
|||
|
|
DefaultRetryTimes = 3
|
|||
|
|
// DefaultRetryDelay 在重试前，延迟等待100毫秒
|
|||
|
|
DefaultRetryDelay = time.Millisecond * 100
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Verify parse the body and verify that it is correct
|
|||
|
|
type RetryVerify func(body []byte) (shouldRetry bool)
|
|||
|
|
|
|||
|
|
func shouldRetry(ctx context.Context, httpCode int) bool {
|
|||
|
|
select {
|
|||
|
|
case <-ctx.Done():
|
|||
|
|
return false
|
|||
|
|
default:
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
switch httpCode {
|
|||
|
|
case
|
|||
|
|
_StatusReadRespErr,
|
|||
|
|
_StatusDoReqErr,
|
|||
|
|
|
|||
|
|
http.StatusRequestTimeout,
|
|||
|
|
http.StatusLocked,
|
|||
|
|
http.StatusTooEarly,
|
|||
|
|
http.StatusTooManyRequests,
|
|||
|
|
|
|||
|
|
http.StatusServiceUnavailable,
|
|||
|
|
http.StatusGatewayTimeout:
|
|||
|
|
|
|||
|
|
return true
|
|||
|
|
|
|||
|
|
default:
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
}
|