支持响应数据类型自适应解析

httpclient/implement/response.go

@@ -8,21 +8,23 @@
 package implement
 
 import (
+	"errors"
 	"fmt"
+	"git.zhangdeman.cn/zhangdeman/consts"
 	"git.zhangdeman.cn/zhangdeman/network/httpclient/define"
 	"git.zhangdeman.cn/zhangdeman/serialize"
 	"github.com/tidwall/gjson"
+	"strings"
 )
 
 // Response 响应结果解析
 type Response struct {
 }
 
-func (r *Response) Parse(reqConfig *define.Request, response *define.Response) {
+func (r *Response) Parse(reqConfig *define.Request, response *define.Response) error {
 	r.fillResponseHeader(response)
 	r.fillResponseCookie(response)
-	r.fillResponseBody(reqConfig, response)
-	return
+	return r.fillResponseBody(reqConfig, response)
 }
 
 // fillResponseHeader 填充响应header
@@ -60,8 +62,38 @@ func (r *Response) fillResponseCookie(response *define.Response) {
 // Author : go_developer@163.com<白茶清欢>
 //
 // Date : 21:38 2024/6/5
-func (r *Response) fillResponseBody(reqCfg *define.Request, response *define.Response) {
-	response.Data = string(response.RestyResponse.Body())
+func (r *Response) fillResponseBody(reqCfg *define.Request, response *define.Response) error {
+	// 解析响应数据类型
+	responseContentType := response.RestyResponse.Header().Get(consts.HeaderKeyContentType.String())
+	if responseContentType == "" {
+		// 返回数据未说明 Content-Type
+		return errors.New("response content type is empty")
+	}
+	typeArr := strings.Split(strings.Split(responseContentType, ";")[0], "/")
+	responseType := "json"
+	if len(typeArr) > 0 {
+		responseType = typeArr[len(typeArr)-1]
+	}
+	parser := ResponseParserTable[responseType]
+	if parser == nil {
+		// 未读取到的, 不支持解析对应的响应数据
+		return errors.New("response type " + responseType + " is not support")
+
+	}
+
+	var (
+		err      error
+		res      map[string]any
+		jsonByte []byte
+	)
+
+	if err = parser.Unmarshal(response.RestyResponse.Body(), &res); nil != err {
+		return errors.New("response parse body error :" + err.Error())
+	}
+	if jsonByte, err = parser.MarshalForByte(res); nil != err {
+		return errors.New("response body Marshal error :" + err.Error())
+	}
+	response.Data = string(jsonByte)
 	response.Code = gjson.Get(response.Data, reqCfg.CodeField).String()
 	response.Message = gjson.Get(response.Data, reqCfg.MessageField).String()
 	businessData := gjson.Get(response.Data, reqCfg.DataField)
@@ -96,6 +128,7 @@ func (r *Response) fillResponseBody(reqCfg *define.Request, response *define.Res
 		response.ExtendData[key.String()] = value.String()
 		return true
 	})
+	return nil
 }
 
 // BusinessSuccess ...