Node.js如何实现类似Python生成器的分段流式发送响应而非一次性返回？

问题原因 #

当前代码未达到预期效果的核心原因有两点：

• 注释处的10秒耗时逻辑如果是同步阻塞代码，会直接卡住Node.js事件循环，两次res.write的输出都会暂存在缓冲区，直到阻塞结束调用res.end()后才会一次性发送给客户端
• 未显式配置流式响应的必要响应头，部分客户端、代理服务会默认缓冲小体积的响应内容

可行实现方案 #

1. 异步耗时场景实现（最常用，适用于IO等待类耗时逻辑） #

如果你的10秒耗时是IO操作（接口请求、数据库查询、文件读写等），直接按异步逻辑编写即可，原生支持流式输出：

const http = require('http');
const server = http.createServer((req, res) => {
  if (req.url === "/" || req.url === "") {
    // 配置流式响应必要头信息
    res.writeHead(200, {
      'Content-Type': 'text/plain; charset=utf-8',
      'Transfer-Encoding': 'chunked',
      'Cache-Control': 'no-cache'
    });
    // 输出第一段响应
    res.write("Hello 1\n");
    // 模拟10秒异步耗时逻辑，替换为实际业务代码即可
    setTimeout(() => {
      res.write("Hello 2\n");
      res.end();
    }, 10000);
  }
});
server.listen(3000, () => console.log('服务启动在3000端口'));

2. 同步CPU密集型耗时场景实现 #

如果你的10秒耗时是CPU密集型同步逻辑，不能放在主线程直接运行，需要用worker_threads子线程处理避免阻塞事件循环：

const http = require('http');
const { Worker } = require('worker_threads');
const server = http.createServer((req, res) => {
  if (req.url === "/" || req.url === "") {
    res.writeHead(200, {
      'Content-Type': 'text/plain; charset=utf-8',
      'Transfer-Encoding': 'chunked',
      'Cache-Control': 'no-cache'
    });
    res.write("Hello 1\n");
    // 子线程处理耗时同步逻辑
    const worker = new Worker(`
      const { parentPort } = require('worker_threads');
      // 此处放置10秒同步耗时逻辑
      const start = Date.now();
      while(Date.now() - start < 10000);
      parentPort.postMessage('done');
    `, { eval: true });
    worker.on('message', () => {
      res.write("Hello 2\n");
      res.end();
    });
  }
});
server.listen(3000);

注意事项 #

• 部分浏览器默认会缓冲前1KB左右的响应内容，如果你返回的初始内容过小将不会实时渲染，可在第一次res.write时填充少量无意义字符（比如空格）绕过缓冲
• 如果使用Express/Koa等框架，核心逻辑不变，直接调用对应响应对象的write方法即可实现分段输出

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