Merge pull request 'feature/upgrade_retry' (#5) from feature/upgrade_retry into master

Reviewed-on: #5

httpclient/client.go

@@ -10,7 +10,6 @@ package httpclient
 import (
 	"fmt"
 	"net"
-	"net/http"
 	"strings"
 	"time"
 
@@ -37,6 +36,18 @@ func NewHttpClient(reqConfig *define.Request, cacheInstance cache.ICache) (*Http
 	if err := validate.RequestConfig(reqConfig); nil != err {
 		return nil, err
 	}
+	if reqConfig.RetryRule == nil {
+		reqConfig.RetryRule = &define.RequestRetryRule{
+			RetryCount:            0,
+			RetryTimeInterval:     0,
+			RetryHttpCodeList:     []int64{},
+			RetryBusinessCodeList: []string{},
+		} // 未指定重试规则, 则使用默认重试规则
+	}
+	// 初始化成功的 http code list
+	if len(reqConfig.SuccessHttpCodeList) == 0 {
+		reqConfig.SuccessHttpCodeList = []int{}
+	}
 	if len(reqConfig.Static) > 0 {
 		for loc, valMap := range reqConfig.Static {
 			if len(valMap) == 0 {
@@ -276,6 +287,20 @@ func (hc *HttpClient) requestBackendApi() *define.Response {
 		response.RequestFinishTime = 0                     // 清空完成时间
 		response.Seq++
 		response.RequestCount++
+		retryHttpCodeTable := make(map[int64]bool)
+		retryBusinessCodeTable := make(map[string]bool)
+		if nil != hc.reqConfig.RetryRule {
+			for _, httpCode := range hc.reqConfig.RetryRule.RetryHttpCodeList {
+				retryHttpCodeTable[httpCode] = true
+			}
+			for _, businessCode := range hc.reqConfig.RetryRule.RetryBusinessCodeList {
+				retryBusinessCodeTable[businessCode] = true
+			}
+		}
+		successHttpCodeTable := make(map[int]bool)
+		for _, itemHttpCode := range hc.reqConfig.SuccessHttpCodeList {
+			successHttpCodeTable[itemHttpCode] = true
+		}
 		if response.RestyResponse, err = hc.request.Send(); nil != err {
 			errType := define.RequestFailTypeSend
 			if err.Error() == resty.ErrRateLimitExceeded.Error() {
@@ -293,6 +318,10 @@ func (hc *HttpClient) requestBackendApi() *define.Response {
 				// 命中限流就不重试了
 				break
 			}
+			if len(retryHttpCodeTable) > 0 && !retryHttpCodeTable[499] {
+				// 未配置超时重试
+				break
+			}
 			time.Sleep(time.Duration(hc.reqConfig.RetryRule.RetryTimeInterval) * time.Millisecond)
 			continue
 		}
@@ -309,7 +338,10 @@ func (hc *HttpClient) requestBackendApi() *define.Response {
 		hc.fillResponseHeader(response)
 		hc.fillResponseCookie(response)
 		hc.fillResponseBody(response)
-		if response.HttpCode != http.StatusOK {
+		// 配置了当前code为成功, 或者未配置任何code, 当前code为2xx, 则认为请求成功
+		isHttpSuccess := successHttpCodeTable[response.HttpCode] || (len(successHttpCodeTable) == 0 && response.HttpCode/100 == 2)
+		if !isHttpSuccess {
+			// 非 成功
 			errType := define.RequestFailTypeServerError
 			if response.HttpCode/100 == 4 {
 				// 客户端错误
@@ -319,6 +351,10 @@ func (hc *HttpClient) requestBackendApi() *define.Response {
 				Type:    errType,
 				Message: "http code is " + response.HttpCodeStatus + ", not success",
 			}
+			if len(retryHttpCodeTable) > 0 && !retryHttpCodeTable[int64(response.HttpCode)] {
+				// 未配置http code重试
+				break
+			}
 			time.Sleep(time.Duration(hc.reqConfig.RetryRule.RetryTimeInterval) * time.Millisecond)
 			continue
 		}
@@ -327,6 +363,10 @@ func (hc *HttpClient) requestBackendApi() *define.Response {
 				Type:    define.RequestFailTypeBusinessError,
 				Message: "business code is " + response.Code + ", not success",
 			}
+			if len(retryBusinessCodeTable) > 0 && !retryBusinessCodeTable[response.Code] {
+				// 未配置业务code重试
+				break
+			}
 			time.Sleep(time.Duration(hc.reqConfig.RetryRule.RetryTimeInterval) * time.Millisecond)
 			continue
 		}

httpclient/define/request.go

@@ -33,6 +33,7 @@ type Request struct {
 	CodeField           string                    `json:"code_field"`             // 业务状态码字段
 	MessageField        string                    `json:"message_field"`          // code描述字段
 	DataReceiver        any                       `json:"-"`                      // 响应data部分数据解析
+	SuccessHttpCodeList []int                     `json:"success_http_code_list"` // 哪些http状态码视为成功, 不配置, 默认2xx
 	SuccessCodeList     []string                  `json:"success_code_list"`      // 哪些业务状态码视为成功
 	ConnectTimeout      int64                     `json:"connect_timeout"`        // 连接超时时间: ms
 	ReadTimeout         int64                     `json:"read_timeout"`           // 读取超时时间

httpclient/resty.go

@@ -63,7 +63,7 @@ func NewRestyClient(reqConfig *define.Request) (*resty.Client, *resty.Request) {
 		if len(pathParamValue) == 0 {
 			continue
 		}
-		reqConfig.FullUrl = strings.ReplaceAll(reqConfig.FullUrl, "{#"+pathParamName+"#}", pathParamValue)
+		reqConfig.FullUrl = strings.ReplaceAll(reqConfig.FullUrl, "{"+pathParamName+"}", pathParamValue)
 	}
 	request.Method = reqConfig.Method // 请求方法
 	cookieList := make([]*http.Cookie, 0)