Node.js 实现聊天室在线人数的更新-JavaScript中文网-JavaScript教程资源分享门户

在一个聊天室中，我们通常需要实时更新在线人数，这对于用户来说是很重要的。在 Node.js 中，我们可以使用 WebSocket 技术来实现在线人数的更新。本文将介绍如何使用 Node.js 和 WebSocket 实现聊天室在线人数的实时更新。

WebSocket 简介

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议。它可以在客户端和服务器之间建立一个持久的连接，这使得服务器可以向客户端发送实时数据。WebSocket 最初是为了解决 HTTP 请求响应模型的局限性而设计的。

实现步骤

1. 安装 WebSocket 模块

我们可以使用 npm 来安装 WebSocket 模块。在命令行中输入以下命令：

npm install websocket

2. 创建 WebSocket 服务器

在 Node.js 中，我们可以使用 WebSocketServer 类来创建 WebSocket 服务器。以下是一个简单的 WebSocket 服务器的示例代码：

// javascriptcn.com 代码示例
const WebSocketServer = require('websocket').server;
const http = require('http');

const server = http.createServer((req, res) => {
  console.log((new Date()) + ' Received request for ' + req.url);
  res.writeHead(404);
  res.end();
});

server.listen(8080, () => {
  console.log((new Date()) + ' Server is listening on port 8080');
});

const wsServer = new WebSocketServer({
  httpServer: server,
  autoAcceptConnections: false
});

wsServer.on('request', (req) => {
  const connection = req.accept('', req.origin);
  console.log((new Date()) + ' Connection accepted.');

  // 在这里处理连接
});

在上面的代码中，我们创建了一个 HTTP 服务器并使用 WebSocketServer 类将其升级为 WebSocket 服务器。我们使用 autoAcceptConnections 选项来防止非法连接，只接受经过验证的连接请求。

3. 处理连接

在接受 WebSocket 连接后，我们需要处理连接。以下是一个处理连接的示例代码：

// javascriptcn.com 代码示例
const connections = [];

wsServer.on('request', (req) => {
  const connection = req.accept('', req.origin);
  console.log((new Date()) + ' Connection accepted.');

  // 将连接添加到 connections 数组中
  connections.push(connection);

  // 发送在线人数给所有连接
  connections.forEach((c) => {
    c.sendUTF(connections.length);
  });

  // 处理关闭连接事件
  connection.on('close', () => {
    console.log((new Date()) + ' Peer ' + connection.remoteAddress + ' disconnected.');
    // 从 connections 数组中删除连接
    connections.splice(connections.indexOf(connection), 1);

    // 发送在线人数给所有连接
    connections.forEach((c) => {
      c.sendUTF(connections.length);
    });
  });
});

在上面的代码中，我们将连接添加到 connections 数组中，并向所有连接发送在线人数。在连接关闭时，我们从 connections 数组中删除连接，并向所有连接发送在线人数。

示例

以下是一个完整的聊天室在线人数更新的示例代码：

// javascriptcn.com 代码示例
const WebSocketServer = require('websocket').server;
const http = require('http');

const server = http.createServer((req, res) => {
  console.log((new Date()) + ' Received request for ' + req.url);
  res.writeHead(404);
  res.end();
});

server.listen(8080, () => {
  console.log((new Date()) + ' Server is listening on port 8080');
});

const wsServer = new WebSocketServer({
  httpServer: server,
  autoAcceptConnections: false
});

const connections = [];

wsServer.on('request', (req) => {
  const connection = req.accept('', req.origin);
  console.log((new Date()) + ' Connection accepted.');

  // 将连接添加到 connections 数组中
  connections.push(connection);

  // 发送在线人数给所有连接
  connections.forEach((c) => {
    c.sendUTF(connections.length);
  });

  // 处理关闭连接事件
  connection.on('close', () => {
    console.log((new Date()) + ' Peer ' + connection.remoteAddress + ' disconnected.');
    // 从 connections 数组中删除连接
    connections.splice(connections.indexOf(connection), 1);

    // 发送在线人数给所有连接
    connections.forEach((c) => {
      c.sendUTF(connections.length);
    });
  });
});

总结

在本文中，我们介绍了如何使用 Node.js 和 WebSocket 实现聊天室在线人数的实时更新。WebSocket 提供了一种可靠的方式来实现实时通信。如果你正在构建一个需要实时更新的应用程序，那么你应该考虑使用 WebSocket。

来源：JavaScript中文网，转载请注明来源本文地址：https://www.javascriptcn.com/post/6577cae8d2f5e1655d180926