Open-IM-Server/internal/rpc/msg/server.go

package msg

import (
	"Open_IM/internal/common/check"
	"Open_IM/pkg/common/db/controller"
	"Open_IM/pkg/common/db/localcache"
	"Open_IM/pkg/common/db/relation"
	tablerelation "Open_IM/pkg/common/db/table/relation"
	discoveryRegistry "Open_IM/pkg/discoveryregistry"
	"github.com/OpenIMSDK/openKeeper"

	promePkg "Open_IM/pkg/common/prometheus"
	"Open_IM/pkg/proto/msg"
	"google.golang.org/grpc"
)

type msgServer struct {
	RegisterCenter discoveryRegistry.SvcDiscoveryRegistry
	MsgInterface   controller.MsgInterface
	Group          *check.GroupChecker
	User           *check.UserCheck
	Conversation   *check.ConversationChecker
	friend         *check.FriendChecker
	*localcache.GroupLocalCache
	black *check.BlackChecker
}

type deleteMsg struct {
	UserID      string
	OpUserID    string
	SeqList     []uint32
	OperationID string
}

func Start(client *openKeeper.ZkClient, server *grpc.Server) error {
	mysql, err := relation.NewGormDB()
	if err != nil {
		return err
	}
	if err := mysql.AutoMigrate(&tablerelation.UserModel{}); err != nil {
		return err
	}
	s := &msgServer{
		Conversation: check.NewConversationChecker(client),
		User:         check.NewUserCheck(client),
		Group:        check.NewGroupChecker(client),
		//MsgInterface: controller.MsgInterface(),
		RegisterCenter:  client,
		GroupLocalCache: localcache.NewGroupMemberIDsLocalCache(client),
		black:           check.NewBlackChecker(client),
		friend:          check.NewFriendChecker(client),
	}
	s.initPrometheus()
	msg.RegisterMsgServer(server, s)
	return nil
}

func (m *msgServer) initPrometheus() {
	promePkg.NewMsgPullFromRedisSuccessCounter()
	promePkg.NewMsgPullFromRedisFailedCounter()
	promePkg.NewMsgPullFromMongoSuccessCounter()
	promePkg.NewMsgPullFromMongoFailedCounter()
	promePkg.NewSingleChatMsgRecvSuccessCounter()
	promePkg.NewGroupChatMsgRecvSuccessCounter()
	promePkg.NewWorkSuperGroupChatMsgRecvSuccessCounter()
	promePkg.NewSingleChatMsgProcessSuccessCounter()
	promePkg.NewSingleChatMsgProcessFailedCounter()
	promePkg.NewGroupChatMsgProcessSuccessCounter()
	promePkg.NewGroupChatMsgProcessFailedCounter()
	promePkg.NewWorkSuperGroupChatMsgProcessSuccessCounter()
	promePkg.NewWorkSuperGroupChatMsgProcessFailedCounter()
}