From 87ad8b7f6811c75ceb2e0bbd1497e4246288f3d8 Mon Sep 17 00:00:00 2001 From: wangchuxiao Date: Fri, 30 Jun 2023 13:13:06 +0800 Subject: [PATCH] api call user once --- internal/api/route.go | 8 ++++---- internal/api/user.go | 4 ++++ 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/internal/api/route.go b/internal/api/route.go index 67bd75f73..b24aa82c9 100644 --- a/internal/api/route.go +++ b/internal/api/route.go @@ -2,6 +2,7 @@ package api import ( "context" + "github.com/OpenIMSDK/Open-IM-Server/pkg/common/config" "github.com/OpenIMSDK/Open-IM-Server/pkg/common/log" "github.com/OpenIMSDK/Open-IM-Server/pkg/common/mw" @@ -27,6 +28,7 @@ func NewGinRouter(discov discoveryregistry.SvcDiscoveryRegistry, rdb redis.Unive } log.ZInfo(context.Background(), "load config", "config", config.Config) r.Use(gin.Recovery(), mw.CorsHandler(), mw.GinParseOperationID()) + u := NewUserApi(discov) if config.Config.Prometheus.Enable { prome.NewApiRequestCounter() prome.NewApiRequestFailedCounter() @@ -36,7 +38,6 @@ func NewGinRouter(discov discoveryregistry.SvcDiscoveryRegistry, rdb redis.Unive } userRouterGroup := r.Group("/user") { - u := NewUserApi(discov) userRouterGroupChild := mw.NewRouterGroup(userRouterGroup, "") userRouterGroupChildToken := mw.NewRouterGroup(userRouterGroup, "", mw.WithGinParseToken(rdb)) userRouterGroupChild.POST("/user_register", u.UserRegister) @@ -105,7 +106,6 @@ func NewGinRouter(discov discoveryregistry.SvcDiscoveryRegistry, rdb redis.Unive authRouterGroup := r.Group("/auth") { a := NewAuthApi(discov) - u := NewUserApi(discov) authRouterGroupChild := mw.NewRouterGroup(authRouterGroup, "") authRouterGroupChildToken := mw.NewRouterGroup(authRouterGroup, "", mw.WithGinParseToken(rdb)) authRouterGroupChild.POST("/user_register", u.UserRegister) //1 @@ -173,9 +173,9 @@ func NewGinRouter(discov discoveryregistry.SvcDiscoveryRegistry, rdb redis.Unive statisticsGroup := r.Group("/statistics") { - s := NewStatisticsApi(discov) + // s := NewStatisticsApi(discov) conversationGroup.Use(mw.GinParseToken(rdb)) - statisticsGroup.POST("/user_register", s.UserRegister) + statisticsGroup.POST("/user_register", u.UserRegisterCount) } return r } diff --git a/internal/api/user.go b/internal/api/user.go index ba3b6f59f..18411ec2a 100644 --- a/internal/api/user.go +++ b/internal/api/user.go @@ -57,3 +57,7 @@ func (u *UserApi) GetUsersOnlineStatus(c *gin.Context) { return } } + +func (s *UserApi) UserRegisterCount(c *gin.Context) { + a2r.Call(user.UserClient.UserRegisterCount, s.Client, c) +}