Initial commit

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
2026-09-15 13:57:10 +08:00
commit 8679200f41
1897 changed files with 257900 additions and 0 deletions
@@ -0,0 +1,282 @@
+package httputil
+
+import (
+	"91porn-server/common/log"
+	"91porn-server/middleware/requestid"
+	"context"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"github.com/ddliu/go-httpclient"
+	"io"
+	"net/http"
+	"strconv"
+)
+
+// HTTPClient htttp客户端
+type HTTPClient struct {
+	client *httpclient.HttpClient
+}
+
+func NewHTTPClient(client *httpclient.HttpClient) *HTTPClient {
+	return &HTTPClient{client}
+}
+
+type httpClientPool struct {
+	cliPool  map[string]*httpclient.HttpClient
+	poolSize int
+	offset   int
+}
+
+// HTTPResponse http响应
+type HTTPResponse struct {
+	StatusCode int                  `json:"statusCode"`
+	Body       *httpclient.Response `json:"boby"`
+}
+
+var httpCliPool = httpClientPool{cliPool: map[string]*httpclient.HttpClient{}, poolSize: 50}
+
+func New() *HTTPClient {
+	h := httpclient.NewHttpClient().WithOptions(httpclient.Map{
+		httpclient.OPT_CONNECTTIMEOUT: 5,
+		httpclient.OPT_TIMEOUT:        10,
+	})
+	return &HTTPClient{client: h}
+}
+
+func NewCtx(ctx context.Context) *HTTPClient {
+	reqID, _ := ctx.Value(requestid.ContextKey).(string)
+	h := httpclient.NewHttpClient().WithOptions(httpclient.Map{
+		httpclient.OPT_CONNECTTIMEOUT: 5,
+		httpclient.OPT_TIMEOUT:        10,
+		httpclient.OPT_CONTEXT:        ctx,
+	}).WithHeader(requestid.HeaderKey, reqID)
+
+	return &HTTPClient{client: h.Begin()}
+}
+
+// GetHTTPClient  获取httpclient
+func Client() *HTTPClient {
+	cliID := "CLIENT-ID-"
+	if len(httpCliPool.cliPool) < httpCliPool.poolSize {
+		h := httpclient.NewHttpClient().Defaults(httpclient.Map{
+			httpclient.OPT_CONNECTTIMEOUT: 5,
+		})
+		cliID = cliID + strconv.FormatInt(int64(len(httpCliPool.cliPool)+1), 10)
+		httpCliPool.cliPool[cliID] = h
+		httpCliPool.offset = httpCliPool.poolSize
+		return &HTTPClient{client: h}
+	}
+	if httpCliPool.offset == httpCliPool.poolSize+1 {
+		httpCliPool.offset = 1
+	}
+	cliID = cliID + strconv.FormatInt(int64(httpCliPool.offset), 10)
+	httpCliPool.offset = httpCliPool.offset + 1
+	h, ok := httpCliPool.cliPool[cliID]
+	if !ok || h == nil {
+		h = httpclient.NewHttpClient().WithOption(httpclient.OPT_CONNECTTIMEOUT, 10)
+		httpCliPool.cliPool[cliID] = h
+		return &HTTPClient{client: h}
+	}
+	return &HTTPClient{client: h}
+}
+
+func jsonUnmarshalResp(bind interface{}, resp *HTTPResponse) error {
+	if bind == nil { //传入空表示只执行 defer
+		return nil
+	}
+	b, err := resp.Body.ReadAll()
+	if err != nil {
+		log.Error("httputil jsonUnmarshalResp readall err", log.E(err))
+		return err
+	}
+	err = json.Unmarshal(b, bind)
+	if err != nil {
+		log.Error("httputil jsonUnmarshalResp Unmarshal err", log.E(err))
+	}
+	return err
+}
+
+// Get Get方法
+func (h *HTTPClient) Get(url string, headers map[string]string, params ...interface{}) (*HTTPResponse, error) {
+	h.client.WithHeaders(headers)
+	response, err := h.client.Get(url, params...)
+	if err != nil {
+		log.Error(fmt.Sprintf("http client Get method failed %+v:", err))
+		return nil, err
+	}
+	return &HTTPResponse{StatusCode: response.StatusCode, Body: response}, nil
+}
+
+func (h *HTTPClient) PGet(p *ProxyCfg, url string, headers map[string]string, params ...interface{}) (*HTTPResponse, error) {
+	purl, pheaders, err := p.Build(url, headers)
+	if err != nil {
+		return nil, err
+	}
+	return h.Get(purl, pheaders, params)
+}
+
+// Get GetBytes
+func (h *HTTPClient) GetBytes(url string, headers map[string]string, params ...interface{}) ([]byte, error) {
+	h.client.WithHeaders(headers)
+	response, err := h.client.Get(url, params...)
+	if err != nil {
+		log.Error(fmt.Sprintf("http client Get method failed %+v:", err))
+		return nil, err
+	}
+	if response.StatusCode != http.StatusOK {
+		log.Error("http client Get method status code not ok", log.Any("url", url), log.Any("headers", headers), log.Any("params", params), log.Any("res", response))
+		return nil, errors.New("http response satus code not ok statusCoe:" + response.Status)
+	}
+	data, err := response.ReadAll()
+	if err != nil {
+		log.ZapLog.Error("resp readAll errror", log.Any("Error", err))
+		return data, err
+	}
+	return data, nil
+}
+
+// GetWithJResp 结果json.Unmarshal到bind中
+func (h *HTTPClient) GetWithJResp(bind interface{}, url string, headers map[string]string, params ...interface{}) (int, error) {
+	resp, err := h.Get(url, headers, params...)
+	if err != nil {
+		return 0, err
+	}
+	defer func() { _ = resp.Body.Body.Close() }()
+	if resp.StatusCode == http.StatusOK {
+		err = jsonUnmarshalResp(bind, resp)
+	}
+	return resp.StatusCode, err
+}
+
+func (h *HTTPClient) PGetWithJResp(p *ProxyCfg, bind interface{}, url string, headers map[string]string, params ...interface{}) (int, error) {
+	purl, pheaders, err := p.Build(url, headers)
+	if err != nil {
+		return 0, err
+	}
+	return h.GetWithJResp(bind, purl, pheaders, params...)
+}
+
+// Post Post方法
+func (h *HTTPClient) Post(url string, headers map[string]string, params interface{}) (*HTTPResponse, error) {
+	h.client.WithHeaders(headers)
+	response, err := h.client.Post(url, params)
+	if err != nil {
+		log.Error(fmt.Sprintf("http client Post method failed %+v:", err))
+		return nil, err
+	}
+	return &HTTPResponse{StatusCode: response.StatusCode, Body: response}, nil
+}
+
+func (h *HTTPClient) PPost(p *ProxyCfg, bind interface{}, url string, headers map[string]string, params interface{}) (*HTTPResponse, error) {
+	purl, pheaders, err := p.Build(url, headers)
+	if err != nil {
+		return nil, err
+	}
+	return h.Post(purl, pheaders, params)
+}
+
+// POSTJson PostJson方法
+func (h *HTTPClient) POSTJson(url string, headers map[string]string, data interface{}) (*HTTPResponse, error) {
+	h.client.WithHeaders(headers)
+	response, err := h.client.PostJson(url, data)
+	if err != nil {
+		log.Error(fmt.Sprintf("http client POSTJson method failed %+v:", err))
+		return nil, err
+	}
+	return &HTTPResponse{StatusCode: response.StatusCode, Body: response}, nil
+}
+
+// POSTJson PostJson方法, bind json.Unmarshal
+func (h *HTTPClient) POSTJsonWithJResp(bind interface{}, url string, headers map[string]string, data interface{}) (int, error) {
+	resp, err := h.POSTJson(url, headers, data)
+	if err != nil {
+		log.Error("POSTJsonWithResp err", log.E(err))
+		return 0, err
+	}
+	defer resp.Body.Body.Close()
+	if resp.StatusCode == http.StatusOK {
+		err = jsonUnmarshalResp(bind, resp)
+	}
+	return resp.StatusCode, err
+}
+
+func (h *HTTPClient) PPOSTJsonWithJResp(p *ProxyCfg, bind interface{}, url string, headers map[string]string, data interface{}) (int, error) {
+	if p == nil {
+		return 0, fmt.Errorf("ProxyCfg is nil")
+	}
+	purl, pheaders, err := p.Build(url, headers)
+	if err != nil {
+		return 0, err
+	}
+	return h.POSTJsonWithJResp(bind, purl, pheaders, data)
+}
+
+// PostMultipart PostMultipart 上传文件
+func (h *HTTPClient) PostMultipart(url string, headers map[string]string, params interface{}) (*HTTPResponse, error) {
+	h.client.WithHeaders(headers)
+	response, err := h.client.PostMultipart(url, params)
+	if err != nil {
+		log.Error(fmt.Sprintf("http client PostMultipart method failed %+v:", err))
+		return nil, err
+	}
+	return &HTTPResponse{StatusCode: response.StatusCode, Body: response}, nil
+}
+
+// Put 流式传输 可用于上传文件
+func (h *HTTPClient) Put(url string, headers map[string]string, body io.Reader) (*HTTPResponse, error) {
+	h.client.WithHeaders(headers)
+	response, err := h.client.Put(url, body)
+	if err != nil {
+		log.Error(fmt.Sprintf("http client PostMultipart method failed %+v:", err))
+		return nil, err
+	}
+	return &HTTPResponse{StatusCode: response.StatusCode, Body: response}, nil
+}
+
+// PutJSON PutJSON方法
+func (h *HTTPClient) PutJSON(url string, headers map[string]string, data interface{}) (*HTTPResponse, error) {
+	h.client.WithHeaders(headers)
+	response, err := h.client.PutJson(url, data)
+	if err != nil {
+		log.Error(fmt.Sprintf("http client PostMultipart method failed %+v:", err))
+		return nil, err
+	}
+	return &HTTPResponse{StatusCode: response.StatusCode, Body: response}, nil
+}
+
+// Header Header
+func (h *HTTPClient) Header(url string) (*HTTPResponse, error) {
+	response, err := h.client.Head(url)
+	if err != nil {
+		log.Error(fmt.Sprintf("http client PostMultipart method failed %+v:", err))
+		return nil, err
+	}
+	return &HTTPResponse{StatusCode: response.StatusCode, Body: response}, nil
+}
+
+// POSTWithJResp 结果json.Unmarshal到bind中  用于上传文件, content-type :"application/x-www-form-urlencoded" 如参数以@开头 则为上传文件
+func (h *HTTPClient) POSTWithJResp(bind interface{}, url string, headers map[string]string, params interface{}) (int, error) {
+	resp, err := h.Post(url, headers, params)
+	if err != nil {
+		return 0, err
+	}
+	defer resp.Body.Body.Close()
+	if resp.StatusCode == http.StatusOK {
+		err = jsonUnmarshalResp(bind, resp)
+	}
+	return resp.StatusCode, err
+}
+
+// PPOSTWithJResp 结果json.Unmarshal到bind中  用于上传文件, content-type :"application/x-www-form-urlencoded" 如参数以@开头 则为上传文件
+func (h *HTTPClient) PPOSTWithJResp(p *ProxyCfg, bind interface{}, url string, headers map[string]string, params interface{}) (int, error) {
+	resp, err := h.PPost(p, bind, url, headers, params)
+	if err != nil {
+		return 0, err
+	}
+	defer resp.Body.Body.Close()
+	if resp.StatusCode == http.StatusOK {
+		err = jsonUnmarshalResp(bind, resp)
+	}
+	return resp.StatusCode, err
+}