From c16cd0bf7cc3d9e5a18b7ad7cb89a8032e33d9e3 Mon Sep 17 00:00:00 2001 From: withchao <993506633@qq.com> Date: Wed, 10 Jan 2024 12:01:55 +0800 Subject: [PATCH] localcache --- internal/rpc/friend/friend.go | 6 ++++++ internal/rpc/msg/server.go | 7 +++++++ pkg/common/config/parse.go | 7 ------- 3 files changed, 13 insertions(+), 7 deletions(-) diff --git a/internal/rpc/friend/friend.go b/internal/rpc/friend/friend.go index 84702f548..02e123bac 100644 --- a/internal/rpc/friend/friend.go +++ b/internal/rpc/friend/friend.go @@ -16,6 +16,7 @@ package friend import ( "context" + "github.com/openimsdk/open-im-server/v3/pkg/common/config" "github.com/OpenIMSDK/tools/tx" @@ -54,6 +55,11 @@ type friendServer struct { } func Start(client registry.SvcDiscoveryRegistry, server *grpc.Server) error { + + config.Config.LocalCache.Friend.Topic = "friend" + config.Config.LocalCache.Friend.SlotNum = 500 + config.Config.LocalCache.Friend.SlotSize = 20000 + // Initialize MongoDB mongo, err := unrelation.NewMongo() if err != nil { diff --git a/internal/rpc/msg/server.go b/internal/rpc/msg/server.go index 0600031cd..1c919372f 100644 --- a/internal/rpc/msg/server.go +++ b/internal/rpc/msg/server.go @@ -16,6 +16,7 @@ package msg import ( "context" + "github.com/openimsdk/open-im-server/v3/pkg/common/config" "github.com/openimsdk/open-im-server/v3/pkg/rpccache" "google.golang.org/grpc" @@ -65,6 +66,11 @@ func (m *msgServer) execInterceptorHandler(ctx context.Context, req *msg.SendMsg } func Start(client discoveryregistry.SvcDiscoveryRegistry, server *grpc.Server) error { + + config.Config.LocalCache.Friend.Topic = "friend" + config.Config.LocalCache.Friend.SlotNum = 500 + config.Config.LocalCache.Friend.SlotSize = 20000 + rdb, err := cache.NewRedis() if err != nil { return err @@ -82,6 +88,7 @@ func Start(client discoveryregistry.SvcDiscoveryRegistry, server *grpc.Server) e userRpcClient := rpcclient.NewUserRpcClient(client) groupRpcClient := rpcclient.NewGroupRpcClient(client) msgDatabase := controller.NewCommonMsgDatabase(msgDocModel, cacheModel) + s := &msgServer{ Conversation: &conversationClient, User: &userRpcClient, diff --git a/pkg/common/config/parse.go b/pkg/common/config/parse.go index 6513079ab..1410a5a4a 100644 --- a/pkg/common/config/parse.go +++ b/pkg/common/config/parse.go @@ -110,10 +110,3 @@ func InitConfig(configFolderPath string) error { return initConfig(&Config.Notification, NotificationFileName, configFolderPath) } - -// todo test -func init() { - Config.LocalCache.Friend.Topic = "friend" - Config.LocalCache.Friend.SlotNum = 500 - Config.LocalCache.Friend.SlotSize = 20000 -}