chore: add api timeout and im admin token cache #241

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 * 10,
+}
+
 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
 	}

pkg/common/apicall/caller.go

@@ -16,7 +16,9 @@ package apicall
 
 import (
 	"context"
-	"fmt"
+	"github.com/OpenIMSDK/tools/log"
+	"sync"
+	"time"
 
 	"github.com/OpenIMSDK/chat/pkg/common/config"
 	"github.com/OpenIMSDK/protocol/auth"
@@ -40,7 +42,11 @@ type CallerInterface interface {
 	FriendUserIDs(ctx context.Context, userID string) ([]string, error)
 }
 
-type Caller struct{}
+type Caller struct {
+	token   string
+	timeout time.Time
+	lock    sync.Mutex
+}
 
 func NewCallerInterface() CallerInterface {
 	return &Caller{}
@@ -58,11 +64,23 @@ func (c *Caller) ImportFriend(ctx context.Context, ownerUserID string, friendUse
 }
 
 func (c *Caller) ImAdminTokenWithDefaultAdmin(ctx context.Context) (string, error) {
-	return c.UserToken(ctx, config.GetDefaultIMAdmin(), constant.AdminPlatformID)
+	c.lock.Lock()
+	defer c.lock.Unlock()
+	if c.token == "" || c.timeout.Before(time.Now()) {
+		userID := config.GetDefaultIMAdmin()
+		token, err := c.UserToken(ctx, config.GetDefaultIMAdmin(), constant.AdminPlatformID)
+		if err != nil {
+			log.ZError(ctx, "get im admin token", err, "userID", userID)
+			return "", err
+		}
+		log.ZDebug(ctx, "get im admin token", "userID", userID)
+		c.token = token
+		c.timeout = time.Now().Add(time.Minute * 5)
+	}
+	return c.token, nil
 }
 
 func (c *Caller) UserToken(ctx context.Context, userID string, platformID int32) (string, error) {
-	fmt.Println(*config.Config.Secret)
 	resp, err := userToken.Call(ctx, &auth.UserTokenReq{
 		Secret:     *config.Config.Secret,
 		PlatformID: platformID,