feat: add api caller timeout

openimsdk · Oct 25, 2023 · fd9b622 · fd9b622
1 parent 02acbfe
commit fd9b622
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/pkg/common/apicall/call.go b/pkg/common/apicall/call.go
@@ -21,6 +21,7 @@ import (
 	"errors"
 	"io"
 	"net/http"
+	"time"
 
 	"github.com/OpenIMSDK/chat/pkg/common/constant"
 	constant2 "github.com/OpenIMSDK/protocol/constant"
@@ -36,6 +37,10 @@ type baseApiResponse[T any] struct {
 	Data    *T     `json:"data"`
 }
 
+var client = &http.Client{
+	Timeout: time.Second * 20,
+}
+
 type ApiCaller[Req, Resp any] interface {
 	Call(ctx context.Context, req *Req) (*Resp, error)
 }
@@ -64,6 +69,10 @@ func (a caller[Req, Resp]) Call(ctx context.Context, req *Req) (*Resp, error) {
 
 func (a caller[Req, Resp]) call(ctx context.Context, req *Req) (*Resp, error) {
 	url := a.prefix() + a.api
+	start := time.Now()
+	defer func() {
+		log.ZDebug(ctx, "call caller", "api", a.api, "cost", time.Since(start))
+	}()
 	log.ZInfo(ctx, "caller req", "addr", url, "req", req)
 	reqBody, err := json.Marshal(req)
 	if err != nil {
@@ -79,7 +88,7 @@ func (a caller[Req, Resp]) call(ctx context.Context, req *Req) (*Resp, error) {
 		request.Header.Set(constant2.Token, token)
 		log.ZDebug(ctx, "req token", "token", token)
 	}
-	response, err := http.DefaultClient.Do(request)
+	response, err := client.Do(request)
 	if err != nil {
 		return nil, err
 	}