如何在http-proxy钩子中异步调用外部服务并解决头设置错误

错误原因 #

你遇到的Error [ERR_HTTP_HEADERS_SENT]是因为在proxyReq事件触发时，代理默认的请求转发逻辑已经准备向目标服务器发送请求了。如果此时异步调用外部服务，等拿到结果再设置请求头，代理早已把请求发出去，自然无法再修改已发送的HTTP头。

解决方案 #

1. HTTP请求：提前异步获取请求头再转发 #

不要在proxyReq事件里做异步操作，而是在代理转发请求之前完成外部服务调用，修改原请求头后再发起代理：

const http = require('http');
const httpProxy = require('http-proxy');

const proxy = httpProxy.createProxyServer({});
const targetServer = 'http://你的目标服务器地址';

// 创建HTTP服务器，先处理请求头再转发
const server = http.createServer(async (req, res) => {
  try {
    // 调用外部服务获取额外请求头
    const extraHeaders = await fetchExternalServiceForReqHeaders(req);
    // 将额外头合并到原请求头
    Object.assign(req.headers, extraHeaders);
    
    // 转发请求到目标服务器
    proxy.web(req, res, { target: targetServer });
  } catch (err) {
    console.error('获取请求头失败:', err);
    res.writeHead(500);
    res.end('代理请求预处理失败');
  }
});

// 处理响应拦截与外部服务调用
proxy.on('proxyRes', async (proxyRes, req, res) => {
  // 暂停响应流，防止异步处理时响应已发给客户端
  proxyRes.pause();

  try {
    // 调用外部服务处理响应数据
    const processResult = await fetchExternalServiceForRes(proxyRes, req);

    // 读取原始响应体，结合外部服务结果修改
    let rawBody = [];
    proxyRes.on('data', chunk => rawBody.push(chunk));
    proxyRes.on('end', () => {
      rawBody = Buffer.concat(rawBody).toString();
      // 根据外部服务结果修改响应体（示例逻辑）
      const modifiedBody = modifyResponse(rawBody, processResult);
      
      // 更新响应头（比如Content-Length）
      res.setHeader('Content-Length', Buffer.byteLength(modifiedBody));
      // 发送修改后的响应给客户端
      res.end(modifiedBody);
    });
  } catch (err) {
    console.error('处理响应失败:', err);
    res.writeHead(500);
    res.end('响应处理失败');
  }
});

server.listen(3000);

2. WebSocket请求：销毁原连接后重新发起 #

WebSocket的处理逻辑类似，需要在proxyReqWs事件中先销毁当前连接，异步获取头后重新发起代理：

proxy.on('proxyReqWs', async (proxyReq, req, socket, options, head) => {
  // 销毁当前socket，避免异步过程中已建立连接
  socket.destroy();

  try {
    // 调用外部服务获取WebSocket请求头
    const extraWsHeaders = await fetchExternalServiceForWsHeaders(req);
    // 修改原请求头
    Object.assign(req.headers, extraWsHeaders);
    
    // 重新发起WebSocket代理请求
    proxy.ws(req, socket, options, head);
  } catch (err) {
    console.error('WebSocket请求预处理失败:', err);
    // 向客户端返回错误
    socket.write('HTTP/1.1 500 Internal Server Error\r\n\r\n');
    socket.destroy();
  }
});

关键注意点 #

• 所有异步操作必须在代理请求/连接发起前完成，避免与代理默认的发送/连接逻辑冲突。
• 处理响应时必须暂停响应流，防止异步处理期间响应已被发送给客户端。
• WebSocket场景下，必须销毁原socket后重新发起，否则原连接会在异步过程中建立，导致无法修改头。

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