perf: http client transport (#1311)

Signed-off-by: rfyiamcool <rfyiamcool@163.com>

pkg/common/http/http_client.go

@@ -31,14 +31,25 @@ import (
 	"github.com/openimsdk/open-im-server/v3/pkg/common/config"
 )
 
-var client http.Client
+var (
+	// define http client.
+	client = &http.Client{
+		Timeout: 15 * time.Second, // max timeout is 15s
+	}
+)
+
+func init() {
+	// reset http default transport
+	http.DefaultTransport.(*http.Transport).MaxConnsPerHost = 100 // default: 2
+}
 
 func Get(url string) (response []byte, err error) {
-	client := http.Client{Timeout: 5 * time.Second}
-	resp, err := client.Get(url)
+	hclient := http.Client{Timeout: 5 * time.Second}
+	resp, err := hclient.Get(url)
 	if err != nil {
 		return nil, err
 	}
+
 	defer resp.Body.Close()
 	body, err := io.ReadAll(resp.Body)
 	if err != nil {
@@ -47,26 +58,23 @@ func Get(url string) (response []byte, err error) {
 	return body, nil
 }
 
-func Post(
-	ctx context.Context,
-	url string,
-	header map[string]string,
-	data interface{},
-	timeout int,
-) (content []byte, err error) {
+func Post(ctx context.Context, url string, header map[string]string, data interface{}, timeout int) (content []byte, err error) {
 	if timeout > 0 {
 		var cancel func()
 		ctx, cancel = context.WithTimeout(ctx, time.Second*time.Duration(timeout))
 		defer cancel()
 	}
+
 	jsonStr, err := json.Marshal(data)
 	if err != nil {
 		return nil, err
 	}
+
 	req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewBuffer(jsonStr))
 	if err != nil {
 		return nil, err
 	}
+
 	if operationID, _ := ctx.Value(constant.OperationID).(string); operationID != "" {
 		req.Header.Set(constant.OperationID, operationID)
 	}
@@ -74,25 +82,22 @@ func Post(
 		req.Header.Set(k, v)
 	}
 	req.Header.Add("content-type", "application/json; charset=utf-8")
+
 	resp, err := client.Do(req)
 	if err != nil {
 		return nil, err
 	}
 	defer resp.Body.Close()
+
 	result, err := io.ReadAll(resp.Body)
 	if err != nil {
 		return nil, err
 	}
+
 	return result, nil
 }
 
-func PostReturn(
-	ctx context.Context,
-	url string,
-	header map[string]string,
-	input, output interface{},
-	timeOutSecond int,
-) error {
+func PostReturn(ctx context.Context, url string, header map[string]string, input, output interface{}, timeOutSecond int) error {
 	b, err := Post(ctx, url, header, input, timeOutSecond)
 	if err != nil {
 		return err
@@ -101,17 +106,13 @@ func PostReturn(
 	return err
 }
 
-func callBackPostReturn(
-	ctx context.Context,
-	url, command string,
-	input interface{},
-	output callbackstruct.CallbackResp,
-	callbackConfig config.CallBackConfig,
-) error {
+func callBackPostReturn(ctx context.Context, url, command string, input interface{}, output callbackstruct.CallbackResp, callbackConfig config.CallBackConfig) error {
 	defer log.ZDebug(ctx, "callback", "url", url, "command", command, "input", input, "callbackConfig", callbackConfig)
+
 	v := urllib.Values{}
 	v.Set(constant.CallbackCommand, command)
 	url = url + "?" + v.Encode()
+
 	b, err := Post(ctx, url, nil, input, callbackConfig.CallbackTimeOut)
 	if err != nil {
 		if callbackConfig.CallbackFailedContinue != nil && *callbackConfig.CallbackFailedContinue {
@@ -120,6 +121,7 @@ func callBackPostReturn(
 		}
 		return errs.ErrNetwork.Wrap(err.Error())
 	}
+
 	if err = json.Unmarshal(b, output); err != nil {
 		if callbackConfig.CallbackFailedContinue != nil && *callbackConfig.CallbackFailedContinue {
 			log.ZWarn(ctx, "callback failed but continue", err, "url", url)
@@ -127,15 +129,10 @@ func callBackPostReturn(
 		}
 		return errs.ErrData.Wrap(err.Error())
 	}
+
 	return output.Parse()
 }
 
-func CallBackPostReturn(
-	ctx context.Context,
-	url string,
-	req callbackstruct.CallbackReq,
-	resp callbackstruct.CallbackResp,
-	callbackConfig config.CallBackConfig,
-) error {
+func CallBackPostReturn(ctx context.Context, url string, req callbackstruct.CallbackReq, resp callbackstruct.CallbackResp, callbackConfig config.CallBackConfig) error {
 	return callBackPostReturn(ctx, url, req.GetCallbackCommand(), req, resp, callbackConfig)
 }