WebSocket 是一种在客户端和服务器之间进行双向通信的协议。在 Web 应用程序中，它可以用于实时通信和数据推送。在本文中，我们将探讨如何在 Node.js 中使用 WebSocket 升级和保持连接。
WebSocket 协议
WebSocket 协议基于 HTTP 协议，它允许客户端和服务器之间建立持久连接，从而实现实时通信。与 HTTP 不同的是，WebSocket 是一种双向通信协议，它允许客户端和服务器之间相互发送消息。此外，WebSocket 还具有以下优点：
- 低延迟
- 减少网络流量
- 降低服务器和客户端的负载
Node.js 中的 WebSocket
在 Node.js 中，我们可以使用 ws 模块来实现 WebSocket。下面是一个简单的示例：
-- -------------------- ---- -------
----- --------- - --------------
----- --- - --- ------------------ ----- ---- ---
-------------------- -------- -------------- -
---------------- -------- ----------------- -
---------------------- ---- ---------
---
--------------- ---------
---在上面的示例中，我们创建了一个 WebSocket 服务器，并在客户端连接时发送了一条欢迎消息。
升级到 WebSocket
在 HTTP 协议中，客户端和服务器之间的通信始终是由客户端发起的。为了建立 WebSocket 连接，客户端需要向服务器发送一个特殊的 HTTP 请求，请求将其连接升级到 WebSocket。在 Node.js 中，我们可以使用 upgrade 事件来处理这个请求。下面是一个示例：
-- -------------------- ---- -------
----- --------- - --------------
----- ---- - ----------------
----- ------ - ----------------------- ---- -- -
------------------ - --------------- ------------ ---
--------------- ---------
---
----- --- - --- ------------------ --------- ---- ---
-------------------- -------- ---------------- ------- ----- -
-------------------------- ------- ----- -------- -------- -
---------------------- --- ---------
---
---
-------------------- -------- -------------- -
---------------- -------- ----------------- -
---------------------- ---- ---------
---
--------------- ---------
---
--------------------在上面的示例中，我们创建了一个 HTTP 服务器，并在客户端连接时将其连接升级到 WebSocket。在 upgrade 事件处理程序中，我们使用 handleUpgrade 方法将连接升级到 WebSocket。一旦连接升级成功，我们就可以像之前一样处理 WebSocket 连接了。
保持 WebSocket 连接
WebSocket 连接是一种持久连接，因此我们需要确保连接一直保持活动状态。在 Node.js 中，我们可以使用 ping 和 pong 消息来保持连接。下面是一个示例：
-- -------------------- ---- -------
----- --------- - --------------
----- --- - --- ------------------ ----- ---- ---
-------------------- -------- -------------- -
---------- - -----
------------- -------- ----------- -
---------- - -----
---
----- -------- - -------------------- ------ -
-- ----------- --- ------ ------ ---------------
---------- - ------
----------- ------ ------
-- -------
-------------- -------- ------- -
------------------------
---
---在上面的示例中，我们使用 ping 和 pong 消息来保持连接。我们在每个 WebSocket 连接对象上设置一个 isAlive 属性，表示连接是否处于活动状态。在每个 pong 消息中，我们将 isAlive 设置为 true。然后，我们使用 setInterval 定时发送 ping 消息。如果在 30 秒内没有收到 pong 消息，则我们终止连接。
结论
在本文中，我们探讨了如何在 Node.js 中使用 WebSocket 升级和保持连接。WebSocket 是一种非常强大的协议，可以用于实时通信和数据推送。在 Node.js 中，我们可以使用 ws 模块来实现 WebSocket。我们还介绍了如何升级连接到 WebSocket 并保持连接处于活动状态。这些技术可以被应用于许多实际场景中，例如在线游戏、聊天应用程序和实时监控系统。
来源：JavaScript中文网 ，转载请注明来源 https://www.javascriptcn.com/post/675ed74be49b4d07161d8683