blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-11-05 03:42:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

new mongo

Browse Source
This commit is contained in:
withchao 2024-05-23 18:39:45 +08:00
parent 82c6b005e9
commit a2bfc907bb
1 changed files with 35 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
pkg/common/db/mgo/friend.go
View File

@ -62,7 +62,26 @@ func (f *FriendMgo) Create(ctx context.Context, friends []*relation.FriendModel)
return Success(func() error { return Success(func() error {
return mongoutil.InsertMany(ctx, f.coll, friends) return mongoutil.InsertMany(ctx, f.coll, friends)
}, func() error { }, func() error {
mp := make(map[string][]string)
for _, friend := range friends {
mp[friend.OwnerUserID] = append(mp[friend.OwnerUserID], friend.FriendUserID)
}
for ownerUserID, friendUserIDs := range mp {
if err := f.owner.WriteLog(ctx, ownerUserID, friendUserIDs, false); err != nil {
return err
}
}
return nil
}, func() error {
mp := make(map[string][]string)
for _, friend := range friends {
mp[friend.FriendUserID] = append(mp[friend.FriendUserID], friend.OwnerUserID)
}
for friendUserID, ownerUserIDs := range mp {
if err := f.friend.WriteLog(ctx, friendUserID, ownerUserIDs, false); err != nil {
return err
}
}
return nil return nil
}) })
} }
@ -78,7 +97,12 @@ func (f *FriendMgo) Delete(ctx context.Context, ownerUserID string, friendUserID
}, func() error { }, func() error {
return f.owner.WriteLog(ctx, ownerUserID, friendUserIDs, true) return f.owner.WriteLog(ctx, ownerUserID, friendUserIDs, true)
}, func() error { }, func() error {
for _, userID := range friendUserIDs {
if err := f.friend.WriteLog(ctx, userID, []string{ownerUserID}, true); err != nil {
return err
}
}
return nil
}) })
} }
@ -95,6 +119,8 @@ func (f *FriendMgo) UpdateByMap(ctx context.Context, ownerUserID string, friendU
return mongoutil.UpdateOne(ctx, f.coll, filter, bson.M{"$set": args}, true) return mongoutil.UpdateOne(ctx, f.coll, filter, bson.M{"$set": args}, true)
}, func() error { }, func() error {
return f.owner.WriteLog(ctx, ownerUserID, []string{friendUserID}, false) return f.owner.WriteLog(ctx, ownerUserID, []string{friendUserID}, false)
}, func() error {
return f.friend.WriteLog(ctx, friendUserID, []string{ownerUserID}, false)
}) })
} }
@ -178,6 +204,13 @@ func (f *FriendMgo) UpdateFriends(ctx context.Context, ownerUserID string, frien
return mongoutil.Ignore(mongoutil.UpdateMany(ctx, f.coll, filter, update)) return mongoutil.Ignore(mongoutil.UpdateMany(ctx, f.coll, filter, update))
}, func() error { }, func() error {
return f.owner.WriteLog(ctx, ownerUserID, friendUserIDs, false) return f.owner.WriteLog(ctx, ownerUserID, friendUserIDs, false)
}, func() error {
for _, userID := range friendUserIDs {
if err := f.friend.WriteLog(ctx, userID, []string{ownerUserID}, true); err != nil {
return err
}
}
return nil
}) })
} }