Update index.js

index.js

@@ -172,8 +172,9 @@ class ResponseHandler {
 
 // WebSocket工具类
 class WebSocketUtils {
-    static isProcessing = false; // 添加静态属性用于并行控制
+    static activeConnections = new Set(); // 跟踪活跃连接
     static TIMEOUT = 5 * 60 * 1000; // 5分钟超时时间
+    static MAX_CONNECTIONS = 10; // 最大并发连接数
 
     // 生成WebSocket密钥
     static generateWebSocketKey() {
@@ -182,11 +183,11 @@ class WebSocketUtils {
 
     // 创建WebSocket客户端
     static async createWebSocketClient(requestPayload) {
-        if (this.isProcessing) {
-            throw new Error('上游服务繁忙，只允许单次请求，请稍后重试！');
+        // 检查当前连接数是否达到上限
+        if (this.activeConnections.size >= this.MAX_CONNECTIONS) {
+            throw new Error(`当前连接数已达到上限 (${this.MAX_CONNECTIONS})，请稍后重试喵！`);
         }
 
-        this.isProcessing = true;
         let timeoutId;
         let ws;
 
@@ -200,11 +201,17 @@ class WebSocketUtils {
                     }
                 });
 
+                // 添加到活跃连接集合
+                this.activeConnections.add(ws);
+                console.log(`当前活跃连接数: ${this.activeConnections.size}/${this.MAX_CONNECTIONS}`);
+
                 // 设置超时处理
                 timeoutId = setTimeout(() => {
                     if (ws.readyState === WebSocket.OPEN) {
                         ws.close();
                     }
+                    this.activeConnections.delete(ws);
+                    console.log(`连接超时，当前活跃连接数: ${this.activeConnections.size}/${this.MAX_CONNECTIONS}`);
                     reject(new Error('WebSocket连接超时（5分钟）'));
                 }, this.TIMEOUT);
 
@@ -227,6 +234,7 @@ class WebSocketUtils {
                 ws.on('message', async (data) => {
                     const message = data.toString();
                     const parsedMessage = JSON.parse(message);
+                    console.log(JSON.stringify(parsedMessage,null,2));
                     
                     switch (parsedMessage.type) {
                         case 'connection_ack':
@@ -254,7 +262,8 @@ class WebSocketUtils {
                 ws.on('error', (err) => {
                     console.error('WebSocket错误:', err);
                     clearTimeout(timeoutId);
-                    this.isProcessing = false;
+                    this.activeConnections.delete(ws);
+                    console.log(`连接错误，当前活跃连接数: ${this.activeConnections.size}/${this.MAX_CONNECTIONS}`);
                     if (ws.readyState === WebSocket.OPEN) {
                         ws.close();
                     }
@@ -264,7 +273,8 @@ class WebSocketUtils {
                 ws.on('close', (code, reason) => {
                     console.log('请求完毕，关闭连接');
                     clearTimeout(timeoutId);
-                    this.isProcessing = false;
+                    this.activeConnections.delete(ws);
+                    console.log(`连接关闭，当前活跃连接数: ${this.activeConnections.size}/${this.MAX_CONNECTIONS}`);
                     if (!isComplete) {
                         reject(new Error('WebSocket closed unexpectedly'));
                     }
@@ -272,9 +282,12 @@ class WebSocketUtils {
             });
         } catch (error) {
             clearTimeout(timeoutId);
-            this.isProcessing = false;
-            if (ws && ws.readyState === WebSocket.OPEN) {
-                ws.close();
+            if (ws) {
+                this.activeConnections.delete(ws);
+                console.log(`发生错误，当前活跃连接数: ${this.activeConnections.size}/${this.MAX_CONNECTIONS}`);
+                if (ws.readyState === WebSocket.OPEN) {
+                    ws.close();
+                }
             }
             throw error;
         }
@@ -330,6 +343,11 @@ class WebSocketUtils {
         }
         return null;
     }
+
+    // 获取当前活跃连接数
+    static getActiveConnectionsCount() {
+        return this.activeConnections.size;
+    }
 }
 
 // 创建Express应用