修复AbortSignal监听器内存泄漏问题

2025-10-04 18:53:04 +08:00
parent 6b9b38a80c
commit 9a4cd48c3b
3 changed files with 277 additions and 19 deletions
@@ -0,0 +1,229 @@
+const { defaultLogger } = require('./logger');
+
+// 创建logger实例
+const logger = defaultLogger.child('AbortControllerManager');
+
+/**
+ * AbortController 管理器 - 防止内存泄漏
+ */
+class AbortControllerManager {
+  constructor() {
+    // 全局 AbortController 跟踪
+    this.controllers = new Map();
+    // 监听器数量限制
+    this.maxListenersPerController = 10;
+    this.maxTotalControllers = 50;
+    
+    // 定期清理间隔（5分钟）
+    this.cleanupInterval = 5 * 60 * 1000;
+    this.cleanupTimer = null;
+    
+    this.startPeriodicCleanup();
+  }
+
+  /**
+   * 创建新的 AbortController
+   */
+  createController(id) {
+    // 检查全局控制器数量限制
+    if (this.controllers.size >= this.maxTotalControllers) {
+      logger.warn(`AbortController 数量已达上限 (${this.maxTotalControllers})，拒绝创建新控制器`);
+      return null;
+    }
+
+    const controller = new AbortController();
+    const controllerInfo = {
+      id,
+      controller,
+      createdAt: Date.now(),
+      listenerCount: 0
+    };
+
+    this.controllers.set(id, controllerInfo);
+    logger.debug(`创建 AbortController: ${id}，当前总数: ${this.controllers.size}`);
+    
+    return controller;
+  }
+
+  /**
+   * 获取 AbortController
+   */
+  getController(id) {
+    const info = this.controllers.get(id);
+    return info ? info.controller : null;
+  }
+
+  /**
+   * 添加监听器（带限制检查）
+   */
+  addListener(id, event, listener) {
+    const info = this.controllers.get(id);
+    if (!info) {
+      logger.warn(`未找到 AbortController: ${id}`);
+      return false;
+    }
+
+    if (info.listenerCount >= this.maxListenersPerController) {
+      logger.warn(`AbortController ${id} 的监听器数量已达上限 (${this.maxListenersPerController})`);
+      return false;
+    }
+
+    info.controller.signal.addEventListener(event, listener);
+    info.listenerCount++;
+    logger.debug(`为 AbortController ${id} 添加监听器，当前数量: ${info.listenerCount}`);
+    
+    return true;
+  }
+
+  /**
+   * 移除监听器
+   */
+  removeListener(id, event, listener) {
+    const info = this.controllers.get(id);
+    if (!info) {
+      return false;
+    }
+
+    info.controller.signal.removeEventListener(event, listener);
+    info.listenerCount = Math.max(0, info.listenerCount - 1);
+    logger.debug(`从 AbortController ${id} 移除监听器，剩余数量: ${info.listenerCount}`);
+    
+    return true;
+  }
+
+  /**
+   * 中断并清理 AbortController
+   */
+  abortAndCleanup(id) {
+    const info = this.controllers.get(id);
+    if (!info) {
+      return false;
+    }
+
+    try {
+      if (!info.controller.signal.aborted) {
+        info.controller.abort();
+      }
+    } catch (error) {
+      logger.warn(`中断 AbortController ${id} 时出错:`, error.message);
+    } finally {
+      this.controllers.delete(id);
+      logger.debug(`清理 AbortController: ${id}，剩余总数: ${this.controllers.size}`);
+    }
+
+    return true;
+  }
+
+  /**
+   * 清理所有控制器
+   */
+  cleanupAll() {
+    logger.info(`清理所有 AbortController，当前数量: ${this.controllers.size}`);
+    
+    for (const [id, info] of this.controllers) {
+      try {
+        if (!info.controller.signal.aborted) {
+          info.controller.abort();
+        }
+      } catch (error) {
+        logger.warn(`清理 AbortController ${id} 时出错:`, error.message);
+      }
+    }
+    
+    this.controllers.clear();
+  }
+
+  /**
+   * 获取统计信息
+   */
+  getStats() {
+    const totalControllers = this.controllers.size;
+    let totalListeners = 0;
+    let oldestController = null;
+    let oldestAge = 0;
+
+    for (const [id, info] of this.controllers) {
+      totalListeners += info.listenerCount;
+      const age = Date.now() - info.createdAt;
+      if (age > oldestAge) {
+        oldestAge = age;
+        oldestController = id;
+      }
+    }
+
+    return {
+      totalControllers,
+      totalListeners,
+      oldestController,
+      oldestAge: Math.round(oldestAge / 1000), // 秒
+      maxTotalControllers: this.maxTotalControllers,
+      maxListenersPerController: this.maxListenersPerController
+    };
+  }
+
+  /**
+   * 开始定期清理检查
+   */
+  startPeriodicCleanup() {
+    if (this.cleanupTimer) {
+      return;
+    }
+
+    this.cleanupTimer = setInterval(() => {
+      const stats = this.getStats();
+      
+      // 记录统计信息
+      if (stats.totalControllers > 0) {
+        logger.debug('AbortController 统计:', {
+          总控制器数: stats.totalControllers,
+          总监听器数: stats.totalListeners,
+          最老控制器: stats.oldestController,
+          最老年龄: `${stats.oldestAge}秒`
+        });
+
+        // 警告检查
+        if (stats.totalControllers > stats.maxTotalControllers * 0.8) {
+          logger.warn(`AbortController 数量接近上限: ${stats.totalControllers}/${stats.maxTotalControllers}`);
+        }
+
+        if (stats.totalListeners > stats.maxTotalControllers * stats.maxListenersPerController * 0.8) {
+          logger.warn(`AbortController 监听器总数过多: ${stats.totalListeners}`);
+        }
+
+        // 清理超过 30 分钟的控制器
+        const maxAge = 30 * 60 * 1000; // 30分钟
+        const now = Date.now();
+        const toCleanup = [];
+
+        for (const [id, info] of this.controllers) {
+          if (now - info.createdAt > maxAge) {
+            toCleanup.push(id);
+          }
+        }
+
+        if (toCleanup.length > 0) {
+          logger.info(`清理 ${toCleanup.length} 个超时的 AbortController`);
+          toCleanup.forEach(id => this.abortAndCleanup(id));
+        }
+      }
+    }, this.cleanupInterval);
+
+    logger.info('AbortController 定期清理已启动');
+  }
+
+  /**
+   * 停止定期清理
+   */
+  stopPeriodicCleanup() {
+    if (this.cleanupTimer) {
+      clearInterval(this.cleanupTimer);
+      this.cleanupTimer = null;
+      logger.info('AbortController 定期清理已停止');
+    }
+  }
+}
+
+// 创建全局实例
+const abortControllerManager = new AbortControllerManager();
+
+module.exports = abortControllerManager;