// Package httpclient ... // // Description : httpclient ... // // Author : go_developer@163.com<白茶清欢> // // Date : 2024-05-31 15:22 package httpclient import ( "git.zhangdeman.cn/gateway/httpclient/define" "git.zhangdeman.cn/gateway/httpclient/validate" "github.com/go-resty/resty/v2" ) // NewHttpClient 获取http client // // Author : go_developer@163.com<白茶清欢> // // Date : 15:27 2024/5/31 func NewHttpClient(reqConfig *define.Request) (*HttpClient, error) { // 验证配置正确性以及初始化默认值 if err := validate.RequestConfig(reqConfig); nil != err { return nil, err } restyClient, restyRequest := NewRestyClient(reqConfig) hc := &HttpClient{ Client: restyClient, request: restyRequest, reqConfig: reqConfig, } return hc, nil } // HttpClient 请求客户端 // // Author : go_developer@163.com<白茶清欢> // // Date : 15:27 2024/5/31 type HttpClient struct { *resty.Client request *resty.Request reqConfig *define.Request } // getRequestValidateMiddleware 请求验证的Middleware // // Author : go_developer@163.com<白茶清欢> // // Date : 15:40 2024/5/31 func (hc *HttpClient) getRequestValidateMiddleware() resty.RequestMiddleware { return func(client *resty.Client, request *resty.Request) error { return nil } } // getResponseValidateMiddleware 获取相应数据验证的middleware // // Author : go_developer@163.com<白茶清欢> // // Date : 15:42 2024/5/31 func (hc *HttpClient) getResponseValidateMiddleware() resty.ResponseMiddleware { return func(client *resty.Client, response *resty.Response) error { return nil } } // SetRestyClient 设置client // // Author : go_developer@163.com<白茶清欢> // // Date : 15:54 2024/5/31 func (hc *HttpClient) SetRestyClient(restyClient *resty.Client) { hc.Client = restyClient } // GetRestyClient 获取 resty client // // Author : go_developer@163.com<白茶清欢> // // Date : 15:57 2024/5/31 func (hc *HttpClient) GetRestyClient() *resty.Client { return hc.Client } // Request 发送请求 // // Author : go_developer@163.com<白茶清欢> // // Date : 15:52 2024/5/31 func (hc *HttpClient) Request() *define.Response { hc.Client.OnBeforeRequest(hc.getRequestValidateMiddleware()) // 请求参数验证中间件必注册 hc.Client.OnAfterResponse(hc.getResponseValidateMiddleware()) // 响应验证中间件必注册 var ( err error ) response := &define.Response{ Header: map[string]string{}, Cookie: map[string]string{}, Body: map[string]any{}, Code: "", Message: "", Data: "", HttpCode: 0, HttpCodeStatus: "", ResponseDataRule: nil, Seq: 0, RequestStartTime: 0, FinishRequestTime: 0, UsedTime: 0, RestyResponse: nil, IsSuccess: false, RequestCount: 0, FailInfoList: make([]any, 0), } for i := 0; i < hc.reqConfig.RetryRule.RetryCount+1; i++ { if response.RestyResponse, err = hc.request.Send(); nil != err { } } return nil }