From 8de12dbcfcf67b758887ccc1c90a87bbe5c93dd0 Mon Sep 17 00:00:00 2001
From: kjqwer <2990346238@qq.com>
Date: Sat, 11 Oct 2025 16:10:12 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E4=BB=85=E5=88=87=E6=8D=A2?=
 =?UTF-8?q?=E6=B3=A8=E5=86=8C=E8=A1=A8=E5=8A=9F=E8=83=BD=EF=BC=8C=E9=98=B2?=
 =?UTF-8?q?=E6=AD=A2=E8=A6=86=E7=9B=96=E5=88=A0=E9=99=A4=E6=95=B0=E6=8D=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/routes/download.js                  |  8 ++
 backend/services/download-registry.js       | 27 ++++++
 ui/src/components/common/RegistryWidget.vue | 98 +++++++++++++++++----
 3 files changed, 114 insertions(+), 19 deletions(-)

diff --git a/backend/routes/download.js b/backend/routes/download.js
index 9597b1b..b9df722 100644
--- a/backend/routes/download.js
+++ b/backend/routes/download.js
@@ -1262,6 +1262,14 @@ router.put('/registry/config', async (req, res) => {
     // 更新配置
     const updatedConfig = await downloadService.cacheConfigManager.updateConfig(updateData);
     
+    // 如果更新了存储模式，需要重新加载存储模式配置
+    if (storageMode) {
+      const reloadResult = await downloadService.downloadRegistry.reloadStorageMode();
+      if (!reloadResult.success) {
+        logger.warn('重新加载存储模式失败:', reloadResult.error);
+      }
+    }
+    
     res.json({
       success: true,
       data: {
diff --git a/backend/services/download-registry.js b/backend/services/download-registry.js
index 63d1e59..63504f1 100644
--- a/backend/services/download-registry.js
+++ b/backend/services/download-registry.js
@@ -73,6 +73,33 @@ class DownloadRegistry {
     }
   }
 
+  /**
+   * 重新加载存储模式配置（用于切换存储模式时）
+   */
+  async reloadStorageMode() {
+    try {
+      await this.loadStorageMode();
+      
+      // 根据新的存储模式重新初始化相应的存储
+      if (this.storageMode === 'database') {
+        if (!this.registryDatabase && this.databaseManager) {
+          await this.initDatabaseStorage();
+        }
+      } else {
+        // JSON模式下确保注册表已加载
+        if (!this.loaded) {
+          await this.initJsonStorage();
+        }
+      }
+      
+      logger.info(`存储模式已切换到: ${this.storageMode}`);
+      return { success: true, storageMode: this.storageMode };
+    } catch (error) {
+      logger.error('重新加载存储模式失败:', error);
+      return { success: false, error: error.message };
+    }
+  }
+
   /**
    * 初始化数据库存储
    */
diff --git a/ui/src/components/common/RegistryWidget.vue b/ui/src/components/common/RegistryWidget.vue
index b8f8a02..8897904 100644
--- a/ui/src/components/common/RegistryWidget.vue
+++ b/ui/src/components/common/RegistryWidget.vue
@@ -110,6 +110,40 @@
                       当前模式: {{ storageMode === 'json' ? 'JSON文件存储' : 'MySQL数据库存储' }}
                     </span>
                   </div>
+
+                  <!-- 迁移选项 -->
+                  <div v-if="hasStorageModeChanges" class="migration-options">
+                    <div class="migration-option-title">切换方式选择：</div>
+                    <div class="migration-option-group">
+                      <div class="migration-option" :class="{ active: migrationMode === 'switch-only' }">
+                        <label>
+                          <input type="radio" v-model="migrationMode" value="switch-only" />
+                          <div class="option-content">
+                            <div class="option-header">
+                              <SvgIcon name="switch" class="option-icon" />
+                              <span class="option-title">仅切换读取方式</span>
+                              <span class="option-badge safe">安全</span>
+                            </div>
+                            <small class="option-description">只改变数据读取方式，不迁移数据，保持两边数据独立</small>
+                          </div>
+                        </label>
+                      </div>
+
+                      <div class="migration-option" :class="{ active: migrationMode === 'migrate-data' }">
+                        <label>
+                          <input type="radio" v-model="migrationMode" value="migrate-data" />
+                          <div class="option-content">
+                            <div class="option-header">
+                              <SvgIcon name="transfer" class="option-icon" />
+                              <span class="option-title">迁移数据并切换</span>
+                              <span class="option-badge warning">覆盖</span>
+                            </div>
+                            <small class="option-description">将当前数据迁移到目标存储并切换，会覆盖目标存储的现有数据</small>
+                          </div>
+                        </label>
+                      </div>
+                    </div>
+                  </div>
                   
                   <div class="action-buttons">
                     <button 
@@ -118,7 +152,7 @@
                       :disabled="migrationLoading || !hasStorageModeChanges || (selectedStorageMode === 'database' && !databaseConnected)"
                     >
                       <SvgIcon name="save" class="btn-icon" />
-                      {{ migrationLoading ? '应用中...' : '应用配置' }}
+                      {{ migrationLoading ? '应用中...' : getApplyButtonText() }}
                     </button>
                     
                     <button 
@@ -322,12 +356,22 @@ const databaseConnected = ref(false);
 const storageMode = ref<'json' | 'database'>('json');
 const selectedStorageMode = ref<'json' | 'database'>('json'); // 用户选择的存储模式
 const migrationLoading = ref(false);
+const migrationMode = ref<'switch-only' | 'migrate-data'>('switch-only'); // 迁移模式选择
 
 // 计算属性：检查存储模式是否有变更
 const hasStorageModeChanges = computed(() => {
   return selectedStorageMode.value !== storageMode.value;
 });
 
+// 计算应用按钮文本
+const getApplyButtonText = () => {
+  if (migrationMode.value === 'switch-only') {
+    return '仅切换读取方式';
+  } else {
+    return '迁移数据并切换';
+  }
+};
+
 // 从store中获取响应式数据
 const { stats, loading, error, config } = storeToRefs(registryStore);
 
@@ -652,9 +696,17 @@ const applyStorageModeConfig = async () => {
     return;
   }
 
-  const confirmMessage = selectedStorageMode.value === 'database' 
-    ? '确定要切换到数据库存储模式吗？这将使用MySQL数据库存储注册表数据。'
-    : '确定要切换到JSON文件存储模式吗？这将使用本地JSON文件存储注册表数据。';
+  // 根据迁移模式生成不同的确认消息
+  let confirmMessage = '';
+  if (migrationMode.value === 'switch-only') {
+    confirmMessage = selectedStorageMode.value === 'database' 
+      ? '确定要切换到数据库存储模式吗？这将仅改变读取方式，不会迁移现有数据。'
+      : '确定要切换到JSON文件存储模式吗？这将仅改变读取方式，不会迁移现有数据。';
+  } else {
+    confirmMessage = selectedStorageMode.value === 'database' 
+      ? '确定要迁移数据并切换到数据库存储模式吗？这将把JSON数据迁移到数据库并覆盖数据库中的现有数据。'
+      : '确定要迁移数据并切换到JSON文件存储模式吗？这将把数据库数据迁移到JSON文件并覆盖JSON文件中的现有数据。';
+  }
 
   if (!confirm(confirmMessage)) {
     return;
@@ -663,24 +715,32 @@ const applyStorageModeConfig = async () => {
   try {
     migrationLoading.value = true;
     
-    if (selectedStorageMode.value === 'database') {
-      // 从JSON迁移到数据库
-      const result = await databaseService.migrateData('json-to-db');
-      if (result.success) {
-        storageMode.value = 'database';
-        showSuccess(`成功迁移到数据库存储，处理了 ${result.data?.recordsProcessed || 0} 条记录`);
+    if (migrationMode.value === 'migrate-data') {
+      // 执行数据迁移
+      if (selectedStorageMode.value === 'database') {
+        // 从JSON迁移到数据库
+        const result = await databaseService.migrateData('json-to-db');
+        if (result.success) {
+          storageMode.value = 'database';
+          showSuccess(`成功迁移到数据库存储，处理了 ${result.data?.recordsProcessed || 0} 条记录`);
+        } else {
+          throw new Error(result.error || '迁移到数据库失败');
+        }
       } else {
-        throw new Error(result.error || '迁移到数据库失败');
+        // 从数据库迁移到JSON
+        const result = await databaseService.migrateData('db-to-json');
+        if (result.success) {
+          storageMode.value = 'json';
+          showSuccess(`成功迁移到JSON存储，处理了 ${result.data?.recordsProcessed || 0} 条记录`);
+        } else {
+          throw new Error(result.error || '迁移到JSON失败');
+        }
       }
     } else {
-      // 从数据库迁移到JSON
-      const result = await databaseService.migrateData('db-to-json');
-      if (result.success) {
-        storageMode.value = 'json';
-        showSuccess(`成功迁移到JSON存储，处理了 ${result.data?.recordsProcessed || 0} 条记录`);
-      } else {
-        throw new Error(result.error || '迁移到JSON失败');
-      }
+      // 仅切换存储模式，不迁移数据
+      storageMode.value = selectedStorageMode.value;
+      const modeText = selectedStorageMode.value === 'database' ? '数据库存储' : 'JSON文件存储';
+      showSuccess(`已切换到${modeText}模式，数据读取方式已更改`);
     }
     
     // 保存存储模式配置到后端