如何用Node.js实现实时通信

发布时间：2025-02-07 10:55:39 作者：小樊
来源：亿速云阅读：111

在Node.js中实现实时通信，可以使用WebSocket协议。WebSocket是一种在单个TCP连接上进行全双工通信的协议，非常适合实时通信场景。以下是一个简单的示例，展示如何使用Node.js和ws库实现实时通信。

1. 安装依赖

首先，你需要安装ws库。你可以使用npm来安装：

npm install ws

2. 创建服务器

创建一个Node.js服务器，并设置WebSocket支持。

const WebSocket = require('ws');

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

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

  // 监听消息事件
  ws.on('message', function incoming(message) {
    console.log('收到消息:', message);

    // 广播消息给所有客户端
    wss.clients.forEach(function each(client) {
      if (client !== ws && client.readyState === WebSocket.OPEN) {
        client.send(message);
      }
    });
  });

  // 监听关闭事件
  ws.on('close', function close() {
    console.log('客户端已断开连接');
  });

  // 监听错误事件
  ws.on('error', function error(err) {
    console.error('发生错误:', err);
  });
});

console.log('WebSocket服务器正在监听端口 8080');

3. 创建客户端

创建一个简单的HTML文件来测试WebSocket客户端。

<!DOCTYPE html>
<html lang="en">
<head>
  <meta charset="UTF-8">
  <meta name="viewport" content="width=device-width, initial-scale=1.0">
  <title>WebSocket Client</title>
</head>
<body>
  <h1>WebSocket Client</h1>
  <input type="text" id="messageInput" placeholder="输入消息">
  <button id="sendButton">发送</button>
  <ul id="messages"></ul>

  <script>
    const socket = new WebSocket('ws://localhost:8080');

    // 监听连接打开事件
    socket.addEventListener('open', function(event) {
      console.log('连接已打开:', event);
    });

    // 监听消息事件
    socket.addEventListener('message', function(event) {
      const messagesList = document.getElementById('messages');
      const newMessage = document.createElement('li');
      newMessage.textContent = event.data;
      messagesList.appendChild(newMessage);
    });

    // 监听关闭事件
    socket.addEventListener('close', function(event) {
      console.log('连接已关闭:', event);
    });

    // 监听错误事件
    socket.addEventListener('error', function(event) {
      console.error('发生错误:', event);
    });

    // 获取DOM元素
    const messageInput = document.getElementById('messageInput');
    const sendButton = document.getElementById('sendButton');

    // 绑定事件处理函数
    sendButton.addEventListener('click', function() {
      const message = messageInput.value;
      socket.send(message);
      messageInput.value = '';
    });
  </script>
</body>
</html>

4. 运行服务器和客户端

启动Node.js服务器：

node server.js

然后在浏览器中打开客户端HTML文件（例如index.html），你应该能够看到实时通信的效果。

总结

以上示例展示了如何使用Node.js和ws库实现基本的实时通信。你可以根据需要扩展这个示例，例如添加身份验证、消息格式化、房间管理等功能。

如何用Node.js实现实时通信

1. 安装依赖

2. 创建服务器

3. 创建客户端

4. 运行服务器和客户端

总结

相关阅读