Nodejs实现多房间简易聊天室功能

下面是详细的Nodejs实现多房间简易聊天室功能攻略。

一、需求分析

首先，我们需要明确聊天室的基本需求。聊天室是一个可以供多个用户在同一时间和空间下进行在线聊天交流的程序。具体的基本需求如下：

支持多人同时在线聊天；
支持多房间创建与加入；
实现聊天信息的即时同步；
具有用户登录和退出功能；
用户发言时可以看到房间内其他用户的发言内容。

二、技术选型

在技术选型方面，我们选择使用Node.js + Socket.io来实现多房间聊天室功能。具体理由如下：

Node.js具有非阻塞I/O、事件驱动等特点，非常适合实时交互应用；
Socket.io是一款实时通信引擎，支持WebSocket、HTTP长轮询等多种传输方式，使用起来非常方便。

三、具体实现步骤

安装依赖：

在项目根目录下运行以下命令：

npm install socket.io express body-parser

创建服务器：

创建app.js文件，输入以下代码：

// 引入依赖
const express = require('express')
const bodyParser = require('body-parser')
const app = express()

// 创建服务器
const server = require('http').createServer(app)
const io = require('socket.io')(server)

// 假设有两个房间
const room1 = 'room1'
const room2 = 'room2'

// 监听客户端连接
io.on('connect', socket => {
  // 监听客户端房间加入事件
  socket.on('join', room => {
    // 判断加入哪个房间
    if (room === room1) {
      socket.join(room1)
      socket.emit('joined', room1) // 发送join成功事件
    } else if (room === room2) {
      socket.join(room2)
      socket.emit('joined', room2)
    } else {
      socket.emit('fail', '无此房间') // 发送join失败事件
    }
  })

  // 监听客户端发送消息事件
  socket.on('message', data => {
    io.to(data.room).emit('new message', data) // 向当前房间的所有客户端广播
  })
})

// 启动服务器
server.listen(3000, () => {
  console.log('running at http://localhost:3000')
})

上述代码中，首先我们引入了express、body-parser、http和socket.io四个依赖，并创建了一个名为app的express应用程序。接下来，在app实例上创建了一个名为server的http服务器，并使用socket.io将这个服务器升级为支持WebSocket的服务器。我们还创建了两个名为room1和room2的房间。

在以上代码中，我们主要监听了connect、join和message三个事件。当客户端连接到服务器时，connect事件会被触发，然后我们监听了客户端的join事件，根据房间名判断客户端加入哪个房间，并使用join方法将客户端加入到该房间中。同时向客户端发送一个joined事件，表示加入成功。如果加入失败，则向客户端发送一个fail事件。最后，监听了客户端对应房间发送消息事件，并使用io.to方法向该房间的所有客户端广播消息。

创建客户端：

在public/js目录下创建main.js文件，输入以下代码：

let socket = io()

// 监听join成功事件
socket.on('joined', room => {
  console.log(`加入房间${room}成功！`)
})

// 监听fail事件
socket.on('fail', message => {
  console.log(message)
})

// 监听new message事件
socket.on('new message', data => {
  console.log(`[${data.username}]: ${data.message}`)
})

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

// 发送消息
socket.emit('message', {
  room: 'room1',
  username: 'user1',
  message: 'hello'
})

上述代码中，我们首先使用io()方法创建了一个名为socket的socket实例。然后，监听了joined、fail和new message三个事件。其中，joined代表客户端成功加入一个房间，fail代表加入失败，new message代表有新的消息传递过来。接下来，我们使用emit方法向服务器发送了两个事件，一个是join事件，表示加入房间，一个是message事件，表示发送消息。

四、实例说明1：基本聊天室

我们假设我们已经创建了上述的服务器和客户端，并且已经进行了一定的样式美化。下面，我们来模拟一下两个用户进行在线聊天的场景。首先，我们需要打开两个浏览器窗口，分别输入以下地址：

http://localhost:3000

在第一个客户端中，我们将用户加入到room1房间中，发送一条消息。

在第二个客户端中，我们将用户加入到room1房间中，发送一条回复消息。

我们会看到两个客户端都能够及时接收到对方发送的消息，实现了一个基本的聊天室。

五、实例说明2：多房间聊天室

下面，我们来实现一个可以同时支持多个房间的聊天室。我们只需要在客户端发送加入房间事件时指定房间名称，然后在服务器端根据房间名称进行加入和广播即可。

现在，我们假设我们新增了一个名为room2的房间。首先，在app.js文件中修改如下代码：

// 假设有两个房间
const room1 = 'room1'
const room2 = 'room2'

// 监听客户端连接
io.on('connect', socket => {
  // 监听客户端房间加入事件
  socket.on('join', room => {
    // 判断加入哪个房间
    if (room === room1) {
      socket.join(room1)
      socket.emit('joined', room1) // 发送join成功事件
    } else if (room === room2) {
      socket.join(room2)
      socket.emit('joined', room2)
    } else {
      socket.emit('fail', '无此房间') // 发送join失败事件
    }
  })

  // 监听客户端发送消息事件
  socket.on('message', data => {
    io.to(data.room).emit('new message', data) // 向当前房间的所有客户端广播
  })
})

然后，在main.js文件中修改如下代码：

// 加入房间
socket.emit('join', 'room2')

// 发送消息
socket.emit('message', {
  room: 'room2',
  username: 'user2',
  message: 'hello room2!'
})

最后，我们就可以在多个房间中同时进行聊天了。

本站文章如无特殊说明，均为本站原创，如若转载，请注明出处：Nodejs实现多房间简易聊天室功能 - Python技术站