merge newest branch

internal/rpc/user/get_user_info.go

@@ -0,0 +1,97 @@
+package user
+
+import (
+	"Open_IM/pkg/common/config"
+	"Open_IM/pkg/common/db/mysql_model/im_mysql_model"
+	"Open_IM/pkg/common/log"
+	"Open_IM/pkg/grpc-etcdv3/getcdv3"
+	pbUser "Open_IM/pkg/proto/user"
+	"Open_IM/pkg/utils"
+	"context"
+	"google.golang.org/grpc"
+	"net"
+	"strconv"
+	"strings"
+)
+
+type userServer struct {
+	rpcPort         int
+	rpcRegisterName string
+	etcdSchema      string
+	etcdAddr        []string
+}
+
+func NewUserServer(port int) *userServer {
+	log.NewPrivateLog("user")
+	return &userServer{
+		rpcPort:         port,
+		rpcRegisterName: config.Config.RpcRegisterName.OpenImUserName,
+		etcdSchema:      config.Config.Etcd.EtcdSchema,
+		etcdAddr:        config.Config.Etcd.EtcdAddr,
+	}
+}
+
+func (s *userServer) Run() {
+	log.Info("", "", "rpc user init....")
+
+	ip := utils.ServerIP
+	registerAddress := ip + ":" + strconv.Itoa(s.rpcPort)
+	//listener network
+	listener, err := net.Listen("tcp", registerAddress)
+	if err != nil {
+		log.InfoByArgs("listen network failed,err=%s", err.Error())
+		return
+	}
+	log.Info("", "", "listen network success, address = %s", registerAddress)
+	defer listener.Close()
+	//grpc server
+	srv := grpc.NewServer()
+	defer srv.GracefulStop()
+	//Service registers with etcd
+	pbUser.RegisterUserServer(srv, s)
+	err = getcdv3.RegisterEtcd(s.etcdSchema, strings.Join(s.etcdAddr, ","), ip, s.rpcPort, s.rpcRegisterName, 10)
+	if err != nil {
+		log.ErrorByArgs("register rpc token to etcd failed,err=%s", err.Error())
+		return
+	}
+	err = srv.Serve(listener)
+	if err != nil {
+		log.ErrorByArgs("listen token failed,err=%s", err.Error())
+		return
+	}
+	log.Info("", "", "rpc token init success")
+}
+
+func (s *userServer) GetUserInfo(ctx context.Context, req *pbUser.GetUserInfoReq) (*pbUser.GetUserInfoResp, error) {
+	log.InfoByKv("rpc get_user_info is server", req.OperationID)
+
+	var userInfoList []*pbUser.UserInfo
+	//Obtain user information according to userID
+	if len(req.UserIDList) > 0 {
+		for _, userID := range req.UserIDList {
+			var userInfo pbUser.UserInfo
+			user, err := im_mysql_model.FindUserByUID(userID)
+			if err != nil {
+				log.ErrorByKv("search userinfo failed", req.OperationID, "userID", userID, "err=%s", err.Error())
+				continue
+			}
+			userInfo.Uid = user.UID
+			userInfo.Icon = user.Icon
+			userInfo.Name = user.Name
+			userInfo.Gender = user.Gender
+			userInfo.Mobile = user.Mobile
+			userInfo.Birth = user.Birth
+			userInfo.Email = user.Email
+			userInfo.Ex = user.Ex
+			userInfoList = append(userInfoList, &userInfo)
+		}
+	} else {
+		return &pbUser.GetUserInfoResp{ErrorCode: 999, ErrorMsg: "uidList is nil"}, nil
+	}
+	log.InfoByKv("rpc get userInfo return success", req.OperationID, "token", req.Token)
+	return &pbUser.GetUserInfoResp{
+		ErrorCode: 0,
+		ErrorMsg:  "",
+		Data:      userInfoList,
+	}, nil
+}

internal/rpc/user/internal_service/get_user_info_to_client.go

