如何在Express路由中连接断开时终止函数执行？

我来帮你梳理下这个问题的核心：当用户断开连接或者切换新过滤器时，后台的耗时MongoDB查询还在继续运行，既浪费服务器资源，还可能导致重复的回调处理。下面是一套实用的解决方案：

核心思路 #

利用Node.js原生的AbortController结合MongoDB驱动的信号支持，在连接断开时主动终止查询；同时在路由层面管理查询实例，避免重复调用的问题。

具体实现步骤 #

1. 修改myFunction，支持取消信号 #

首先要让你的耗时查询函数接收一个AbortSignal参数，这样外部可以主动触发取消操作：

module.myFunction = async (req, signal) => {
  try {
    // 把signal传递给MongoDB查询，这是终止查询的关键
    const result = await db.collection('yourTargetCollection')
      .find({ /* 你的大数据过滤条件 */ })
      // 其他查询选项，比如sort、limit等
      .toArray({ signal });

    return { err: null, result };
  } catch (err) {
    // 区分"查询被取消"的错误和普通业务错误
    if (err.name === 'AbortError') {
      console.log('耗时查询已被主动终止');
      return { err: '查询已取消', result: null };
    }
    return { err, result: null };
  }
};

2. 在路由中管理查询生命周期，监听连接断开 #

在路由处理函数里，创建AbortController实例，监听请求的断开事件，同时处理重复调用的情况：

app.post('/routeName', async (req, res, next) => {
  // 标记当前是否有活跃查询，避免重复执行
  let isQueryRunning = false;
  const controller = new AbortController();
  const { signal } = controller;

  // 统一处理连接断开/请求中止的逻辑
  const cancelQuery = () => {
    if (!signal.aborted) {
      controller.abort();
      console.log('客户端断开连接，已终止后台查询');
    }
  };
  req.on('close', cancelQuery);
  req.on('aborted', cancelQuery);

  try {
    // 如果已有查询在运行，先取消旧的再执行新的
    if (isQueryRunning) {
      controller.abort();
    }
    isQueryRunning = true;

    // 调用带取消信号的查询函数
    const { err, result } = await module.myFunction(req, signal);

    if (err) {
      if (err === '查询已取消') {
        return res.status(499).send('请求已被取消'); // 499是Nginx定义的客户端断开状态码，可选
      }
      return next(err);
    }

    // 正常返回查询结果
    res.json(result);
  } catch (err) {
    next(err);
  } finally {
    isQueryRunning = false;
    // 移除事件监听，避免内存泄漏
    req.off('close', cancelQuery);
    req.off('aborted', cancelQuery);
  }
});

关键细节说明 #

• AbortController的优势：这是Node.js原生的异步操作取消API，MongoDB驱动（v4.10及以上版本）已经支持通过signal参数终止查询，能真正让MongoDB停止处理请求，而不是仅仅在客户端忽略结果。
• 监听两个事件：close事件在客户端主动断开连接时触发，aborted事件在请求被中止（比如用户刷新页面）时触发，两者都需要处理才能覆盖所有断开场景。
• 避免重复调用：通过isQueryRunning标记，当用户切换过滤器发起新请求时，先终止之前的查询，再执行新的，确保同一时间只有一个查询在运行。
• 清理事件监听：在finally块中移除事件监听，防止长期运行后出现内存泄漏问题。

额外提示 #

如果你的MongoDB驱动版本较低，不支持signal参数，可以尝试用cursor.close()终止查询游标，但这种方式不如AbortController可靠——游标关闭后，MongoDB可能还在后台处理剩余的数据。建议优先升级到最新的MongoDB驱动版本。

内容的提问来源于stack exchange，提问作者DannyY2K4StackOverflow