在 Node.js 中创建 WebSocket 服务器的步骤是什么？

在 Node.js 中实现 WebSocket 服务器是一个简单的过程，这要归功于各种 WebSocket 库的可用性。在本文中，我们将使用流行的库“ws”在 Node.js 中创建 WebSocket 服务器。

安装 “ws” 库

首先，我们需要安装 “ws” 库。您可以使用 npm（节点包管理器）通过在终端或命令提示符下运行以下命令来安装它：

npm install ws

创建 WebSocket 服务器

安装库后，我们可以创建WebSocket服务器。我们首先在 Node.js 文件中导入 “ws” 库：

const WebSocket = require('ws');

接下来，我们通过传递 HTTP 服务器作为参数来创建一个新的 WebSocket 服务器：

const server = new WebSocket.Server({ port: 8080 });

在上面的代码中，我们在端口 8080 上创建了一个新的 WebSocket 服务器。您可以将端口号更改为要使用的任何有效端口号。

处理 WebSocket 连接

创建服务器后，我们需要处理传入的 WebSocket 连接。我们使用“连接”事件来执行此操作：

server.on('connection', (socket) => {  console.log('Client connected');  socket.on('message', (message) => {    console.log(`Received message: ${message}`);  });  socket.send('Welcome to the WebSocket server!');});

在上面的代码中，我们侦听“连接”事件，并在新客户端连接时记录一条消息。我们还侦听“消息”事件并记录收到的消息。最后，我们向客户端发送一条消息以确认连接。

广播消息

我们可以使用 for 循环向所有连接的客户端广播消息：

server.on('connection', (socket) => {  // ...  socket.on('message', (message) => {    console.log(`Received message: ${message}`);    server.clients.forEach((client) => {      if (client.readyState === WebSocket.OPEN) {        client.send(message);      }    });  });  // ...});

在上面的代码中，我们遍历所有客户端并将收到的消息发送到当前连接的每个客户端。

关闭连接

我们可以处理“close”事件以在客户端断开连接时关闭连接：

server.on('connection', (socket) => {  // ...  socket.on('close', () => {    console.log('Client disconnected');  });  // ...});

在上面的代码中，我们侦听“close”事件，并在客户端断开连接时记录一条消息。

错误处理

在实现 WebSocket 服务器时处理错误非常重要。我们可以使用“error”事件来处理错误：

server.on('connection', (socket) => {  // ...  socket.on('error', (error) => {    console.error(`WebSocket error: ${error}`);  });  // ...});

在上面的代码中，我们侦听“error”事件并记录发生的任何错误。

1. 完整代码

以下是我们的 WebSocket 服务器的完整代码：

const WebSocket = require('ws');const server = new WebSocket.Server({ port: 8080 });server.on('connection', (socket) => {  console.log('Client connected');  socket.on('message', (message) => {    console.log(`Received message: ${message}`);    server.clients.forEach((client) => {      if (client.readyState === WebSocket.OPEN) {        client.send(message);      }    });  });  socket.send('Welcome to the WebSocket server!');  socket.on('close', () => {    console.log('Client disconnected');  });  socket.on('error', (error) => {    console.error(`WebSocket error: ${error}`);  });});console.log('WebSocket server is running on port 8080');

在上面的代码中，我们导入“ws”库，创建一个新的WebSocket服务器，处理传入连接，广播消息，关闭连接和处理错误。

测试服务器

为了测试我们的 WebSocket 服务器，我们可以使用 WebSocket 客户端，例如基于浏览器的“简单 WebSocket 客户端”。只需打开客户端并连接到“ws://localhost:8080”。您应该看到“欢迎使用 WebSocket 服务器！”消息，并且能够发送和接收消息。

结论

在本文中，我们已经了解了如何使用“ws”库在Node.js中创建WebSocket服务器。我们已经介绍了处理连接、广播消息、关闭连接和错误处理的基础知识。有了这些知识，您现在可以在 Node.js 中构建更高级的 WebSocket 应用程序。