From 0d7ba874adaf34c9bfb95289e656b2e28edc013c Mon Sep 17 00:00:00 2001
From: milky0217 <m943790568@163.com>
Date: Sat, 30 May 2026 09:21:11 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E8=BF=81=E7=A7=BB=E8=84=9A=E6=9C=AC?=
 =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E8=87=AA=E5=A2=9E=E5=BA=8F=E5=88=97=E9=87=8D?=
 =?UTF-8?q?=E7=BD=AE+=E6=94=B9=E8=BF=9B=E5=9B=9E=E6=BB=9A=E8=AF=B4?=
 =?UTF-8?q?=E6=98=8E?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 scripts/migrate-prod-to-dev.sh | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/scripts/migrate-prod-to-dev.sh b/scripts/migrate-prod-to-dev.sh
index 09871e3..4b57cc1 100755
--- a/scripts/migrate-prod-to-dev.sh
+++ b/scripts/migrate-prod-to-dev.sh
@@ -89,6 +89,15 @@ log "导入数据到 ${DST_DB}..."
 docker exec -i "$DB_CONTAINER" psql -U "$DB_USER" -d "$DST_DB" < "$TMP_SQL" 2>> "$LOG_FILE"
 log "  导入完成 ✅"
 
+# ---------- 步骤 5: 重置自增序列 ----------
+log "重置自增序列..."
+docker exec "$DB_CONTAINER" psql -U "$DB_USER" -d "$DST_DB" -tAc "
+SELECT SETVAL('users_id_seq', COALESCE((SELECT MAX(id) FROM users), 1));
+SELECT SETVAL('weather_data_id_seq', COALESCE((SELECT MAX(id) FROM weather_data), 1));
+SELECT SETVAL('payment_orders_id_seq', COALESCE((SELECT MAX(id) FROM payment_orders), 1));
+" 2>> "$LOG_FILE"
+log "  序列重置完成 ✅"
+
 # ---------- 清理 ----------
 rm -f "$TMP_SQL"
 
@@ -110,4 +119,5 @@ echo ""
 echo "日志: ${LOG_FILE}"
 echo "备份: ${BACKUP_DIR}/${DST_DB}_pre_migration_*.dump"
 echo ""
-echo "⚠️  如需回滚：gunzip -c ${BACKUP_DIR}/*.dump.gz | docker exec -i ... pg_restore"
+echo "⚠️  如需回滚：
+  docker exec -i ${DB_CONTAINER} pg_restore -U ${DB_USER} -d ${DST_DB} --clean \n    < ${BACKUP_DIR}/${DST_DB}_pre_migration_*.dump"