From 4f609481ccb31249ebf964d255bb2cabed048818 Mon Sep 17 00:00:00 2001
From: Leufolium <Caeruscaemon@outlook.com>
Date: Sat, 16 Mar 2024 23:54:18 +0800
Subject: [PATCH] by Robin at 20240316

---
 cmd/main.go    |  7 +++++--
 mongo/mongo.go | 17 +++++++++++++++++
 2 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/cmd/main.go b/cmd/main.go
index aca258a..1224650 100644
--- a/cmd/main.go
+++ b/cmd/main.go
@@ -18,8 +18,11 @@ func main() {
 	}
 
 	ctx := context.Background()
-	account, _ := client.GetAccountListByMid(ctx, 2)
-	if account != nil {
+
+	list, _ := client.GetAccountList(ctx)
+	fmt.Printf("共%v个account待更新", len(list))
+
+	for _, account := range list {
 		account.UserIdString = goproto.String(fmt.Sprint(*account.UserId))
 		client.UpdateAccount(ctx, account)
 	}
diff --git a/mongo/mongo.go b/mongo/mongo.go
index b34da55..675c5da 100644
--- a/mongo/mongo.go
+++ b/mongo/mongo.go
@@ -152,6 +152,23 @@ func (m *Mongo) GetAccountListByMid(ctx context.Context, mid int64) (*dbstruct.A
 	return account, err
 }
 
+func (m *Mongo) GetAccountList(ctx context.Context) ([]*dbstruct.Account, error) {
+	col := m.getColAccount()
+	list := make([]*dbstruct.Account, 0)
+
+	query := qmgo.M{}
+
+	err := col.Find(ctx, query).All(&list)
+	if err == qmgo.ErrNoSuchDocuments {
+		err = nil
+		return nil, err
+	}
+	if err != nil {
+		return nil, err
+	}
+	return list, err
+}
+
 func (m *Mongo) UpdateAccount(ctx context.Context, account *dbstruct.Account) error {
 	col := m.getColAccount()
 	set := qmgo.M{