mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-11-04 19:32:17 +08:00
97 lines
3.3 KiB
Go
97 lines
3.3 KiB
Go
package friend
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/openimsdk/open-im-server/v3/internal/rpc/incrversion"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/authverify"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/storage/model"
|
|
"github.com/openimsdk/protocol/relation"
|
|
"github.com/openimsdk/tools/errs"
|
|
)
|
|
|
|
func (s *friendServer) NotificationUserInfoUpdate(ctx context.Context, req *relation.NotificationUserInfoUpdateReq) (*relation.NotificationUserInfoUpdateResp, error) {
|
|
if req.NewUserInfo == nil {
|
|
var err error
|
|
req.NewUserInfo, err = s.userRpcClient.GetUserInfo(ctx, req.UserID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
if req.UserID != req.NewUserInfo.UserID {
|
|
return nil, errs.ErrArgs.WrapMsg("req.UserID != req.NewUserInfo.UserID")
|
|
}
|
|
userIDs, err := s.friendDatabase.FindFriendUserID(ctx, req.UserID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(userIDs) > 0 {
|
|
if err := s.friendDatabase.UpdateFriendUserInfo(ctx, req.UserID, userIDs, req.NewUserInfo.Nickname, req.NewUserInfo.FaceURL); err != nil {
|
|
return nil, err
|
|
}
|
|
s.notificationSender.FriendsInfoUpdateNotification(ctx, req.UserID, userIDs)
|
|
}
|
|
return &relation.NotificationUserInfoUpdateResp{}, nil
|
|
}
|
|
|
|
func (s *friendServer) SearchFriends(ctx context.Context, req *relation.SearchFriendsReq) (*relation.SearchFriendsResp, error) {
|
|
if err := s.userRpcClient.Access(ctx, req.UserID); err != nil {
|
|
return nil, err
|
|
}
|
|
if req.Keyword == "" {
|
|
total, friends, err := s.friendDatabase.PageOwnerFriends(ctx, req.UserID, req.Pagination)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &relation.SearchFriendsResp{
|
|
Total: total,
|
|
Friends: friendsDB2PB(friends),
|
|
}, nil
|
|
}
|
|
total, friends, err := s.friendDatabase.SearchFriend(ctx, req.UserID, req.Keyword, req.Pagination)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &relation.SearchFriendsResp{
|
|
Total: total,
|
|
Friends: friendsDB2PB(friends),
|
|
}, nil
|
|
}
|
|
|
|
func (s *friendServer) GetIncrementalFriends(ctx context.Context, req *relation.GetIncrementalFriendsReq) (*relation.GetIncrementalFriendsResp, error) {
|
|
if err := authverify.CheckAccessV3(ctx, req.UserID, s.config.Share.IMAdminUserID); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
opt := incrversion.Option[*relation.FriendInfo, relation.GetIncrementalFriendsResp]{
|
|
Ctx: ctx,
|
|
VersionKey: req.UserID,
|
|
VersionID: req.VersionID,
|
|
VersionNumber: req.Version,
|
|
SyncLimit: s.config.RpcConfig.FriendSyncCount,
|
|
Version: s.friendDatabase.FindFriendIncrVersion,
|
|
CacheMaxVersion: s.friendDatabase.FindMaxFriendVersionCache,
|
|
SortID: s.friendDatabase.FindSortFriendUserIDs,
|
|
Find: func(ctx context.Context, ids []string) ([]*relation.FriendInfo, error) {
|
|
friends, err := s.friendDatabase.FindFriendsWithError(ctx, req.UserID, ids)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return friendsDB2PB(friends), nil
|
|
},
|
|
|
|
ID: func(elem *relation.FriendInfo) string { return elem.FriendUserID },
|
|
Resp: func(version *model.VersionLog, delIDs []string, list []*relation.FriendInfo, full bool) *relation.GetIncrementalFriendsResp {
|
|
return &relation.GetIncrementalFriendsResp{
|
|
VersionID: version.ID.Hex(),
|
|
Version: uint64(version.Version),
|
|
Full: full,
|
|
SyncCount: uint32(s.config.RpcConfig.FriendSyncCount),
|
|
DeleteUserIds: delIDs,
|
|
Changes: list,
|
|
}
|
|
},
|
|
}
|
|
return opt.Build()
|
|
}
|