在Web应用中使用WebSocket实现实时聊天功能

const socket = new WebSocket('ws://localhost:8000/chat');

// 连接建立成功后的回调函数
socket.onopen = function() {
  console.log('WebSocket连接已建立');
};

// 接收到消息时的回调函数
socket.onmessage = function(event) {
  const message = JSON.parse(event.data);
  console.log('收到消息：', message);
};

// 关闭连接时的回调函数
socket.onclose = function() {
  console.log('WebSocket连接已关闭');
};

// 发送消息的函数
function sendMessage(message) {
  socket.send(jsON.stringify(message));
}

const WebSocket = require('ws');

// 创建WebSocket服务器
const wss = new WebSocket.Server({ port: 8000 });

// 保存所有连接到服务器的客户端
const clients = new Set();

// 处理客户端连接事件
wss.on('connection', function(ws) {
  console.log('客户端已连接');

  // 将客户端添加到集合中
  clients.add(ws);

  // 处理收到消息的事件
  ws.on('message', function(message) {
    console.log('收到消息：', message);

    // 将消息发送给所有连接的客户端
    clients.forEach(function(client) {
      client.send(message);
    });
  });

  // 处理连接关闭事件
  ws.on('close', function() {
    console.log('客户端已关闭');

    // 将客户端从集合中移除
    clients.delete(ws);
  });
});

node server.js