WebSocket 是一种全双工协议，能够在浏览器和服务器之间创建持久性连接。websocket-connection-stream 是一个开源的 npm 包，提供了一种简单而有效的方法将 WebSocket 连接转换为 Node.js 中的流。本文将介绍如何使用这个包。
安装
可以通过 npm 安装 websocket-connection-stream 包：
npm install websocket-connection-stream --save
基本用法
创建一个 WebSocket 流需要一个 WebSocket 链接。可以使用 ws 库的 WebSocket 类或 nodejs-websocket 库的 WebSocketConnection 类创建 WebSocket 连接，然后将其传递给 websocket-connection-stream 的 createWebSocketStream 函数。
const { createWebSocketStream } = require('websocket-connection-stream');
const WebSocket = require('ws');
const ws = new WebSocket('wss://example.com');
const websocketStream = createWebSocketStream(ws);这里创建了一个到 wss://example.com 的 WebSocket 连接，然后通过 createWebSocketStream 创建了一个 WebSocket 流。
现在，可以像使用其他流一样使用 WebSocket 流进行数据读写：
websocketStream.on('readable', () => {
let chunk;
while ((chunk = websocketStream.read()) !== null) {
console.log(chunk.toString());
}
});
websocketStream.write('foo');在这个例子中，当 websocketStream 可读时，数据将从流中读取，并在控制台上打印出来。然后一个字符串 'foo' 被写入 WebSocket 流，通过服务器发送。
示例代码
接下来是一个稍微复杂一些的示例，演示如何使用 WebSocket 流与服务器进行实时通信。以下代码演示如何使用 Express 和 websocket-connection-stream 将 WebSocket 流作为 HTTP 响应的一部分返回给客户端。
-- -------------------- ---- -------
----- ------- - -------------------
----- - --------------------- - - ---------------------------------------
----- --------- - --------------
----- --- - ----------
----- ------ - ---------------- -- -- -
---------------------- -- ---- ------
---
------------------ ----- ---- -- -
----- -- - --- -------------------------------
----- --------------- - --------------------------
--------------------------
--------------- -- -- -
--------------------------
---
--在上面的代码中，当客户端访问 /stream 路径时，代码将尝试通过 WebSocket 流与 wss://example.com 建立连接。websocketStream 然后被传递给 HTTP 响应，使客户端能够获取实时数据。
请注意，在客户端关闭 HTTP 连接时，websocketStream.destroy() 被调用以关闭流。
结论
websocket-connection-stream 为开发人员提供了一种有效的方式，可以将 WebSocket 连接转换为 Node.js 中的流。通过简单易懂的 API，可以轻松地创建、读取和写入 WebSocket 流。它在实时应用和实时通信的场景下非常有用。
来源：JavaScript中文网 ，转载请注明来源 https://www.javascriptcn.com/post/600671068dd3466f61ffddc7