feat: 增强用户数据库同步可靠性

backend/app/database_user.py

@@ -11,6 +11,7 @@ import secrets
 import hashlib
 import hmac
 import logging
+import time
 from dotenv import load_dotenv
 
 # 加载 .env 环境变量
@@ -170,20 +171,30 @@ def sync_user_db_from_hf() -> None:
         logger.info("HF_TOKEN or DATASET_REPO_ID not set, skip user db download")
         return
 
-    try:
-        downloaded = hf_hub_download(
-            repo_id=DATASET_REPO_ID,
-            filename=USER_DB_FILENAME,
-            repo_type="dataset",
-            token=HF_TOKEN,
-        )
-        src = Path(downloaded)
-        dst = Path(USER_DB_PATH)
-        if src.exists() and src.resolve() != dst.resolve():
-            dst.write_bytes(src.read_bytes())
-        logger.info(f"User db synced from HF: {USER_DB_FILENAME}")
-    except Exception as e:
-        logger.warning(f"Failed to sync user db from HF: {e}")
+    max_retries = 5
+    retry_delay = 2  # seconds
+    for attempt in range(max_retries):
+        try:
+            downloaded = hf_hub_download(
+                repo_id=DATASET_REPO_ID,
+                filename=USER_DB_FILENAME,
+                repo_type="dataset",
+                token=HF_TOKEN,
+            )
+            src = Path(downloaded)
+            dst = Path(USER_DB_PATH)
+            if src.exists() and src.resolve() != dst.resolve():
+                dst.write_bytes(src.read_bytes())
+            logger.info(f"User db synced from HF: {USER_DB_FILENAME}")
+            return  # Success, exit the loop
+        except Exception as e:
+            logger.warning(f"Attempt {attempt + 1} failed to sync user db from HF: {e}")
+            if attempt < max_retries - 1:
+                logger.info(f"Retrying in {retry_delay} seconds...")
+                time.sleep(retry_delay)
+            else:
+                logger.error(f"All {max_retries} attempts failed to sync user db from HF. Application will not start properly.")
+                raise  # Re-raise the exception to prevent application startup
 
 
 def sync_qr_codes_from_hf() -> None:
@@ -221,17 +232,32 @@ def upload_user_db_to_hf() -> None:
         logger.warning("User db file not found, skip upload")
         return
 
-    try:
-        upload_file(
-            path_or_fileobj=str(db_path),
-            path_in_repo=USER_DB_FILENAME,
-            repo_id=DATASET_REPO_ID,
-            repo_type="dataset",
-            token=HF_TOKEN,
-        )
-        logger.info(f"User db uploaded to HF: {USER_DB_FILENAME}")
-    except Exception as e:
-        logger.error(f"Failed to upload user db to HF: {e}")
+    # Check if the database file is too small (likely empty), prevent overwriting
+    file_size = db_path.stat().st_size
+    if file_size < 1024:  # Less than 1KB, likely empty
+        logger.warning(f"User db file is too small ({file_size} bytes), skip upload to prevent overwriting remote data")
+        return
+
+    max_retries = 3
+    retry_delay = 2  # seconds
+    for attempt in range(max_retries):
+        try:
+            upload_file(
+                path_or_fileobj=str(db_path),
+                path_in_repo=USER_DB_FILENAME,
+                repo_id=DATASET_REPO_ID,
+                repo_type="dataset",
+                token=HF_TOKEN,
+            )
+            logger.info(f"User db uploaded to HF: {USER_DB_FILENAME}")
+            return  # Success, exit the loop
+        except Exception as e:
+            logger.error(f"Attempt {attempt + 1} failed to upload user db to HF: {e}")
+            if attempt < max_retries - 1:
+                logger.info(f"Retrying in {retry_delay} seconds...")
+                time.sleep(retry_delay)
+            else:
+                logger.error(f"All {max_retries} attempts failed to upload user db to HF")
 
 
 def sync_user_db_after_update(func: Callable[..., Any]) -> Callable[..., Any]: