withchao
1 year ago
parent
2f97933f6c
commit
1f02bdc267
@ -0,0 +1,26 @@
|
||||
package friend
|
||||
|
||||
import (
|
||||
relationtb "github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
||||
"github.com/openimsdk/protocol/friend"
|
||||
"github.com/openimsdk/tools/utils/datautil"
|
||||
)
|
||||
|
||||
func friendDB2PB(db *relationtb.FriendModel) *friend.FriendInfo {
|
||||
return &friend.FriendInfo{
|
||||
OwnerUserID: db.OwnerUserID,
|
||||
FriendUserID: db.FriendUserID,
|
||||
FriendNickname: db.FriendNickname,
|
||||
FriendFaceURL: db.FriendFaceURL,
|
||||
Remark: db.Remark,
|
||||
CreateTime: db.CreateTime.UnixMilli(),
|
||||
AddSource: db.AddSource,
|
||||
OperatorUserID: db.OperatorUserID,
|
||||
Ex: db.Ex,
|
||||
IsPinned: db.IsPinned,
|
||||
}
|
||||
}
|
||||
|
||||
func friendsDB2PB(db []*relationtb.FriendModel) []*friend.FriendInfo {
|
||||
return datautil.Slice(db, friendDB2PB)
|
||||
}
|
||||
@ -0,0 +1,65 @@
|
||||
package friend
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/md5"
|
||||
"encoding/binary"
|
||||
"encoding/json"
|
||||
"github.com/openimsdk/open-im-server/v3/pkg/common/db/dataver"
|
||||
"github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
||||
pbfriend "github.com/openimsdk/protocol/friend"
|
||||
)
|
||||
|
||||
func (s *friendServer) SearchFriends(ctx context.Context, req *pbfriend.SearchFriendsReq) (*pbfriend.SearchFriendsResp, error) {
|
||||
//TODO implement me
|
||||
panic("implement me")
|
||||
}
|
||||
|
||||
func (s *friendServer) sortFriendUserIDsHash(userIDs []string) uint64 {
|
||||
data, _ := json.Marshal(userIDs)
|
||||
sum := md5.Sum(data)
|
||||
return binary.BigEndian.Uint64(sum[:])
|
||||
}
|
||||
|
||||
func (s *friendServer) IncrSyncFriends(ctx context.Context, req *pbfriend.IncrSyncFriendsReq) (*pbfriend.IncrSyncFriendsResp, error) {
|
||||
var limit int
|
||||
if req.Version > 0 {
|
||||
limit = s.config.RpcConfig.FriendSyncCount
|
||||
}
|
||||
incrVer, err := s.friendDatabase.FindFriendIncrVersion(ctx, req.UserID, uint(req.Version), limit)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
sortUserIDs, err := s.friendDatabase.FindSortFriendUserIDs(ctx, req.UserID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if len(sortUserIDs) == 0 {
|
||||
return &pbfriend.IncrSyncFriendsResp{
|
||||
Version: uint64(incrVer.Version),
|
||||
Full: true,
|
||||
SyncCount: uint32(s.config.RpcConfig.FriendSyncCount),
|
||||
}, nil
|
||||
}
|
||||
var changes []*relation.FriendModel
|
||||
res := dataver.NewSyncResult(incrVer, sortUserIDs)
|
||||
if len(res.Changes) > 0 {
|
||||
changes, err = s.friendDatabase.FindFriendsWithError(ctx, req.UserID, res.Changes)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
calcHash := s.sortFriendUserIDsHash(sortUserIDs)
|
||||
if calcHash == req.IdHash {
|
||||
sortUserIDs = nil
|
||||
}
|
||||
return &pbfriend.IncrSyncFriendsResp{
|
||||
Version: uint64(res.Version),
|
||||
Full: res.Full,
|
||||
SyncCount: uint32(s.config.RpcConfig.FriendSyncCount),
|
||||
SortUserIdHash: calcHash,
|
||||
SortUserIds: sortUserIDs,
|
||||
DeleteUserIds: res.DeleteEID,
|
||||
Changes: friendsDB2PB(changes),
|
||||
}, nil
|
||||
}
|
||||
@ -1,32 +1,31 @@
|
||||
package dataver
|
||||
|
||||
type SyncResult[T any] struct {
|
||||
import "github.com/openimsdk/tools/utils/datautil"
|
||||
|
||||
type SyncResult struct {
|
||||
Version uint
|
||||
DeleteEID []string
|
||||
Changes []T
|
||||
Changes []string
|
||||
Full bool
|
||||
}
|
||||
|
||||
func NewSyncResult[T any](wl *WriteLog, find func(eIds []string) ([]T, error)) (*SyncResult[T], error) {
|
||||
func NewSyncResult(wl *WriteLog, fullIDs []string) *SyncResult {
|
||||
var findEIDs []string
|
||||
var res SyncResult[T]
|
||||
var res SyncResult
|
||||
if wl.Full() {
|
||||
res.Changes = fullIDs
|
||||
res.Full = true
|
||||
} else {
|
||||
idSet := datautil.SliceSet(fullIDs)
|
||||
for _, l := range wl.Logs {
|
||||
if l.Deleted {
|
||||
res.DeleteEID = append(res.DeleteEID, l.EID)
|
||||
} else {
|
||||
findEIDs = append(findEIDs, l.EID)
|
||||
if _, ok := idSet[l.EID]; ok {
|
||||
findEIDs = append(findEIDs, l.EID)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
if res.Full || len(findEIDs) > 0 {
|
||||
var err error
|
||||
res.Changes, err = find(findEIDs)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
return &res, nil
|
||||
return &res
|
||||
}
|
||||
|
||||
Loading…
Reference in new issue