Upload unified-server.js

unified-server.js

@@ -723,6 +723,7 @@ class ProxyServerSystem extends EventEmitter {
       maxRetries: 3, retryDelay: 2000, browserExecutablePath: null,
       apiKeys: [],
       immediateSwitchStatusCodes: [],
+      initialAuthIndex: null, // 新增：为 initialAuthIndex 提供默认值
     };
 
     const configPath = path.join(__dirname, 'config.json');
@@ -746,6 +747,14 @@ class ProxyServerSystem extends EventEmitter {
     if (process.env.API_KEYS) {
       config.apiKeys = process.env.API_KEYS.split(',');
     }
+    // 新增：处理环境变量，它会覆盖 config.json 中的设置
+    if (process.env.INITIAL_AUTH_INDEX) {
+        const envIndex = parseInt(process.env.INITIAL_AUTH_INDEX, 10);
+        if (!isNaN(envIndex) && envIndex > 0) {
+            config.initialAuthIndex = envIndex;
+        }
+    }
+
 
     // NEW: 统一处理 immediateSwitchStatusCodes，环境变量优先于 config.json
     let rawCodes = process.env.IMMEDIATE_SWITCH_STATUS_CODES;
@@ -779,6 +788,10 @@ class ProxyServerSystem extends EventEmitter {
     this.logger.info(`  HTTP 服务端口: ${this.config.httpPort}`);
     this.logger.info(`  监听地址: ${this.config.host}`);
     this.logger.info(`  流式模式: ${this.config.streamingMode}`);
+    // 新增：在日志中显示初始索引的配置
+    if (this.config.initialAuthIndex) {
+        this.logger.info(`  指定初始认证索引: ${this.config.initialAuthIndex}`);
+    }
     // MODIFIED: 日志输出已汉化
     this.logger.info(`  失败计数切换: ${this.config.failureThreshold > 0 ? `连续 ${this.config.failureThreshold} 次失败后切换` : '已禁用'}`);
     this.logger.info(`  立即切换状态码: ${this.config.immediateSwitchStatusCodes.length > 0 ? this.config.immediateSwitchStatusCodes.join(', ') : '已禁用'}`);
@@ -796,14 +809,15 @@ class ProxyServerSystem extends EventEmitter {
     try {
       // 决定启动时使用的认证索引
       let startupIndex = this.authSource.getFirstAvailableIndex();
-      const suggestedIndex = parseInt(process.env.INITIAL_AUTH_INDEX, 10);
+      // 修改：从 this.config 读取，而不是直接从 process.env
+      const suggestedIndex = this.config.initialAuthIndex;
 
       if (suggestedIndex) {
         if (this.authSource.getAvailableIndices().includes(suggestedIndex)) {
-            this.logger.info(`[System] 使用环境变量指定的有效启动索引: ${suggestedIndex}`);
+            this.logger.info(`[System] 使用配置中指定的有效启动索引: ${suggestedIndex}`);
             startupIndex = suggestedIndex;
         } else {
-            this.logger.warn(`[System] 环境变量指定的启动索引 ${suggestedIndex} 无效或不存在，将使用第一个可用索引: ${startupIndex}`);
+            this.logger.warn(`[System] 配置中指定的启动索引 ${suggestedIndex} 无效或不存在，将使用第一个可用索引: ${startupIndex}`);
         }
       } else {
         this.logger.info(`[System] 未指定启动索引，将自动使用第一个可用索引: ${startupIndex}`);
@@ -814,7 +828,8 @@ class ProxyServerSystem extends EventEmitter {
       await this._startWebSocketServer();
       this.logger.info(`[System] 代理服务器系统启动完成。`);
       this.emit('started');
-    } catch (error) {
+    } catch (error)
+    {
       this.logger.error(`[System] 启动失败: ${error.message}`);
       this.emit('error', error);
       throw error;