前言
随着移动互联网和 Web 技术的发展，实时性越来越重要，特别是在在线聊天、游戏、股票行情等场景下。传统的 HTTP 请求-响应模式对于实时通信的支持比较有限，一般需要轮询或短连接等方法。而 WebSocket 技术的出现，可以轻松地实现双向实时通信，大大提高了应用的实时性和用户体验。
本文将介绍如何在 Node.js 中使用 WebSocket 进行实时通信，包括 WebSocket 的基础知识、如何搭建 WebSocket 服务器、如何进行双向通信以及如何处理异常情况。
WebSocket 简介
WebSocket 是一种基于 TCP 协议的双向通信协议，它的主要特点是可以同时进行双向实时通信，且数据格式比较轻量级，通信的数据帧只包含真正的数据，不包含头部，可以大大减少通信的数据量，提高传输效率。WebSocket 可以用于 Web 和移动场景下的实时通信，可以轻松实现在线聊天、游戏、股票行情等场景。
WebSocket 的使用过程包括握手和数据交换两个阶段。握手阶段是一个 HTTP 请求-响应的过程，客户端发起一次 HTTP 请求，包含特殊的头部信息，服务器进行一系列的协商和校验，如果成功，返回一个 HTTP 响应，表明双方可以建立 WebSocket 连接。握手完成后，就可以进行数据交换了。WebSocket 数据交换使用带有特殊控制信号的数据帧进行，数据帧包含有效载荷 (payload)，用于存储传输数据的内容。数据帧的控制信号可以用于保证通信的可靠性、安全性等。
搭建 WebSocket 服务器
在 Node.js 中，提供了一系列非常好用的 WebSocket 库，可以快速搭建 WebSocket 服务器。
安装 WebSocket 库
首先需要安装 WebSocket 库，在命令行执行以下命令：
npm install ws
创建 WebSocket 服务器
创建 WebSocket 服务器非常简单，只需要几行代码即可实现。
-- -------------------- ---- -------
----- --------- - --------------
----- --- - --- ------------------ ----- ---- ---
-------------------- ------------ -
------------------- ------------
---------------- -- --------- ---------
---------------- ----------------- -
--------------------- ---------- ---------
------------ ----- - - ---------
---
-------------- ---------- -
------------------- ---------------
---
---以上代码创建了一个 WebSocket 服务器，监听 8080 端口。当客户端连接成功后，服务器会打印一条连接信息，并且向客户端发送一条欢迎消息。当客户端发送消息时，服务器会处理消息并且返回一条回复消息。当客户端关闭连接时，服务器会打印一条断开连接的信息。
WebSocket 实现双向通信
WebSocket 可以支持双向通信，客户端和服务器都可以发送和接收消息。
客户端
在浏览器端，可以使用原生的 WebSocket 对象来与服务器进行双向通信。创建 WebSocket 对象时，需要指定连接的地址。
-- -------------------- ---- ------- ----- ------ - --- --------------------------------- ------------- - ---------- - ---------------------- ------------ --------------------- -- ---------------- - --------------- - ----- ------- - ----------- --------------------- ---------- --------- -- -------------- - ---------- - ---------------------- --------------- --
以上代码创建了一个 WebSocket 对象，连接到 localhost 的 8080 端口。当连接成功时，浏览器打印一条连接信息，并向服务器发送一条消息。当服务器接收到消息时，会打印一条接收到消息的信息。当连接关闭时，浏览器会打印一条断开连接的信息。
服务器端
在服务器端，可以使用监听事件的方式来进行 WebSocket 的双向通信。当客户端连接成功时，WebSocket 服务器会触发 connection 事件。在 connection 事件中，可以获取到客户端的 WebSocket 对象。通过 WebSocket 对象的 send 方法，可以向客户端发送消息。当客户端发送消息时，WebSocket 对象会触发 message 事件，在 message 事件中可以获取到客户端发送的消息。通过 WebSocket 对象的 send 方法，可以向客户端发送回复消息。当客户端关闭连接时，WebSocket 对象会触发 close 事件。
处理异常情况
在使用 WebSocket 进行实时通信时，可能会遇到各种异常情况，需要进行处理。
连接失败
在 WebSocket 连接失败的情况下，WebSocket 对象会触发 error 事件。在 error 事件中，可以处理连接失败的逻辑。例如，可以进行重试、提示用户等操作。
const socket = new WebSocket('ws://localhost:8080');
socket.onerror = function(error) {
console.log('websocket error:', error);
};以上代码在 WebSocket 连接失败时，打印一条错误信息。
断开连接
在 WebSocket 断开连接时，WebSocket 对象会触发 close 事件。在 close 事件中，可以进行断开连接的处理。例如，可以重新连接、提示用户等操作。
const socket = new WebSocket('ws://localhost:8080');
socket.onclose = function() {
console.log('websocket disconnected');
};以上代码在 WebSocket 断开连接时，打印一条断开连接的信息。
总结
本文介绍了在 Node.js 中使用 WebSocket 进行实时通信的基础知识、如何搭建 WebSocket 服务器以及如何进行双向通信，同时还介绍了如何处理 WebSocket 连接异常情况。WebSocket 技术相对于传统的 HTTP 请求-响应协议，在实时性和数据传输效率等方面具有明显优势。在实际项目中，可以根据具体场景和需求，选择合适的技术和库来进行 WebSocket 实时通信的开发。
来源：JavaScript中文网 ，转载请注明来源 https://www.javascriptcn.com/post/645ce8b5968c7c53b0f6f4a2