From 0ba0f7dcedb354068904ec585337c384ee8b664e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BD=E8=8C=B6=E6=B8=85=E6=AC=A2?= Date: Fri, 31 May 2024 18:37:35 +0800 Subject: [PATCH] =?UTF-8?q?=E8=AE=BE=E7=BD=AE=E8=AF=B7=E6=B1=82header?= =?UTF-8?q?=E4=B8=8E=E8=AF=B7=E6=B1=82url?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- client.go | 3 +++ resty.go | 7 +++++-- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/client.go b/client.go index 8f9def7..a3d9512 100644 --- a/client.go +++ b/client.go @@ -89,5 +89,8 @@ func (hc *HttpClient) GetRestyClient() *resty.Client { // // Date : 15:52 2024/5/31 func (hc *HttpClient) Request(reqConfig *define.Request) *define.Response { + hc.Client.OnBeforeRequest(hc.getRequestValidateMiddleware()) // 请求参数验证中间件必注册 + hc.Client.OnAfterResponse(hc.getResponseValidateMiddleware()) // 响应验证中间件必注册 + hc.request.Send() return nil } diff --git a/resty.go b/resty.go index 7309836..8330355 100644 --- a/resty.go +++ b/resty.go @@ -38,6 +38,8 @@ func NewRestyClient(reqConfig *define.Request) (*resty.Client, *resty.Request) { request.SetPathParams(reqConfig.PathParam) // 设置path中的参数 request.SetQueryParams(reqConfig.Query) // 设置query参数 request.SetHeaders(reqConfig.Header) // 设置header + request.URL = reqConfig.FullUrl // 请求接口 + request.Method = reqConfig.Method // 请求方法 cookieList := make([]*http.Cookie, 0) for cookieName, cookieValue := range reqConfig.Cookie { cookieList = append(cookieList, &http.Cookie{ @@ -45,8 +47,8 @@ func NewRestyClient(reqConfig *define.Request) (*resty.Client, *resty.Request) { Value: cookieValue, }) } - request.SetCookies(cookieList) // 设置cookie - setRestyBody(reqConfig, request) + request.SetCookies(cookieList) // 设置cookie + setRestyBody(reqConfig, request) // 设置请求Body return client, request } @@ -71,6 +73,7 @@ func setRestyBody(reqConfig *define.Request, request *resty.Request) { bodyData[key.String()] = value.String() return true }) + request.SetFormData(bodyData) } return }