好的,以下是一个简单的示例代码,可以让 Node.js 创建一个 WebSocket 服务,并实现群发消息:
const WebSocket = require('ws');
// 创建 WebSocket server 并将其绑定到 HTTP server 上
const httpServer = new WebSocket.Server({ port: 8080 });
// 存储所有已连接的 WebSocket 客户端
let clients = [];
httpServer.on('connection', (websocket, req) => {
console.log(`New WebSocket connection from ${req.socket.remoteAddress}:${req.socket.remotePort}`);
// 将该连接添加到所有客户端列表中
clients.push(websocket);
websocket.on('message', (message) => {
console.log(`Received message from client: ${message}`);
broadcastMessage(message);
});
websocket.on('close', () => {
console.log(`WebSocket connection closed from ${req.socket.remoteAddress}:${req.socket.remotePort}`);
clients.splice(clients.indexOf(websocket), 1);
});
});
function broadcastMessage(message) {
// 广播消息给所有已连接的客户端
clients.forEach(client => {
if (client.readyState === WebSocket.OPEN) {
client.send(message);
}
});
}
console.log('WebSocket server is running on port 8080');
这个例子中,Node.js 创建了一个本地 WebSocket server 监听在端口号 8080 上。当有新的 WebSocket 连接时,将其添加到存储所有客户端的数组中,并在收到消息时将其转发给所有已连接的客户端;当某个 WebSocket 连接关闭时,从客户端列表中删除该连接。函数 broadcastMessage 实现了广播消息给所有已连接的客户端的功能。
你可以使用浏览器或者 WebSocket 客户端来测试这个服务。例如,你可以在多个浏览器窗口或标签页中打开 http://localhost:8080 并分别向该页面发送消息,即可看到所有窗口都收到了相同的消息。
希望这个示例可以帮助你了解如何在 Node.js 中创建 WebSocket 服务,并实现群发消息。




