在前端开发中，实时通讯是一个十分重要的功能，多人在线聊天室是实时通讯中最经典的应用场景之一。而 WebSocket 提供了一种全新的实现实时通讯的方式，它通过建立长连接，实现了双向的即时通讯能力。在 Node.js 中，通过使用 WebSocket 可以较为轻松地实现多人在线聊天室。
WebSocket 的基本使用
在使用之前，需要安装 WebSocket 的依赖：
npm install ws
然后，我们可以在 Node.js 中创建 WebSocket 服务，如下所示：
const WebSocket = require('ws');
const wss = new WebSocket.Server({ port: 3000 });
wss.on('connection', (ws) => {
console.log('client connected');
});这段代码中，我们使用 Node.js 内置的 ws 模块创建了一个 WebSocket 服务器，并监听了端口 3000。当有客户端连接成功时，WebSocket 服务器将会调用回调函数，可以在回调函数中处理客户端发送的数据。
例如，我们可以给客户端发送消息，代码如下所示：
wss.on('connection', (ws) => {
console.log('client connected');
ws.send('hello, client');
});而客户端在连接到 WebSocket 服务器后，就可以接收到服务端发送的消息。可以参考下面的代码：
const ws = new WebSocket('ws://localhost:3000');
ws.onmessage = (message) => {
console.log('received:', message.data);
};在这个代码片段中，我们创建了一个 WebSocket 连接，并监听了连接上收到的消息。当 WebSocket 服务器向连接发送消息时，客户端将会打印出消息的内容。
实现多人在线聊天室
有了 WebSocket，我们可以相对轻松地实现多人在线聊天室，下面是一份完整的示例代码：
-- -------------------- ---- -------
----- --------- - --------------
----- --- - --- ------------------ ----- ---- ---
----- ----- - --- ------
-------------------- ---- -- -
------------------- ------------
--- ------ - -----
---------------- --------- -- -
----- ---- - --------------------
---------------------- ---------------
------ ----------- -
---- -------
------ - ------------
-- -------------------- -
----------------- --- -------
-
--------------------------
------------------ -
----- --------------
----- ------------------------------------ -- -------
---
------
---- --------
-- ------- -- ------------------ -
-----------------------------
------------------ -
----- ------------
----- ------
---
-
------
---- ----------
-- ------- -- ------------------ -
----- ----------- - -
----- ----------
----- -
--- ------
----- ----------
--
--
------------------ -------------
-
------
--------
-------------------- ------- ----- ---------------
------
-
---
-------------- -- -- -
-- -------- -
-----------------------------
------------------ -
----- ------------
----- ------
---
-
---
---
-------- ------------------ ----- -
-- -------------------- -
----------------- --- ------- ------------
- ---- -
------------------------------ -- -------------------------------
-
-这份代码中，我们建立了一个 rooms 数据结构，用于存储每个聊天室的连接集合。当客户端加入一个聊天室时，我们会将它的连接加入到对应聊天室的连接集合中。离开时则相反，从集合中删除该连接即可。
客户端可以向服务器发送三种类型的消息：
join：加入一个聊天室leave：离开当前聊天室message：发送消息
服务器会根据消息的类型，对集合中所有连接广播消息。例如，当有用户加入聊天室时，服务器会向该聊天室内的所有连接广播一个 user-joined 类型的消息，消息的 data 属性则为该聊天室内所有用户的 ID。
{
"type": "user-joined",
"data": ["id1", "id2", "id3"]
}当有用户退出聊天室时，服务端会广播一个 user-left 类型的消息，消息的 data 属性为删除连接的用户的 ID。
{
"type": "user-left",
"data": "id1"
}所有的聊天消息都会被以 message 类型广播出去，消息的 data 属性为该消息的发送者 ID 与 消息的文本内容。
{
"type": "message",
"data": {
"id": "id1",
"text": "hello, world!"
}
}总结
Node.js 中的 WebSocket 为多人在线聊天室等实时通讯应用提供了十分便利的开发方式。通过建立长连接，WebSocket 的双向通讯能力使得实时通讯成为了可能。在实际应用中，还需要考虑性能等方面的优化，以及如何处理异常情况等。但总的来说，使用 WebSocket 建立多人在线聊天室的实现相对轻松，并且在实际应用中也有广泛的使用价值。
来源：JavaScript中文网 ，转载请注明来源 https://www.javascriptcn.com/post/64fc4ceef6b2d6eab321c384