From e37cea2d6469368a860b1efa23973a0c6c510a50 Mon Sep 17 00:00:00 2001
From: hawklin2017 <32898629+hawklin2017@users.noreply.github.com>
Date: Wed, 6 May 2026 21:08:38 +0800
Subject: [PATCH] update area_code

---
 pkg/common/storage/database/mgo/user.go | 51 ++++++++++++-------------
 1 file changed, 24 insertions(+), 27 deletions(-)

diff --git a/pkg/common/storage/database/mgo/user.go b/pkg/common/storage/database/mgo/user.go
index 15f86fd07..d5f64a5ab 100644
--- a/pkg/common/storage/database/mgo/user.go
+++ b/pkg/common/storage/database/mgo/user.go
@@ -69,36 +69,33 @@ func (u *UserMgo) UpdateByMap(ctx context.Context, userID string, args map[strin
 	if err := mongoutil.UpdateOne(ctx, u.coll, filter, update, true); err != nil {
 		return err
 	}
-
-	return nil
-
 	// Keep user attributes in sync for consumers that read from the "attribute" collection.
 	// Only sync the allowed attribute fields.
-	//attributeSet := make(map[string]any)
-	//for _, key := range []string{
-	//	"nickname",
-	//	"first_name",
-	//	"last_name",
-	//	"full_name",
-	//	"remark",
-	//	"face_url",
-	//	"phone_number",
-	//	"area_code",
-	//} {
-	//	if v, ok := args[key]; ok {
-	//		attributeSet[key] = v
-	//	}
-	//}
+	attributeSet := make(map[string]any)
+	for _, key := range []string{
+		"nickname",
+		"first_name",
+		"last_name",
+		"full_name",
+		"remark",
+		"face_url",
+		"phone_number",
+		"area_code",
+	} {
+		if v, ok := args[key]; ok {
+			attributeSet[key] = v
+		}
+	}
 	//// user collection uses "phone"; attribute collection uses "phone_number".
-	//if v, ok := args["phone"]; ok {
-	//	attributeSet["phone_number"] = v
-	//}
-	//if len(attributeSet) == 0 {
-	//	return nil
-	//}
-
-	//attributeColl := u.coll.Database().Collection("attribute")
-	//return mongoutil.UpdateOne(ctx, attributeColl, filter, bson.M{"$set": attributeSet}, true)
+	if v, ok := args["phone"]; ok {
+		attributeSet["phone_number"] = v
+	}
+	if len(attributeSet) == 0 {
+		return nil
+	}
+
+	attributeColl := u.coll.Database().Collection("attribute")
+	return mongoutil.UpdateOne(ctx, attributeColl, filter, bson.M{"$set": attributeSet}, true)
 }
 
 func (u *UserMgo) Find(ctx context.Context, userIDs []string) (users []*model.User, err error) {