@@ -0,0 +1,16 @@
+package internal_service
+
+import (
+	"Open_IM/pkg/common/config"
+	"Open_IM/pkg/grpc-etcdv3/getcdv3"
+	pbUser "Open_IM/pkg/proto/user"
+	"context"
+	"strings"
+)
+
+func GetUserInfoClient(req *pbUser.GetUserInfoReq) (*pbUser.GetUserInfoResp, error) {
+	etcdConn := getcdv3.GetConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImUserName)
+	client := pbUser.NewUserClient(etcdConn)
+	RpcResp, err := client.GetUserInfo(context.Background(), req)
+	return RpcResp, err
+}

internal/rpc/user/management_user.go

@@ -0,0 +1,61 @@
+/*
+** description("").
+** copyright('open-im,www.open-im.io').
+** author("fg,Gordon@tuoyun.net").
+** time(2021/9/15 10:28).
+ */
+package user
+
+import (
+	"Open_IM/pkg/common/config"
+	"Open_IM/pkg/common/db/mysql_model/im_mysql_model"
+	"Open_IM/pkg/common/log"
+	pbUser "Open_IM/pkg/proto/user"
+	"Open_IM/pkg/utils"
+	"context"
+)
+
+func (s *userServer) DeleteUsers(_ context.Context, req *pbUser.DeleteUsersReq) (*pbUser.DeleteUsersResp, error) {
+	log.InfoByKv("rpc DeleteUsers arrived server", req.OperationID, "args", req.String())
+	var resp pbUser.DeleteUsersResp
+	c, err := utils.ParseToken(req.Token)
+	if err != nil {
+		log.ErrorByKv("parse token failed", req.OperationID, "err", err.Error())
+		return &pbUser.DeleteUsersResp{CommonResp: &pbUser.CommonResp{ErrorCode: config.ErrParseToken.ErrCode, ErrorMsg: err.Error()}, FailedUidList: req.DeleteUidList}, nil
+	}
+	if !utils.IsContain(c.UID, config.Config.Manager.AppManagerUid) {
+		log.ErrorByKv(" Authentication failed", req.OperationID, "args", c)
+		return &pbUser.DeleteUsersResp{CommonResp: &pbUser.CommonResp{ErrorCode: 401, ErrorMsg: "not authorized"}, FailedUidList: req.DeleteUidList}, nil
+	}
+	for _, uid := range req.DeleteUidList {
+		err = im_mysql_model.UserDelete(uid)
+		if err != nil {
+			resp.CommonResp.ErrorCode = 201
+			resp.CommonResp.ErrorMsg = "some uid deleted failed"
+			resp.FailedUidList = append(resp.FailedUidList, uid)
+		}
+	}
+	return &resp, nil
+
+}
+
+func (s *userServer) GetAllUsersUid(_ context.Context, req *pbUser.GetAllUsersUidReq) (*pbUser.GetAllUsersUidResp, error) {
+	log.InfoByKv("rpc GetAllUsersUid arrived server", req.OperationID, "args", req.String())
+	c, err := utils.ParseToken(req.Token)
+	if err != nil {
+		log.InfoByKv("parse token failed", req.OperationID, "err", err.Error())
+		return &pbUser.GetAllUsersUidResp{CommonResp: &pbUser.CommonResp{ErrorCode: config.ErrParseToken.ErrCode, ErrorMsg: err.Error()}}, nil
+	}
+	if !utils.IsContain(c.UID, config.Config.Manager.AppManagerUid) {
+		log.ErrorByKv(" Authentication failed", req.OperationID, "args", c)
+		return &pbUser.GetAllUsersUidResp{CommonResp: &pbUser.CommonResp{ErrorCode: 401, ErrorMsg: "not authorized"}}, nil
+	}
+	uidList, err := im_mysql_model.SelectAllUID()
+	if err != nil {
+		log.ErrorByKv("db get failed", req.OperationID, "err", err.Error())
+		return &pbUser.GetAllUsersUidResp{CommonResp: &pbUser.CommonResp{ErrorCode: config.ErrMysql.ErrCode, ErrorMsg: err.Error()}}, nil
+	} else {
+		return &pbUser.GetAllUsersUidResp{CommonResp: &pbUser.CommonResp{ErrorCode: 0, ErrorMsg: ""}, UidList: uidList}, nil
+	}
+
+}

