NestJS如何直接获取指定Socket？无需fetchSockets遍历

你完全可以不用先拉取所有Socket再遍历，Socket.IO本身提供了直接通过ID定位Socket的方式，不需要依赖fetchSockets()。

Socket.IO的Server实例下，sockets.sockets是一个Map结构，其中键就是Socket的ID，值对应具体的Socket实例。你可以直接通过ID从这个Map中获取目标Socket，时间复杂度是O(1)，比遍历全列表高效得多。

修改后的代码示例：

@WebSocketServer()
server: Server = new Server<ServerToClientEvents, ClientToServerEvents>();

@SubscribeMessage('event')
async handelInitial(@MessageBody() data: any): Promise<void> {
  // 直接通过ID获取目标Socket
  const targetSocket = this.server.sockets.sockets.get('required socket id');
  
  if (targetSocket) {
    // 找到Socket后的逻辑处理
    console.log('找到目标Socket:', targetSocket.id);
  } else {
    console.log('未找到对应ID的Socket');
  }
}

注意事项 #

如果你的WebSocket使用了自定义命名空间（而非默认的根命名空间），需要先指定对应的命名空间再获取Socket：

// 假设命名空间为'/chat'
const targetSocket = this.server.of('/chat').sockets.get('required socket id');

这个方法直接复用了Socket.IO内部维护的Socket映射表，避免了全量拉取和遍历的开销，完全符合你的需求。

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