认识并理解webSocket

今天逛牛客，看到有大佬分享说前端面试的时候遇到了关于webSocket的问题，一看自己都没见过这个知识点，赶紧学习一下，在此记录！

WebSocket 是一种网络通信协议，提供了全双工通信渠道，即客户端和服务器可以同时发送和接收数据。这与传统的HTTP请求不同，后者是单向的，客户端发起请求，服务器响应请求。WebSocket 允许服务器主动向客户端发送消息，这使得实时通信成为可能，例如在线聊天应用、实时游戏、股票行情更新等场景。

WebSocket 的基本概念

连接建立：客户端通过发送一个HTTP请求来发起WebSocket连接，这个请求中包含特定的头部，表明这是一个WebSocket握手请求。
握手：服务器接收到请求后，如果支持WebSocket，则响应一个HTTP响应，完成握手过程，建立WebSocket连接。
数据传输：一旦连接建立，客户端和服务器就可以通过这个连接发送数据。数据可以是文本或二进制格式。
连接关闭：任何一方都可以关闭WebSocket连接。

WebSocket 的使用步骤

创建WebSocket实例：在客户端，首先需要创建一个WebSocket实例，指定服务器的URL。
```
const ws = new WebSocket('ws://example.com/socket');
```

处理连接事件：当WebSocket连接建立时，会触发open事件。

ws.onopen = function(event) {console.log('WebSocket connection opened:', event);
};

发送数据：使用send方法向服务器发送数据。
```
ws.send('Hello Server!');
```

接收数据：服务器发送的数据可以通过message事件接收。

ws.onmessage = function(event) {console.log('Message from server:', event.data);
};

处理错误和关闭连接：WebSocket连接可能会遇到错误，或者需要主动关闭。

ws.onerror = function(error) {console.error('WebSocket Error:', error);
};ws.onclose = function(event) {console.log('WebSocket connection closed:', event);
};// 可以主动关闭连接
ws.close();

使用nodejs实现一个简单的在线聊天demo。

客户端代码

<!DOCTYPE html>
<html lang="en">
<head><meta charset="UTF-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><title>Document</title>
</head>
<body><!-- 简单的HTML界面 -->
<textarea id="messageInput" placeholder="Type a message..."></textarea>
<button id="sendButton">Send</button>
<div id="messages"></div><script>// 绑定按钮点击事件document.getElementById('sendButton').addEventListener('click', sendMessage);// 创建WebSocket连接const ws = new WebSocket('ws://localhost:8080');const messageInput = document.getElementById('messageInput');const messagesDiv = document.getElementById('messages');const sendButton = document.getElementById('sendButton');// 连接打开时触发ws.onopen = function(event) {console.log('Connected to the server');// 可以在这里禁用或启用按钮等sendButton.disabled = false;};// 接收到消息时触发ws.onmessage = function(event) {// 将接收到的消息添加到消息显示区域const messageElement = document.createElement('div');messageElement.textContent = event.data;messagesDiv.appendChild(messageElement);// 滚动到消息区域底部messagesDiv.scrollTop = messagesDiv.scrollHeight;};// 发送消息的函数function sendMessage() {if (ws.readyState === WebSocket.OPEN) {const message = messageInput.value;if (message) { // 确保消息不为空ws.send(message);messageInput.value = ''; // 清空输入框}} else {console.error('WebSocket is not connected. Cannot send message.');}}// 客户端连接错误时触发ws.onerror = function(error) {console.error('WebSocket error observed:', error);// 可以在这里显示错误信息或禁用发送按钮sendButton.disabled = true;};// 客户端关闭连接时触发ws.onclose = function(event) {console.log('WebSocket connection closed:', event);// 可以在这里禁用发送按钮或显示断开连接的信息sendButton.disabled = true;};
</script>
</body>
</html>

服务端代码

（注意要先使用npm install ws命令安装需要的库）

const WebSocket = require('ws');
const wss = new WebSocket.Server({ port: 8080 });// 存储所有客户端的集合
const clients = new Set();wss.on('connection', function connection(ws) {console.log('Client connected');// 将新的客户端WebSocket对象添加到集合中clients.add(ws);ws.on('message', function incoming(message) {console.log('received: %s', message);});ws.on('close', function() {console.log('Client disconnected');// 从集合中移除客户端clients.delete(ws);});// 可以在这里发送欢迎消息给新连接的客户端ws.send('Welcome to the chat!');
});// 假设我们有一个函数，用来向所有客户端广播消息
function broadcastMessage(message) {clients.forEach(client => {if (client.readyState === WebSocket.OPEN) {client.send(message);}});
}// 示例：每隔5秒向所有客户端发送当前时间
setInterval(() => {const currentTime = new Date().toLocaleTimeString();broadcastMessage(`Current time: ${currentTime}`);
}, 5000);