conn

internal/api/statistics.go

@@ -2,29 +2,34 @@ package api
 
 import (
 	"context"
+
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/a2r"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/config"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/discoveryregistry"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/user"
 	"github.com/gin-gonic/gin"
+	"google.golang.org/grpc"
 )
 
 func NewStatistics(discov discoveryregistry.SvcDiscoveryRegistry) *Statistics {
-	return &Statistics{discov: discov}
+	conn, err := discov.GetConn(context.Background(), config.Config.RpcRegisterName.OpenImUserName)
+	if err != nil {
+		panic(err)
+	}
+	client := user.NewUserClient(conn)
+	return &Statistics{discov: discov, client: client, conn: conn}
 }
 
 type Statistics struct {
+	conn   *grpc.ClientConn
+	client user.UserClient
 	discov discoveryregistry.SvcDiscoveryRegistry
 }
 
-func (s *Statistics) userClient(ctx context.Context) (user.UserClient, error) {
-	conn, err := s.discov.GetConn(ctx, config.Config.RpcRegisterName.OpenImUserName)
-	if err != nil {
-		return nil, err
-	}
-	return user.NewUserClient(conn), nil
+func (s *Statistics) Client() user.UserClient {
+	return s.client
 }
 
 func (s *Statistics) UserRegister(c *gin.Context) {
-	a2r.Call(user.UserClient.UserRegisterCount, s.userClient, c)
+	a2r.Call(user.UserClient.UserRegisterCount, s.Client, c)
 }