From 4367f7fb9266ed2fe179a54e753affe6b289b4be Mon Sep 17 00:00:00 2001
From: dsh <dsh@lmve.net>
Date: Wed, 19 Aug 2026 11:27:00 -0400
Subject: [PATCH] =?UTF-8?q?fix(migrate):=20mailbox=5Fstates=20=E6=94=B9?=
 =?UTF-8?q?=E7=94=A8=E5=8E=9F=E7=94=9F=20SQL=20=E8=BF=81=E7=A7=BB=EF=BC=8C?=
 =?UTF-8?q?=E4=B8=8D=E4=BE=9D=E8=B5=96=20MailboxState=20=E6=A8=A1=E5=9E=8B?=
 =?UTF-8?q?=EF=BC=88=E5=85=BC=E5=AE=B9=E6=9C=AA=E5=90=88=E5=B9=B6=20UIDVAL?=
 =?UTF-8?q?IDITY=20=E7=89=B9=E6=80=A7=E7=9A=84=E4=B8=8A=E6=B8=B8=EF=BC=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cmd/migrate/main.go | 56 ++++++++++++++++++++++++++++++++++-----------
 1 file changed, 43 insertions(+), 13 deletions(-)

diff --git a/cmd/migrate/main.go b/cmd/migrate/main.go
index 71ce115..d2419d2 100644
--- a/cmd/migrate/main.go
+++ b/cmd/migrate/main.go
@@ -43,6 +43,8 @@ func main() {
 	// 关闭 GORM 自动时间戳（保留原始 CreatedAt/UpdatedAt）
 	mw := mdb.Session(&gorm.Session{SkipHooks: true})
 
+	stateWant := int64(-1) // mailbox_states 期望行数；-1 = 源库无此表不校验
+
 	// 按外键依赖顺序复制：domains → users → messages → attachments → 其余
 	// 所有时间统一 UTC（MySQL DATETIME 无时区）。
 	utc := func(t time.Time) time.Time {
@@ -177,19 +179,45 @@ func main() {
 	}
 	log.Printf("protocol_logs: %d", len(logs))
 
-	// ---- mailbox_states ----
-	var states []db.MailboxState
-	if err := sdb.Order("user_id ASC, folder ASC").Find(&states).Error; err != nil {
-		log.Fatalf("读 mailbox_states: %v", err)
+	// ---- mailbox_states（原生 SQL：该表随 UIDVALIDITY 特性存在，旧版本源库可能没有）----
+	var stateCount int64
+	sdb.Raw("SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name='mailbox_states'").Scan(&stateCount)
+	if stateCount > 0 {
+		srows, err := sdb.Raw("SELECT user_id, folder, uid_validity, created_at, updated_at FROM mailbox_states ORDER BY user_id, folder").Rows()
+		if err != nil {
+			log.Fatalf("读 mailbox_states: %v", err)
+		}
+		defer srows.Close()
+		states := 0
+		for srows.Next() {
+			var (
+				userID   uint
+				folder   string
+				validity uint32
+				created  *time.Time
+				updated  *time.Time
+			)
+			if err := srows.Scan(&userID, &folder, &validity, &created, &updated); err != nil {
+				log.Fatalf("扫 mailbox_states: %v", err)
+			}
+			norm := func(t *time.Time) *time.Time {
+				if t == nil || t.IsZero() {
+					return nil
+				}
+				u := t.UTC()
+				return &u
+			}
+			if err := mdb.Exec("INSERT INTO mailbox_states (user_id, folder, uid_validity, created_at, updated_at) VALUES (?,?,?,?,?)",
+				userID, folder, validity, norm(created), norm(updated)).Error; err != nil {
+				log.Fatalf("写 mailbox_states: %v", err)
+			}
+			states++
+		}
+		log.Printf("mailbox_states: %d", states)
+		stateWant = int64(states)
+	} else {
+		log.Println("mailbox_states: 源库无此表，跳过")
 	}
-	for i := range states {
-		states[i].CreatedAt = states[i].CreatedAt.UTC()
-		states[i].UpdatedAt = states[i].UpdatedAt.UTC()
-	}
-	if err := mw.Create(&states).Error; err != nil {
-		log.Fatalf("写 mailbox_states: %v", err)
-	}
-	log.Printf("mailbox_states: %d", len(states))
 
 	// ---- 校验 ----
 	check := func(table string, want int64) {
@@ -209,7 +237,9 @@ func main() {
 	check("outbound_messages", int64(len(outs)))
 	check("ban_entries", int64(bans))
 	check("protocol_logs", int64(len(logs)))
-	check("mailbox_states", int64(len(states)))
+	if stateWant >= 0 {
+		check("mailbox_states", stateWant)
+	}
 
 	log.Println("迁移完成 ✅")
 }