Node.js中如何在Websocket与Socket均连接时执行第三方回调函数？

这里提供两种实用的实现方式，根据你的业务场景选择：

方法一：状态标记法（简单直观） #

通过两个布尔变量跟踪连接状态，每次连接建立时检查是否双端就绪，就绪则触发目标回调：

const net = require('net');

// 初始化连接状态标记
let hasWsClient = false;
let hasPythonConn = false;

// 定义双连接就绪后要执行的逻辑
function handleBothConnected() {
  console.log("WebSocket客户端和Python服务器均已连接，开始执行后续操作");
  // 在这里编写你的业务代码，比如消息转发、状态同步等
}

// WebSocket客户端连接处理
io.on('connection', (clientSocket) => {
  hasWsClient = true;
  // 检查双连接是否都就绪
  if (hasPythonConn) {
    handleBothConnected();
  }

  // 可选：处理客户端断开连接，重置状态
  clientSocket.on('disconnect', () => {
    hasWsClient = false;
  });

  // 原客户端逻辑保留
  // Do some stuff with the client
});

// Python Socket连接处理（注意：优先监听connect事件确认连接建立，而非data）
const pythonSocket = net.createConnection({ port: 你的Python服务端口, host: 'Python服务IP' });

pythonSocket.on('connect', () => {
  hasPythonConn = true;
  // 检查双连接是否都就绪
  if (hasWsClient) {
    handleBothConnected();
  }
});

// 可选：处理Python服务断开连接，重置状态
pythonSocket.on('close', () => {
  hasPythonConn = false;
});

// 原Python数据处理逻辑保留
pythonSocket.on('data', (msg) => {
  // Do some stuff with the Python server
});

方法二：Promise.all异步流程管理（优雅可控） #

将两个连接事件包装成Promise，利用Promise.all等待双端就绪后统一执行回调，还能方便捕获连接错误：

const net = require('net');

// 包装WebSocket连接事件为Promise（等待第一个客户端连接）
const waitForWsClient = new Promise((resolve) => {
  io.on('connection', (clientSocket) => {
    resolve(clientSocket); // 可将客户端Socket传入后续逻辑
    // 原客户端逻辑保留
    // Do some stuff with the client
  });
});

// 包装Python Socket连接事件为Promise
const waitForPythonConn = new Promise((resolve, reject) => {
  const pythonSocket = net.createConnection({ port: 你的Python服务端口, host: 'Python服务IP' });
  
  pythonSocket.on('connect', () => {
    resolve(pythonSocket); // 可将Python Socket传入后续逻辑
  });

  // 处理连接失败的情况
  pythonSocket.on('error', (err) => {
    reject(err);
  });

  // 原Python数据处理逻辑保留
  pythonSocket.on('data', (msg) => {
    // Do some stuff with the Python server
  });
});

// 等待双连接就绪后执行逻辑
Promise.all([waitForWsClient, waitForPythonConn])
  .then(([clientSocket, pythonSocket]) => {
    console.log("双连接已就绪，执行目标操作");
    // 示例：实现WebSocket客户端与Python服务器的消息转发
    clientSocket.on('message', (data) => {
      pythonSocket.write(data);
    });
  })
  .catch((err) => {
    console.error("连接失败:", err);
  });

注意事项 #

• 若你的WebSocket需要支持多客户端，方法一中的hasWsClient可以改为统计在线客户端数量，判断数量>0即可；方法二则可以修改为监听所有客户端连接，根据需求调整。
• 原代码中监听Python Socket的data事件是处理收到的数据，而非确认连接建立，建议通过connect事件判断Python服务是否连接成功。

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