通过 Fastify 实现 WebSocket 服务

简介

WebSocket 是一种支持双向通信的协议，它能够在客户端和服务器之间建立一个持久化的连接，以实现实时数据的传输。在前端开发中，我们经常需要使用 WebSocket 来实现即时通讯、实时更新等功能。本文将介绍如何通过 Fastify 框架来实现 WebSocket 服务。

Fastify 简介

Fastify 是一个快速、低开销且可扩展的 Node.js Web 框架。它的设计目标是提供最佳性能，同时保持易于开发和维护。Fastify 提供了许多有用的功能，如请求验证、路由、插件等，使得它成为一个非常流行的 Web 框架。

WebSocket 实现

在 Fastify 中实现 WebSocket 服务非常简单。我们可以使用 fastify-websocket 库来实现 WebSocket 服务。

安装 fastify-websocket

安装 fastify-websocket 可以使用 npm 或 yarn 安装。在终端中输入以下命令：

npm install fastify-websocket

或

yarn add fastify-websocket

实现代码

我们可以通过以下代码来实现 WebSocket 服务：

const fastify = require('fastify')();
const WebSocket = require('fastify-websocket');

fastify.register(WebSocket);

fastify.get('/websocket', { websocket: true }, (connection, req) => {
  connection.socket.on('message', message => {
    connection.socket.send(`You sent ${message}`);
  });
});

fastify.listen(3000, (err, address) => {
  if (err) {
    console.error(err);
    process.exit(1);
  }
  console.log(`Server listening on ${address}`);
});

在上面的代码中，我们首先引入了 fastify 和 fastify-websocket 库。然后我们注册了 fastify-websocket 插件。接着，我们定义了一个路由 '/websocket'，并将 websocket 参数设置为 true，以指示这是一个 WebSocket 路由。在回调函数中，我们监听了 'message' 事件，并将接收到的消息返回给客户端。

测试 WebSocket 服务

我们可以使用 WebSocket 客户端来测试我们的服务。下面是一个简单的 WebSocket 客户端代码：

const WebSocket = require('ws');

const ws = new WebSocket('ws://localhost:3000/websocket');

ws.on('open', () => {
  console.log('Connected to WebSocket server');
  ws.send('Hello, WebSocket server!');
});

ws.on('message', message => {
  console.log(`Received message: ${message}`);
});

在上面的代码中，我们首先引入了 WebSocket 库，并创建了一个 WebSocket 客户端。然后我们监听了 'open' 事件，在连接建立后向服务器发送了一条消息。在 'message' 事件中，我们接收并打印服务器返回的消息。

我们可以在终端中运行上面的代码，如果一切正常，我们将看到以下输出：

Connected to WebSocket server
Received message: You sent Hello, WebSocket server!

总结

通过 Fastify 实现 WebSocket 服务非常简单。我们只需要使用 fastify-websocket 插件，定义一个 WebSocket 路由，并在回调函数中处理消息即可。这个示例代码可以作为一个基础模板，用于实现更复杂的 WebSocket 服务。

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