Open-IM-Server/internal/rpc/friend/get_friend_apply_list.go

package friend

import (
	"Open_IM/pkg/common/config"
	"Open_IM/pkg/common/db/mysql_model/im_mysql_model"
	"Open_IM/pkg/common/log"
	pbFriend "Open_IM/pkg/proto/friend"
	"Open_IM/pkg/utils"
	"context"
	"fmt"
	"strconv"
)

func (s *friendServer) GetFriendApplyList(ctx context.Context, req *pbFriend.GetFriendApplyReq) (*pbFriend.GetFriendApplyResp, error) {
	log.Info(req.Token, req.OperationID, "rpc get friend apply list is server,args=%s", req.String())
	var appleUserList []*pbFriend.ApplyUserInfo
	//Parse token, to find current user information
	claims, err := utils.ParseToken(req.Token)
	if err != nil {
		log.Error(req.Token, req.OperationID, "err=%s,parse token failed", err.Error())
		return &pbFriend.GetFriendApplyResp{ErrorCode: config.ErrParseToken.ErrCode, ErrorMsg: config.ErrParseToken.ErrMsg}, nil
	}
	//	Find the  current user friend applications received
	ApplyUsersInfo, err := im_mysql_model.FindFriendsApplyFromFriendReq(claims.UID)
	if err != nil {
		log.Error(req.Token, req.OperationID, "err=%s,search applyInfo failed", err.Error())
		return &pbFriend.GetFriendApplyResp{ErrorCode: config.ErrMysql.ErrCode, ErrorMsg: config.ErrMysql.ErrMsg}, nil
	}
	for _, applyUserInfo := range ApplyUsersInfo {
		var userInfo pbFriend.ApplyUserInfo
		//Find friend application status
		userInfo.Flag = applyUserInfo.Flag
		userInfo.ReqMessage = applyUserInfo.ReqMessage
		userInfo.ApplyTime = strconv.FormatInt(applyUserInfo.CreateTime.Unix(), 10)
		//Find  user information
		us, err := im_mysql_model.FindUserByUID(applyUserInfo.ReqId)
		if err != nil {
			log.Error(req.Token, req.OperationID, "err=%s,search userInfo failed", err.Error())
			continue
		}
		userInfo.Uid = us.UID
		userInfo.Icon = us.Icon
		userInfo.Name = us.Name
		userInfo.Gender = us.Gender
		userInfo.Mobile = us.Mobile
		userInfo.Birth = us.Birth
		userInfo.Email = us.Email
		userInfo.Ex = us.Ex
		appleUserList = append(appleUserList, &userInfo)

	}
	log.Info(req.Token, req.OperationID, fmt.Sprintf("rpc get friendapplylist success return"))
	return &pbFriend.GetFriendApplyResp{Data: appleUserList}, nil
}

func (s *friendServer) GetSelfApplyList(ctx context.Context, req *pbFriend.GetFriendApplyReq) (*pbFriend.GetFriendApplyResp, error) {
	log.Info(req.Token, req.OperationID, "rpc get self apply list is server,args=%s", req.String())
	var selfApplyOtherUserList []*pbFriend.ApplyUserInfo
	//Parse token, to find current user information
	claims, err := utils.ParseToken(req.Token)
	if err != nil {
		log.Error(req.Token, req.OperationID, "err=%s,parse token failed", err.Error())
		return &pbFriend.GetFriendApplyResp{ErrorCode: config.ErrParseToken.ErrCode, ErrorMsg: config.ErrParseToken.ErrMsg}, nil
	}
	//	Find the self add other userinfo
	usersInfo, err := im_mysql_model.FindSelfApplyFromFriendReq(claims.UID)
	if err != nil {
		log.Error(req.Token, req.OperationID, "err=%s,search self to other user Info failed", err.Error())
		return &pbFriend.GetFriendApplyResp{ErrorCode: config.ErrMysql.ErrCode, ErrorMsg: config.ErrMysql.ErrMsg}, nil
	}
	for _, selfApplyOtherUserInfo := range usersInfo {
		var userInfo pbFriend.ApplyUserInfo
		//Find friend application status
		userInfo.Flag = selfApplyOtherUserInfo.Flag
		userInfo.ReqMessage = selfApplyOtherUserInfo.ReqMessage
		userInfo.ApplyTime = strconv.FormatInt(selfApplyOtherUserInfo.CreateTime.Unix(), 10)
		//Find  user information
		us, err := im_mysql_model.FindUserByUID(selfApplyOtherUserInfo.Uid)
		if err != nil {
			log.Error(req.Token, req.OperationID, "err=%s,search userInfo failed", err.Error())
			continue
		}
		userInfo.Uid = us.UID
		userInfo.Icon = us.Icon
		userInfo.Name = us.Name
		userInfo.Gender = us.Gender
		userInfo.Mobile = us.Mobile
		userInfo.Birth = us.Birth
		userInfo.Email = us.Email
		userInfo.Ex = us.Ex
		selfApplyOtherUserList = append(selfApplyOtherUserList, &userInfo)
	}
	log.Info(req.Token, req.OperationID, fmt.Sprintf("rpc get self apply list success return"))
	return &pbFriend.GetFriendApplyResp{Data: selfApplyOtherUserList}, nil
}