AWS Lambda中实现Express req.on("close")与req.on("end")等效功能

问题描述 #

我有一段Node.js/Express代码，可以检测用户是否在调用response.end();之前关闭了连接。现在我想在AWS Lambda中实现类似功能：

const express = require('express');
const app = express();

app.get('/', function (req, res) {
  res.setHeader('Content-Type', 'text/html; charset=utf-8');
  let count = 0;
  let gracefulClose = false;
  req.on("close", function () {
    if (!gracefulClose) {
      console.log('Connection closed before res.end() called');
    }
  });
  req.on("end", function () {
    gracefulClose = true;
    console.log('res.end() called');
  });
  setInterval(() => {
    if (count === 3) {
      res.end();
    } else {
      count += 1;
    }
  }, 1000);
});

app.listen(3000);

请问能否通过Lambda处理器传入的context对象实现该功能？我了解过serverless-express，但尽可能不想使用外部库。

解决方案 #

Lambda的context对象本身无法直接实现类似Express的连接关闭检测——这是因为两者执行模型完全不同：Express直接维护与客户端的TCP连接，能监听连接的close/end事件；而Lambda由触发源（如API Gateway）转发请求，客户端的连接状态不会直接暴露给Lambda函数。

不过可以通过以下方式实现类似的检测逻辑：

1. 监听进程SIGTERM信号

当客户端提前关闭连接时，触发源（如API Gateway）会终止Lambda执行，此时Node.js进程会收到SIGTERM信号。我们可以监听这个信号来判断是否是用户提前断开连接。

2. 代码示例

exports.handler = async (event, context) => {
  let gracefulClose = false;
  let intervalId;

  // 监听SIGTERM：客户端断开或Lambda超时前会触发
  const handleTermination = () => {
    if (!gracefulClose) {
      console.log('Connection closed before response sent');
    }
    if (intervalId) clearInterval(intervalId);
  };
  process.on('SIGTERM', handleTermination);

  try {
    return new Promise((resolve) => {
      let count = 0;
      intervalId = setInterval(() => {
        if (count === 3) {
          gracefulClose = true;
          console.log('Response sent successfully');
          clearInterval(intervalId);
          process.off('SIGTERM', handleTermination);
          resolve({
            statusCode: 200,
            body: JSON.stringify({ message: 'Done' })
          });
        } else {
          count += 1;
        }
      }, 1000);
    });
  } catch (error) {
    gracefulClose = true;
    process.off('SIGTERM', handleTermination);
    console.error('Error occurred:', error);
    return {
      statusCode: 500,
      body: JSON.stringify({ error: error.message })
    };
  }
};

3. 注意事项

• SIGTERM信号不仅会在客户端断开连接时触发，Lambda执行超时、手动终止函数时也会触发，无法100%区分场景。
• 发送响应后必须清理信号监听和异步任务，避免内存泄漏。
• 若需等待异步任务完成，可设置context.callbackWaitsForEmptyEventLoop = true，但会增加函数执行时间，需权衡使用。

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