Merge pull request #11 from xxcheng123/develop

feat(proxy): 多线程支持自定义线程和块大小

Author: xxcheng123

configs/init.go

@@ -82,6 +82,7 @@ func Init() {
 		new(models.VirtualFile),
 		new(models.UserGroup),
 		new(models.Group2File),
+		new(models.SettingDict),
 	); err != nil {
 		panic(err)
 	}
@@ -110,6 +111,20 @@ func Init() {
 		panic(err)
 	}
 
+	var dicts = make([]*models.SettingDict, 0)
+	if err = db.Model(new(models.SettingDict)).Find(&dicts).Error; err != nil {
+		panic(err)
+	}
+
+	for _, dict := range dicts {
+		switch dict.Key {
+		case models.SettingDictKeyMultipleStreamThreadCount:
+			shared.MultipleStreamThreadCount = dict.Value.Int()
+		case models.SettingDictKeyMultipleStreamChunkSize:
+			shared.MultipleStreamChunkSize = dict.Value.Int64()
+		}
+	}
+
 	shared.Setting = setting
 
 	logger.Info("binary build info",

fe/src/api/setting.ts

@@ -8,14 +8,16 @@ export interface Setting {
   enableAuth: boolean
   localProxy: boolean
   multipleStream: boolean
-  runTimes: number  // 后端实际返回的字段名
+  runTimes: number  // 已经运行的时间
   createdAt: string
   updatedAt: string
   baseURL: string
   enableTopFileAutoRefresh: boolean
   initialized: boolean // 系统是否初始化完成
   jobThreadCount: number // 任务线程数
   autoRefreshMinutes: number // 自动刷新间隔（分钟）
+  multipleStreamThreadCount: number // 多线程流线程数
+  multipleStreamChunkSize: number // 多线程流块大小
 }
 
 export interface InitSystemRequest {
@@ -58,6 +60,21 @@ export interface ModifyAutoRefreshMinutesRequest {
   autoRefreshMinutes: number
 }
 
+// 新增：修改多线程流线程数请求
+export interface ModifyMultipleStreamThreadCountRequest {
+  multipleStreamThreadCount: number // 1-64之间
+}
+
+// 新增：修改多线程流块大小请求
+export interface ModifyMultipleStreamChunkSizeRequest {
+  multipleStreamChunkSize: number // 512KB-32MB之间
+}
+
+// 新增：修改操作的通用响应
+export interface ModifyResponse {
+  rowsAffected: number
+}
+
 // 设置API
 export const settingApi = {
   // 获取设置
@@ -110,6 +127,16 @@ export const settingApi = {
     return api.post('/setting/modify_auto_refresh_minutes', data)
   },
 
+  // 新增：修改多线程流线程数
+  modifyMultipleStreamThreadCount: (data: ModifyMultipleStreamThreadCountRequest): Promise<ModifyResponse> => {
+    return api.post('/setting/modify_multiple_stream_thread_count', data)
+  },
+
+  // 新增：修改多线程流块大小
+  modifyMultipleStreamChunkSize: (data: ModifyMultipleStreamChunkSizeRequest): Promise<ModifyResponse> => {
+    return api.post('/setting/modify_multiple_stream_chunk_size', data)
+  },
+
   // 初始化系统
   initSystem: (data: InitSystemRequest): Promise<void> => {
     return api.post('/setting/init_system', data)

fe/src/stores/setting.ts

@@ -137,6 +137,36 @@ export const useSettingStore = defineStore('setting', () => {
     }
   }
 
+  // 新增：修改多线程流线程数
+  const modifyMultipleStreamThreadCount = async (threadCount: number) => {
+    try {
+      await settingApi.modifyMultipleStreamThreadCount({
+        multipleStreamThreadCount: threadCount
+      })
+      if (setting.value) {
+        setting.value.multipleStreamThreadCount = threadCount
+      }
+    } catch (error) {
+      console.error('修改多线程流线程数失败:', error)
+      throw error
+    }
+  }
+
+  // 新增：修改多线程流块大小
+  const modifyMultipleStreamChunkSize = async (multipleStreamChunkSize: number) => {
+    try {
+      await settingApi.modifyMultipleStreamChunkSize({
+        multipleStreamChunkSize: multipleStreamChunkSize
+      })
+      if (setting.value) {
+        setting.value.multipleStreamChunkSize = multipleStreamChunkSize
+      }
+    } catch (error) {
+      console.error('修改多线程流块大小失败:', error)
+      throw error
+    }
+  }
+
   // 初始化系统
   const initSystem = async (data: InitSystemRequest) => {
     try {
@@ -162,6 +192,8 @@ export const useSettingStore = defineStore('setting', () => {
     toggleEnableTopFileAutoRefresh,
     modifyJobThreadCount,
     modifyAutoRefreshMinutes,
+    modifyMultipleStreamThreadCount,
+    modifyMultipleStreamChunkSize,
     initSystem
   }
 })