From ece2498d18731bcac3d497863f21c2e95fb42b43 Mon Sep 17 00:00:00 2001 From: wangchuxiao Date: Mon, 10 Jul 2023 10:34:44 +0800 Subject: [PATCH] cron use rpc mw --- internal/tools/msg.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/internal/tools/msg.go b/internal/tools/msg.go index f962c0a01..ef2175bf0 100644 --- a/internal/tools/msg.go +++ b/internal/tools/msg.go @@ -15,10 +15,13 @@ import ( "github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/unrelation" "github.com/OpenIMSDK/Open-IM-Server/pkg/common/log" "github.com/OpenIMSDK/Open-IM-Server/pkg/common/mcontext" + "github.com/OpenIMSDK/Open-IM-Server/pkg/common/mw" "github.com/OpenIMSDK/Open-IM-Server/pkg/discoveryregistry/zookeeper" "github.com/OpenIMSDK/Open-IM-Server/pkg/rpcclient" "github.com/OpenIMSDK/Open-IM-Server/pkg/rpcclient/notification" "github.com/OpenIMSDK/Open-IM-Server/pkg/utils" + "google.golang.org/grpc" + "google.golang.org/grpc/credentials/insecure" ) type MsgTool struct { @@ -61,6 +64,7 @@ func InitMsgTool() (*MsgTool, error) { if err != nil { return nil, err } + discov.AddOption(mw.GrpcClient(), grpc.WithTransportCredentials(insecure.NewCredentials())) userDB := relation.NewUserGorm(db) msgDatabase := controller.InitCommonMsgDatabase(rdb, mongo.GetDatabase()) userDatabase := controller.NewUserDatabase(userDB, cache.NewUserCacheRedis(rdb, relation.NewUserGorm(db), cache.GetDefaultOpt()), tx.NewGorm(db))