internal/rpc/user/update_user_info.go

@@ -0,0 +1,81 @@
+package user
+
+import (
+	"Open_IM/internal/push/logic"
+	"Open_IM/pkg/common/config"
+	"Open_IM/pkg/common/constant"
+	"Open_IM/pkg/common/db/mysql_model/im_mysql_model"
+	"Open_IM/pkg/common/log"
+	"Open_IM/pkg/grpc-etcdv3/getcdv3"
+	pbChat "Open_IM/pkg/proto/chat"
+	pbFriend "Open_IM/pkg/proto/friend"
+	pbUser "Open_IM/pkg/proto/user"
+	"Open_IM/pkg/utils"
+	"context"
+	"strings"
+)
+
+func (s *userServer) UpdateUserInfo(ctx context.Context, req *pbUser.UpdateUserInfoReq) (*pbUser.CommonResp, error) {
+	log.Info(req.Token, req.OperationID, "rpc modify user is server,args=%s", req.String())
+	claims, err := utils.ParseToken(req.Token)
+	if err != nil {
+		log.Error(req.Token, req.OperationID, "err=%s,parse token failed", err.Error())
+		return &pbUser.CommonResp{ErrorCode: config.ErrParseToken.ErrCode, ErrorMsg: err.Error()}, nil
+	}
+
+	ownerUid := ""
+	//if claims.UID == config.Config.AppManagerUid {
+	if utils.IsContain(claims.UID, config.Config.Manager.AppManagerUid) {
+		ownerUid = req.Uid
+	} else {
+		ownerUid = claims.UID
+	}
+
+	err = im_mysql_model.UpDateUserInfo(ownerUid, req.Name, req.Icon, req.Mobile, req.Birth, req.Email, req.Ex, req.Gender)
+	if err != nil {
+		log.Error(req.Token, req.OperationID, "update user some attribute failed,err=%s", err.Error())
+		return &pbUser.CommonResp{ErrorCode: config.ErrModifyUserInfo.ErrCode, ErrorMsg: config.ErrModifyUserInfo.ErrMsg}, nil
+	}
+	etcdConn := getcdv3.GetConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImFriendName)
+	client := pbFriend.NewFriendClient(etcdConn)
+	newReq := &pbFriend.GetFriendListReq{
+		OperationID: req.OperationID,
+		Token:       req.Token,
+	}
+
+	RpcResp, err := client.GetFriendList(context.Background(), newReq)
+	if err != nil {
+		log.ErrorByKv("get friend list rpc server failed", req.OperationID, "err", err.Error(), "req", req.String())
+		return &pbUser.CommonResp{}, nil
+	}
+	if RpcResp.ErrorCode != 0 {
+		log.ErrorByKv("get friend list rpc server failed", req.OperationID, "err", err.Error(), "req", req.String())
+		return &pbUser.CommonResp{}, nil
+	}
+	self, err := im_mysql_model.FindUserByUID(ownerUid)
+	if err != nil {
+		log.ErrorByKv("get self info failed", req.OperationID, "err", err.Error(), "req", req.String())
+		return &pbUser.CommonResp{}, nil
+	}
+	var name, faceUrl string
+	if self != nil {
+		name, faceUrl = self.Name, self.Icon
+	}
+	for _, v := range RpcResp.Data {
+		logic.SendMsgByWS(&pbChat.WSToMsgSvrChatMsg{
+			SendID:         ownerUid,
+			RecvID:         v.Uid,
+			SenderNickName: name,
+			SenderFaceURL:  faceUrl,
+			Content:        ownerUid + "'s info has changed",
+			SendTime:       utils.GetCurrentTimestampByNano(),
+			MsgFrom:        constant.SysMsgType,
+			ContentType:    constant.SetSelfInfoTip,
+			SessionType:    constant.SingleChatType,
+			OperationID:    req.OperationID,
+			Token:          req.Token,
+		})
+
+	}
+	return &pbUser.CommonResp{}, nil
+}