Node.js 的 WebSocket 库：ws、Socket.IO 和 uWebSockets.js

Node.js 的 WebSocket 库：从简单聊天到高负载实时系统

WebSocket 是一种协议，可在客户端和服务器之间提供持久的双向连接。与 HTTP 不同（HTTP 中请求-响应由客户端发起），WebSocket 允许服务器随时发送数据。对于以其异步性和事件驱动架构著称的 Node.js 来说，WebSocket 已成为创建聊天、在线游戏、交易终端和流式数据传输的自然选择。

在本文中，我们将介绍三个主要的库：ws（轻量且快速）、Socket.IO（便捷且具有回退机制）和 uWebSockets.js（性能卓越）。您将了解如何根据您的任务选择合适的库，以及如何使用每个库编写一个简单的聊天程序。

1. ws 库——极简与速度

ws 是 Node.js 最流行的 WebSocket 库。它实现了纯 RFC 6455 协议，没有多余的附加功能。主要优点：

• 最大性能——单核每秒可处理超过 200,000 条消息。
• 零依赖——仅使用纯 JavaScript。
• 完全控制连接：握手、帧、ping/pong。
• 支持 Node.js Buffer 和 Streams 处理二进制数据。

缺点：没有自动重连，没有房间（rooms），没有 HTTP long-polling 回退。所有功能都需要手动实现或通过额外的包来实现。

示例：基于 ws 的简单回声服务器

const WebSocket = require('ws');

const server = new WebSocket.Server({ port: 8080 });

server.on('connection', (ws) => {  console.log('客户端已连接');

  ws.on('message', (message) => {    console.log('收到:', message.toString());    // 发送回去（回声）    ws.send(`回声: ${message}`);  });

  ws.on('close', () => {    console.log('客户端已断开');  });

  // 连接时发送欢迎信息  ws.send('欢迎来到 WebSocket 服务器！');});

console.log('服务器已启动在 ws://localhost:8080');

何时选择 ws？——用于微服务、高负载游戏（每一帧都很重要）、嵌入到自定义 HTTP 服务器中（通过 server.on('upgrade')）。

2. Socket.IO——生产环境的便捷性与可靠性

Socket.IO 不仅仅是一个 WebSocket，而是一个完整的实时平台。它使用 WebSocket 作为主要传输方式，但如果 WebSocket 不被支持（代理、防火墙、旧浏览器），它会自动切换到 HTTP long-polling。

• 房间（Rooms）——对客户端进行分组以实现广播。
• 命名空间（Namespaces）——应用程序的逻辑划分（例如 /chat 和 /game）。
• 自动重连——带有指数退避延迟。
• 事件支持——您处理事件，而不是原始消息。
• 内置序列化——无需手动解析即可传输 JSON 对象。

缺点：协议开销（带有元数据的额外数据包），更多依赖项，比纯 ws 稍慢。

示例：基于 Socket.IO 的带房间聊天（服务器端）

const http = require('http');const { Server } = require('socket.io');

const server = http.createServer();const io = new Server(server, {  cors: { origin: '*' } // 用于开发});

io.on('connection', (socket) => {  console.log('新客户端:', socket.id);

  // 加入房间  socket.on('join_room', (roomName) => {    socket.join(roomName);    socket.emit('message', `您已加入房间 ${roomName}`);    socket.to(roomName).emit('message', `用户 ${socket.id} 已进入房间`);  });

  // 向房间发送消息