WebSocket 是一种全双工通信协议，它允许客户端和服务器之间进行实时数据交互。在前端开发中，WebSocket 经常被用来实现实时聊天、实时数据展示等功能。而在 Node.js 环境中，你同样可以使用 WebSocket 来实现服务器端的实时数据交互功能。
WebSocket 原理介绍
在 HTTP 协议中，客户端发送请求给服务器后，服务器处理完请求并返回响应后，连接就会断开。而 WebSocket 协议在建立连接之后，客户端和服务器之间的连接就会一直保持，直到被手动关闭或发生异常断开。
WebSocket 协议建立连接需要通过 HTTP 请求和响应来完成，具体步骤如下：
- 客户端向服务器发出一个 HTTP 请求，请求头中包含一个 Upgrade 字段，该字段的值为 websocket。
- 服务器收到请求后，发送一个 HTTP 响应，响应头中包含一个 Upgrade 字段，该字段的值为 websocket。同时，除了 Upgrade 字段以外的其它请求头和响应头与普通的 HTTP 请求和响应一样。
- 客户端收到响应后，建立起一个 WebSocket 连接，之后客户端和服务器可以互相发送消息。
Node.js 中使用 WebSocket
在 Node.js 中，你可以使用 ws 模块来处理 WebSocket 请求。这个模块可以帮助你快速实现一个 WebSocket 服务器，下面是一个示例代码：
-- -------------------- ---- -------
----- --------- - --------------
----- --- - --- ------------------ ----- ---- ---
-------------------- -------- -------------- -
------------------- ------------
---------------- -------- ----------------- -
--------------------- ------- -- -------------
--------------- -------- -------- -------------
---
-------------- -------- ------- -
------------------- ---------------
---
---在这个示例中，我们使用 WebSocket.Server 类创建了一个 WebSocket 服务器，并通过 wss.on('connection', callback) 监听客户端连接事件。当有客户端连接时，服务器会调用 callback 中的函数处理连接。
在 callback 中，我们可以使用 ws 对象来处理客户端发送的消息和连接关闭事件。在消息处理部分，我们使用 ws.send 方法向客户端发送消息。
除了这个示例之外，ws 模块还提供了其它方法来帮助你处理 WebSocket 请求，比如：
ws.clients：一个 Set 对象，包含所有连接到 WebSocket 服务器的客户端。ws.on('headers', callback)：设置请求头。ws.on('ping', callback)：监听客户端发送的 ping 消息。
总结
WebSocket 协议是一种用于实时数据交互的协议，在 Node.js 中，我们可以使用 ws 模块快速构建一个 WebSocket 服务器。在使用中，我们需要注意一些细节，比如处理客户端发送的消息，处理连接关闭事件等。
使用 WebSocket 可以帮助我们实现实时数据交互的功能，比如实时聊天、实时数据展示等。如果你在项目中需要实现类似的功能，可以尝试使用 WebSocket。
来源：JavaScript中文网 ，转载请注明来源 https://www.javascriptcn.com/post/650118a695b1f8cacdee914a