在现代 Web 应用中，WebSocket 成为了最常见的实时通信方式。然而，由于网络的不稳定性和客户端设备的多样性，WebSocket 连接可能会在任何时候断开。为了解决这个问题，需要实现一个 WebSocket 心跳检测机制，以确保连接的持久性和稳定性。在本文中，我们将介绍如何使用 Fastify 框架实现 WebSocket 心跳检测机制的方法，并提供示例代码。
Fastify 框架简介
Fastify 是一个最近流行起来的 Node.js Web 框架，它具有以下特点：
- 高效。Fastify 的性能比 Express 快 2-5 倍。
- 强大。Fastify 提供了许多方便的插件和扩展，例如日志记录、验证、路由管理等。
- 简单。Fastify 的 API 简单易用，易于上手和学习。
Fastify 也具有内置的 WebSocket 支持，可以方便地构建一个实时通信应用。
WebSocket 心跳检测
WebSocket 心跳检测是一种保持 WebSocket 连接活跃的机制，它通过定期发送 ping 消息检查连接的状态。如果 WebSocket 连接一段时间内没有收到任何消息，则有可能已经断开，此时需要关闭该连接，避免浪费服务器资源。
实现方法
要实现 WebSocket 心跳检测，可以使用 Fastify 的内置 WebSocket 扩展和 Node.js 的 setInterval 函数。
首先，在服务器端，创建一个建立 WebSocket 连接的函数，启动定时器发送 ping 消息：
-- -------------------- ---- ------- ----- ------- - --------------------- ----- --------- - -------------- -- -- --------- -- ------------------ - ---------- ---- -- ------------ ---- -- - ----- -- - ------------------ -- -------- ----- -------- - -------------- -- - ------------------------ ----- ------ ---- -- ------ -- ------- ---------------- ------ -- - ------------------ --- -- -- --------- ---- -------------- -- -- - ------------------------ --- --- -------------------- ----- -- - -- ----- - ----------------- ---------------- - ------------------- --------- -- ---- ---------- ---展开代码
上述代码中，我们创建了一个 /ws 路由，用于建立 WebSocket 连接。在连接建立后，我们启动了一个定时器，每 5 秒发送一次 ping 消息，以检查连接的状态。同时，我们还监听了客户端发送过来的消息和 WebSocket 关闭事件。
然后，在客户端，创建一个 WebSocket 连接，并监听服务器发送过来的消息和心跳消息：
-- -------------------- ---- ------- ----- -- - --- ------------------------------------ --------- - ------- -- - ---------------------- ---------- ---------- -- ------------ - ------- -- - ----- ---- - ----------------------- -- ---------- --- ------- - ------------------------ ----- ------ ---- - ------------------ -- ---------- - ------- -- - ---------------------- ---------- ---------- --展开代码
上述代码中，我们创建了一个 WebSocket 连接，当连接打开后，我们监听了服务器发送过来的消息和心跳消息。如果收到的是心跳消息，则返回一个 pong 消息，以告诉服务器该连接仍然是活跃的。
示例代码
完整的示例代码可以从 GitHub 上下载：
https://github.com/MrLishijie/fastify-websocket-heartbeat
指导意义
本文介绍了如何使用 Fastify 框架实现 WebSocket 心跳检测机制，以确保 WebSocket 连接的持久性和稳定性。该方法可以适用于大多数实时通信应用，可以防止 WebSocket 连接在长时间没有消息交换时被自动断开。通过这篇文章的学习，读者可以深入了解 WebSocket 技术和 Fastify 框架的使用，从而在实际项目中运用这些知识。
来源：JavaScript中文网 ，转载请注明来源 https://www.javascriptcn.com/post/67c2710a314edc2684bb5785
