如何强制socket等待SMTP服务器响应？（net.socket、tls.TLSSocket）

你的SMTP服务器在对接外部服务时，因对方发送数据速度过快（延迟约0.5ms），服务器响应生成速度跟不上，出现对方发送MAIL FROM/RCPT TO命令时，服务器仍在发送响应头的异常，最终导致连接故障。以下是针对net.Socket和tls.TLSSocket的强制等待响应解决方案：

1. 实现命令排队与背压控制 #

Node.js的Socket默认支持背压机制，但需手动结合命令队列确保处理顺序，避免并发处理导致的响应混乱：

• 收到客户端命令时，若当前正在处理响应，则将命令加入队列
• 完成当前响应发送（包括等待缓冲区排空）后，再处理队列中的下一个命令

示例代码：

const net = require('net');
const tls = require('tls');

function handleConnection(socket) {
  let isProcessing = false;
  const commandQueue = [];

  socket.on('data', (rawData) => {
    // 拆分SMTP命令（按CRLF分隔）
    const commands = rawData.toString().split('\r\n').filter(cmd => cmd.trim());
    commands.forEach(cmd => {
      isProcessing ? commandQueue.push(cmd) : processCommand(socket, cmd);
    });
  });

  async function processCommand(socket, cmd) {
    isProcessing = true;
    // 生成SMTP响应（替换为你的实际业务逻辑）
    const response = await generateResponse(cmd);
    
    // 发送响应，缓冲区满时等待drain事件
    const canSendImmediate = socket.write(`${response}\r\n`);
    if (!canSendImmediate) {
      await new Promise(resolve => socket.once('drain', resolve));
    }

    // 处理队列中的下一条命令
    isProcessing = false;
    if (commandQueue.length) {
      processCommand(socket, commandQueue.shift());
    }
  }

  async function generateResponse(cmd) {
    // 模拟响应生成耗时（如数据库查询、权限校验）
    await new Promise(resolve => setTimeout(resolve, 8));
    switch (true) {
      case cmd.startsWith('EHLO'):
        return '250 example.com Hello there';
      case cmd.startsWith('MAIL FROM'):
        return '250 OK Sender accepted';
      case cmd.startsWith('RCPT TO'):
        return '250 OK Recipient accepted';
      default:
        return '500 Unknown command';
    }
  }
}

// 启动普通SMTP服务器（25端口）
net.createServer(handleConnection).listen(25, '0.0.0.0');

// 启动TLS加密SMTP服务器（465端口）
tls.createServer({
  key: require('fs').readFileSync('./server-key.pem'),
  cert: require('fs').readFileSync('./server-cert.pem')
}, handleConnection).listen(465, '0.0.0.0');

2. 严格遵循SMTP协议的命令顺序 #

SMTP是命令-响应的同步协议，客户端必须等待服务器响应后才能发送下一条命令。若外部服务违反该规范：

• 跟踪连接状态（如是否已完成EHLO握手），对乱序命令返回503 Bad sequence of commands
• 多次违反协议的客户端直接断开连接，避免资源占用

3. 优化响应生成效率 #

从根源降低响应延迟：

• 缓存固定响应（如EHLO的欢迎信息），避免重复计算
• 异步操作复用资源（如数据库连接池、Redis缓存）
• 简化响应生成逻辑，减少不必要的IO或计算

4. 调整Socket缓冲区（辅助手段） #

通过socket.setSendBufferSize(size)调整发送缓冲区大小，但该值受操作系统限制，仅作为优化补充，核心仍需依赖命令排队机制。

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