feat: use robot to migrate code

Signed-off-by: kubbot & kubecub <3293172751ysy@gmail.com>
2026-04-28 14:29:19 +08:00 · 2023-06-30 09:45:02 +08:00
parent 2d41819008
commit 539e0fdfb6
529 changed files with 64588 additions and 54413 deletions
@@ -0,0 +1,119 @@
+package msg
+
+import (
+	"context"
+
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/constant"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/cache"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/controller"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/localcache"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/tx"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/unrelation"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/prome"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/discoveryregistry"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/conversation"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/msg"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/rpcclient"
+	"google.golang.org/grpc"
+)
+
+type MessageInterceptorChain []MessageInterceptorFunc
+type msgServer struct {
+	RegisterCenter         discoveryregistry.SvcDiscoveryRegistry
+	MsgDatabase            controller.CommonMsgDatabase
+	ExtendMsgDatabase      controller.ExtendMsgDatabase
+	Group                  *rpcclient.GroupRpcClient
+	User                   *rpcclient.UserRpcClient
+	Conversation           *rpcclient.ConversationRpcClient
+	friend                 *rpcclient.FriendRpcClient
+	GroupLocalCache        *localcache.GroupLocalCache
+	ConversationLocalCache *localcache.ConversationLocalCache
+	MessageLocker          MessageLocker
+	Handlers               MessageInterceptorChain
+	notificationSender     *rpcclient.NotificationSender
+}
+
+func (m *msgServer) addInterceptorHandler(interceptorFunc ...MessageInterceptorFunc) {
+	m.Handlers = append(m.Handlers, interceptorFunc...)
+}
+
+func (m *msgServer) execInterceptorHandler(ctx context.Context, req *msg.SendMsgReq) error {
+	for _, handler := range m.Handlers {
+		msgData, err := handler(ctx, req)
+		if err != nil {
+			return err
+		}
+		req.MsgData = msgData
+	}
+	return nil
+}
+
+func Start(client discoveryregistry.SvcDiscoveryRegistry, server *grpc.Server) error {
+	rdb, err := cache.NewRedis()
+	if err != nil {
+		return err
+	}
+	mongo, err := unrelation.NewMongo()
+	if err != nil {
+		return err
+	}
+	if err := mongo.CreateMsgIndex(); err != nil {
+		return err
+	}
+	cacheModel := cache.NewMsgCacheModel(rdb)
+	msgDocModel := unrelation.NewMsgMongoDriver(mongo.GetDatabase())
+	extendMsgModel := unrelation.NewExtendMsgSetMongoDriver(mongo.GetDatabase())
+	extendMsgCacheModel := cache.NewExtendMsgSetCacheRedis(rdb, extendMsgModel, cache.GetDefaultOpt())
+	extendMsgDatabase := controller.NewExtendMsgDatabase(extendMsgModel, extendMsgCacheModel, tx.NewMongo(mongo.GetClient()))
+	msgDatabase := controller.NewCommonMsgDatabase(msgDocModel, cacheModel)
+	conversationClient := rpcclient.NewConversationRpcClient(client)
+	userRpcClient := rpcclient.NewUserRpcClient(client)
+	groupRpcClient := rpcclient.NewGroupRpcClient(client)
+	friendRpcClient := rpcclient.NewFriendRpcClient(client)
+	s := &msgServer{
+		Conversation:           &conversationClient,
+		User:                   &userRpcClient,
+		Group:                  &groupRpcClient,
+		MsgDatabase:            msgDatabase,
+		ExtendMsgDatabase:      extendMsgDatabase,
+		RegisterCenter:         client,
+		GroupLocalCache:        localcache.NewGroupLocalCache(client),
+		ConversationLocalCache: localcache.NewConversationLocalCache(client),
+		friend:                 &friendRpcClient,
+		MessageLocker:          NewLockerMessage(cacheModel),
+	}
+	s.notificationSender = rpcclient.NewNotificationSender(rpcclient.WithLocalSendMsg(s.SendMsg))
+	s.addInterceptorHandler(MessageHasReadEnabled)
+	s.initPrometheus()
+	msg.RegisterMsgServer(server, s)
+	return nil
+}
+
+func (m *msgServer) initPrometheus() {
+	prome.NewMsgPullFromRedisSuccessCounter()
+	prome.NewMsgPullFromRedisFailedCounter()
+	prome.NewMsgPullFromMongoSuccessCounter()
+	prome.NewMsgPullFromMongoFailedCounter()
+	prome.NewSingleChatMsgRecvSuccessCounter()
+	prome.NewGroupChatMsgRecvSuccessCounter()
+	prome.NewWorkSuperGroupChatMsgRecvSuccessCounter()
+	prome.NewSingleChatMsgProcessSuccessCounter()
+	prome.NewSingleChatMsgProcessFailedCounter()
+	prome.NewGroupChatMsgProcessSuccessCounter()
+	prome.NewGroupChatMsgProcessFailedCounter()
+	prome.NewWorkSuperGroupChatMsgProcessSuccessCounter()
+	prome.NewWorkSuperGroupChatMsgProcessFailedCounter()
+}
+
+func (m *msgServer) conversationAndGetRecvID(conversation *conversation.Conversation, userID string) (recvID string) {
+	if conversation.ConversationType == constant.SingleChatType || conversation.ConversationType == constant.NotificationChatType {
+		if userID == conversation.OwnerUserID {
+			recvID = conversation.UserID
+		} else {
+			recvID = conversation.OwnerUserID
+		}
+	} else if conversation.ConversationType == constant.SuperGroupChatType {
+		recvID = conversation.GroupID
+	}
+	return
+}