From a2bfc907bb9e68c53b63c9183ff0887a635b0fd7 Mon Sep 17 00:00:00 2001
From: withchao <993506633@qq.com>
Date: Thu, 23 May 2024 18:39:45 +0800
Subject: [PATCH] new mongo

---
 pkg/common/db/mgo/friend.go | 37 +++++++++++++++++++++++++++++++++++--
 1 file changed, 35 insertions(+), 2 deletions(-)

diff --git a/pkg/common/db/mgo/friend.go b/pkg/common/db/mgo/friend.go
index 6e7a6db98..b82d1c3ab 100644
--- a/pkg/common/db/mgo/friend.go
+++ b/pkg/common/db/mgo/friend.go
@@ -62,7 +62,26 @@ func (f *FriendMgo) Create(ctx context.Context, friends []*relation.FriendModel)
 	return Success(func() error {
 		return mongoutil.InsertMany(ctx, f.coll, friends)
 	}, 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
 	})
 }
@@ -78,7 +97,12 @@ func (f *FriendMgo) Delete(ctx context.Context, ownerUserID string, friendUserID
 	}, func() error {
 		return f.owner.WriteLog(ctx, ownerUserID, friendUserIDs, true)
 	}, 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)
 	}, func() error {
 		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))
 	}, func() error {
 		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
 	})
 }