From 38d3c5d94dc879b97e649605c3bfdc7938d1cb4a Mon Sep 17 00:00:00 2001 From: Gordon <1432970085@qq.com> Date: Thu, 16 Mar 2023 13:01:42 +0800 Subject: [PATCH] user update --- internal/rpc/user/user.go | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/internal/rpc/user/user.go b/internal/rpc/user/user.go index 7f340665b..b7d231b22 100644 --- a/internal/rpc/user/user.go +++ b/internal/rpc/user/user.go @@ -24,7 +24,7 @@ type userServer struct { controller.UserDatabase notification *notification.Check userCheck *check.UserCheck - ConversationChecker *check.ConversationChecker + conversationChecker *check.ConversationChecker RegisterCenter registry.SvcDiscoveryRegistry friendCheck *check.FriendChecker } @@ -45,10 +45,12 @@ func Start(client registry.SvcDiscoveryRegistry, server *grpc.Server) error { users = append(users, &tablerelation.UserModel{UserID: v, Nickname: config.Config.Manager.Nickname[k]}) } u := &userServer{ - UserDatabase: controller.NewUserDatabase(relation.NewUserGorm(db)), - notification: notification.NewCheck(client), - userCheck: check.NewUserCheck(client), - RegisterCenter: client, + UserDatabase: controller.NewUserDatabase(relation.NewUserGorm(db)), + notification: notification.NewCheck(client), + userCheck: check.NewUserCheck(client), + friendCheck: check.NewFriendChecker(client), + conversationChecker: check.NewConversationChecker(client), + RegisterCenter: client, } pbuser.RegisterUserServer(server, u) return u.UserDatabase.InitOnce(context.Background(), users)