面试题：Node.js中WebSocket即时通讯功能实现的基本步骤

Question

请简述在Node.js中使用WebSocket实现即时通讯功能的基本步骤，包括引入相关模块、创建WebSocket服务器、处理连接和消息收发等环节。

Answer 1

1. 引入相关模块： 在Node.js中，使用ws模块来实现WebSocket功能。首先需要安装ws模块，如果是在项目目录下，通过npm install ws进行安装。然后在代码中引入该模块：

const WebSocket = require('ws');

2. 创建WebSocket服务器： 使用WebSocket.Server来创建一个WebSocket服务器实例，并指定服务器监听的端口。例如：

const wss = new WebSocket.Server({ port: 8080 });

3. 处理连接： 通过监听connection事件来处理客户端的连接。当有客户端连接到服务器时，会触发该事件，回调函数中的ws参数代表与该客户端建立的WebSocket连接。

wss.on('connection', function connection(ws) {
    console.log('Client connected');
    // 这里可以进行一些连接建立后的初始化操作
});

4. 处理消息收发：
   • 接收消息：监听连接ws的message事件，当服务器接收到客户端发送的消息时会触发该事件，message参数即为接收到的消息内容。

ws.on('message', function incoming(message) {
    console.log('Received: %s', message);
    // 可以在这里对消息进行处理，比如解析JSON等
});

- **发送消息**：使用连接`ws`的`send`方法向客户端发送消息。例如：

ws.send('Hello, client!');

5. 处理连接关闭： 监听连接ws的close事件，当客户端与服务器的连接关闭时会触发该事件。可以在回调函数中进行一些清理操作等。

ws.on('close', function close() {
    console.log('Connection closed');
});