Node.js中如何实现Socket重用：连接失败复用及两次http.request调用复用

我来帮你搞定这两个Socket重用的需求，结合你遇到的自定义Agent中Socket被提前移除的问题，给你具体的实现方案：

一、连接失败后重用Socket #

默认情况下，Node.js的http.ClientRequest在连接失败（比如超时、服务器拒绝）时会自动销毁Socket，但我们可以手动接管Socket的生命周期，阻止默认销毁行为，从而在后续请求中复用它。

实现思路： #

• 维护一个全局的Socket引用，用于保存可复用的连接
• 发起请求时手动指定socket选项，尝试复用已有Socket
• 监听Socket的error和timeout事件，不执行默认的销毁操作
• 请求失败后，保留Socket引用，下次请求继续复用

示例代码： #

const http = require('http');

// 保存可复用的Socket引用
let reusableSocket = null;

function createReusableRequest(options) {
  // 发起请求时传入已有的Socket
  const req = http.request({
    ...options,
    socket: reusableSocket
  });

  req.on('socket', (socket) => {
    if (!reusableSocket) {
      reusableSocket = socket;
      // 监听错误，阻止默认销毁
      socket.on('error', (err) => {
        console.error('Socket 出错:', err);
        // 这里不要调用socket.destroy()，保留Socket以便复用
      });
      // 监听超时，同样保留Socket
      socket.on('timeout', () => {
        console.warn('Socket 超时');
      });
    }
  });

  req.on('error', (err) => {
    console.error('请求失败:', err);
    // 请求失败后，Socket依然保留，下次请求继续尝试复用
  });

  return req;
}

// 第一次请求（假设连接失败）
const firstReq = createReusableRequest({
  hostname: 'example.com',
  port: 80,
  path: '/',
  method: 'GET'
});
firstReq.end();

// 2秒后发起第二次请求，复用同一个Socket
setTimeout(() => {
  const secondReq = createReusableRequest({
    hostname: 'example.com',
    port: 80,
    path: '/test',
    method: 'GET'
  });
  secondReq.end();
}, 2000);

二、用同一个Socket发起两次http.request调用 #

你提到的Socket被提前移除的问题，根源在于默认http.Agent的逻辑：当请求响应完成后，Agent会检查Socket是否符合复用条件（比如Connection头是否为keep-alive、是否在keep-alive超时内），如果不符合就会把Socket从池中移除。要强制复用同一个Socket，我们需要自定义Agent，重写关键方法来保留Socket。

实现思路： #

• 自定义Agent继承自http.Agent，设置maxSockets: 1确保池子里只有一个Socket
• 重写createConnection方法，优先返回已有的可用Socket，没有则创建新的并保存引用
• 重写onFree方法，强制将未销毁的Socket放回连接池，忽略默认的Connection头检查
• 确保消费响应数据（调用res.resume()），避免Socket被挂起无法复用

示例代码： #

const http = require('http');

class SingleSocketAgent extends http.Agent {
  constructor(options) {
    // 开启keepAlive，同时限制最大Socket数为1
    super({ ...options, keepAlive: true, maxSockets: 1 });
    this.reusableSocket = null;
  }

  createConnection(options, callback) {
    // 如果已有可用Socket，直接返回
    if (this.reusableSocket && !this.reusableSocket.destroyed) {
      process.nextTick(() => callback(null, this.reusableSocket));
      return;
    }
    // 没有可用Socket时，创建新的并保存引用
    super.createConnection(options, (err, socket) => {
      if (!err) {
        this.reusableSocket = socket;
        // Socket关闭时清理引用
        socket.on('close', () => {
          this.reusableSocket = null;
        });
      }
      callback(err, socket);
    });
  }

  onFree(socket) {
    // 重写onFree，强制保留未销毁的Socket
    if (!socket.destroyed) {
      // 将Socket放回freeSockets，忽略默认的复用条件检查
      this.freeSockets.set(socket._host, [socket]);
    } else {
      // Socket已销毁，调用父类逻辑清理
      super.onFree(socket);
    }
  }
}

// 初始化自定义Agent
const agent = new SingleSocketAgent();

// 第一次请求
const firstReq = http.request({
  hostname: 'example.com',
  port: 80,
  path: '/',
  method: 'GET',
  agent: agent
}, (res) => {
  console.log('第一次请求状态:', res.statusCode);
  // 必须消费响应数据，否则Socket会被挂起
  res.resume();
  res.on('end', () => {
    // 响应结束后发起第二次请求，复用同一个Socket
    const secondReq = http.request({
      hostname: 'example.com',
      port: 80,
      path: '/test',
      method: 'GET',
      agent: agent
    }, (res2) => {
      console.log('第二次请求状态:', res2.statusCode);
      res2.resume();
    });
    secondReq.end();
  });
});
firstReq.end();

关键注意点： #

• 两次请求必须指向同一个主机和端口，因为Socket是和特定端点绑定的
• 如果服务器主动关闭了Socket（比如keep-alive超时），要在close事件中清理引用，下次请求会自动创建新的Socket
• 对于HTTPS请求，只需把http换成https，继承https.Agent即可，逻辑完全一致
• 一定要调用res.resume()消费响应数据，否则Node.js会认为请求还未完成，Socket无法被复用

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