fix: 修复顺运宝采集任务孤儿状态 (#165)

2026-08-11 17:00:33 +08:00
parent c8b1fd686a
commit 9e328984e9
17 changed files with 508 additions and 67 deletions
@@ -585,6 +585,71 @@ func TestMySQLMigrate_V11回填顺运宝店铺且可重放(t *testing.T) {
 	}
 }

+func TestMySQLMigrate_V11升级V12并修复孤儿采集中状态(t *testing.T) {
+	db := openMySQLMigrationTestDB(t)
+	defer db.Close()
+	cleanMySQLTestSchema(t, db)
+	defer cleanMySQLTestSchema(t, db)
+	if err := MigrateMySQL(db); err != nil {
+		t.Fatal(err)
+	}
+	// 模拟已在 v11 的生产库：去掉 v12 版本及其附加表，保留全部 v1-v11 结构。
+	mustExec(t, db, `DELETE FROM schema_migrations WHERE version=12`)
+	mustExec(t, db, `DROP TABLE task_syb_sources`)
+	now := model.NowISO()
+	mustExec(t, db, `INSERT INTO pdd_products(goods_id,url,collect_status,created_at,updated_at) VALUES
+		('ORPHAN','https://mobile.yangkeduo.com/goods.html?goods_id=1','collecting',?,?),
+		('ACTIVE','https://mobile.yangkeduo.com/goods.html?goods_id=2','collecting',?,?)`, now, now, now, now)
+	mustExec(t, db, `INSERT INTO tasks(task_id,task_type,status,pdd_goods_url,pdd_goods_id,created_at,updated_at)
+		VALUES('COL-ACTIVE','collect','pending','https://mobile.yangkeduo.com/goods.html?goods_id=2','ACTIVE',?,?)`, now, now)
+
+	if err := MigrateMySQL(db); err != nil {
+		t.Fatalf("v11 升级 v12 失败: %v", err)
+	}
+	if err := MigrateMySQL(db); err != nil {
+		t.Fatalf("v12 重放失败: %v", err)
+	}
+	var orphanStatus, activeStatus string
+	if err := db.QueryRow(`SELECT collect_status FROM pdd_products WHERE goods_id='ORPHAN'`).Scan(&orphanStatus); err != nil {
+		t.Fatal(err)
+	}
+	if err := db.QueryRow(`SELECT collect_status FROM pdd_products WHERE goods_id='ACTIVE'`).Scan(&activeStatus); err != nil {
+		t.Fatal(err)
+	}
+	if orphanStatus != "pending" || activeStatus != "collecting" {
+		t.Fatalf("v12 状态修复错误：orphan=%s active=%s", orphanStatus, activeStatus)
+	}
+	var versionCount int
+	if err := db.QueryRow(`SELECT COUNT(*) FROM schema_migrations WHERE version=12`).Scan(&versionCount); err != nil || versionCount != 1 {
+		t.Fatalf("v12 版本记录错误: count=%d err=%v", versionCount, err)
+	}
+}
+
+func TestMySQLMigrate_V12形状错误不记录版本(t *testing.T) {
+	db := openMySQLMigrationTestDB(t)
+	defer db.Close()
+	cleanMySQLTestSchema(t, db)
+	defer cleanMySQLTestSchema(t, db)
+	if err := MigrateMySQL(db); err != nil {
+		t.Fatal(err)
+	}
+	mustExec(t, db, `DELETE FROM schema_migrations WHERE version=12`)
+	mustExec(t, db, `DROP TABLE task_syb_sources`)
+	mustExec(t, db, `CREATE TABLE task_syb_sources (
+		task_id VARCHAR(191) COLLATE utf8mb4_bin NOT NULL,
+		syb_id VARCHAR(191) COLLATE utf8mb4_bin NOT NULL,
+		created_at VARCHAR(35) NOT NULL,
+		PRIMARY KEY(syb_id,task_id)
+	) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci`)
+	if err := MigrateMySQL(db); err == nil {
+		t.Fatal("错误主键和缺失外键必须阻止 v12")
+	}
+	var versionCount int
+	if err := db.QueryRow(`SELECT COUNT(*) FROM schema_migrations WHERE version=12`).Scan(&versionCount); err != nil || versionCount != 0 {
+		t.Fatalf("v12 自检失败时不得记录版本：count=%d err=%v", versionCount, err)
+	}
+}
+
 func openMySQLMigrationTestDB(t *testing.T) *sql.DB {
 	t.Helper()
 	if os.Getenv("CMAUTOBUY_MYSQL_TEST") != "1" {