javascript - 如何在 Node js 中使用 ws (Websocket) 包创建/加入聊天室

Question

我正在使用 ws 在服务器端打包，我希望客户端在服务器套接字中创建/加入房间。并在它们不再连接时将它们从创建的房间中删除。
PS:我不想使用socketIo。

Answer 1

您可以尝试以下方法:

const rooms = {};

wss.on("connection", socket => {
  const uuid = ...; // create here a uuid for this connection

  const leave = room => {
    // not present: do nothing
    if(! rooms[room][uuid]) return;

    // if the one exiting is the last one, destroy the room
    if(Object.keys(rooms[room]).length === 1) delete rooms[room];
    // otherwise simply leave the room
    else delete rooms[room][uuid];
  };

  socket.on("message", data => {
    const { message, meta, room } = data;

    if(meta === "join") {
      if(! rooms[room]) rooms[room] = {}; // create the room
      if(! rooms[room][uuid]) rooms[room][uuid] = socket; // join the room
    }
    else if(meta === "leave") {
      leave(room);
    }
    else if(! meta) {
      // send the message to all in the room
      Object.entries(rooms[room]).forEach(([, sock]) => sock.send({ message }));
    }
  });

  socket.on("close", () => {
    // for each room, remove the closed socket
    Object.keys(rooms).forEach(room => leave(room));
  });
});

这只是一个草图:您需要处理离开房间、与客户断开连接(离开所有房间)以及在没有人再进来时删除房间。