WebSocket 聊天室多房间源码大公开

在上一篇文章中，我们介绍了WebSocket的基本概念和一些简单的应用场景。在这篇文章中，我们将继续深入探讨WebSocket，并实现一个多房间的聊天室。

多房间聊天室的实现

1. 服务端实现

首先，我们需要在服务端创建一个WebSocket服务器。可以使用Node.js来实现，可以使用Socket.IO库来简化WebSocket的开发。

// 引入必要的库
const socketIO = require('socket.io');

// 创建WebSocket服务器
const io = socketIO(3000);

// 当客户端连接时触发
io.on('connection', (socket) => {
  console.log('客户端已连接');

  // 当客户端发送消息时触发
  socket.on('message', (message) => {
    console.log('收到客户端消息：', message);

    // 将消息广播给所有客户端
    io.emit('message', message);
  });

  // 当客户端断开连接时触发
  socket.on('disconnect', () => {
    console.log('客户端已断开连接');
  });
});

2. 客户端实现

接下来，我们需要在客户端创建一个WebSocket客户端。可以使用JavaScript来实现，可以使用WebSocket API或Socket.IO客户端库来连接WebSocket服务器。

// 引入必要的库
const socketIO = require('socket.io-client');

// 创建WebSocket客户端
const socket = socketIO('http://localhost:3000');

// 当客户端连接到服务器时触发
socket.on('connect', () => {
  console.log('已连接到WebSocket服务器');

  // 发送消息到服务器
  socket.emit('message', 'Hello from client');
});

// 当客户端收到服务器的消息时触发
socket.on('message', (message) => {
  console.log('收到服务器消息：', message);
});

// 当客户端断开连接时触发
socket.on('disconnect', () => {
  console.log('已断开连接');
});

3. 多房间聊天室的实现

为了实现多房间聊天室，我们需要在服务端和客户端分别做一些修改。

在服务端，我们需要为每个房间创建一个单独的命名空间。命名空间可以理解为一个独立的聊天室，用户可以自由地加入或离开不同的命名空间。

// 创建一个命名空间
const room1 = io.of('/room1');

// 当客户端连接到命名空间时触发
room1.on('connection', (socket) => {
  console.log('客户端已连接到房间1');

  // 将客户端加入房间1
  socket.join('room1');

  // 当客户端发送消息时触发
  socket.on('message', (message) => {
    console.log('收到客户端消息：', message);

    // 将消息广播给房间1中的所有客户端
    room1.emit('message', message);
  });

  // 当客户端断开连接时触发
  socket.on('disconnect', () => {
    console.log('客户端已断开连接');
  });
});

在客户端，我们需要在连接到WebSocket服务器后，加入到指定的房间。

// 加入房间1
socket.emit('join', 'room1');

// 当客户端收到服务器的消息时触发
socket.on('message', (message) => {
  console.log('收到服务器消息：', message);
});

这样，我们就实现了一个多房间的聊天室。用户可以自由地在不同的房间中进行聊天。

总结

在这篇文章中，我们介绍了如何使用WebSocket和Socket.IO来实现一个多房间的聊天室。这个聊天室可以允许用户自由地在不同的房间中进行聊天。

WebSocket是一个非常强大的技术，可以实现多种实时通讯的场景。例如，还可以使用WebSocket来实现在线游戏、多人协作编辑器、股票行情推送等应用。