Node.js 实现 WebSocket

WebSocket 在前端的应用场景

WebSocket 是一种在客户端和服务器之间建立持续连接的协议，它提供了一种实时、双向的通信方式。在前端方面，它的主要应用场景如下：

实时通信： WebSocket 可以与服务器建立持久的连接，通过它可以进行实时的消息传递，如聊天室、游戏等。
实时数据更新： 对于需要实时更新数据的场景，如股票行情、快递物流信息等，通过 WebSocket 能够快速地将数据推送给客户端，实现实时更新。
实时通知： 在需要告知用户的情况下，无需用户发出请求，服务器可以直接通过 WebSocket 向客户端推送通知，例如新邮件、提醒等。

Nodejs 实现 WebSocket

前端用的最多的后端语言就是 Nodejs，今天带大家使用 Nodejs 实现一个 WebSocket 服务，开始！

安装插件

使用 nodejs-websocket 插件，开发人员可以在 Node.js 中创建 WebSocket 服务器和客户端。

我们需要使用 npm 进行安装：

npm init
npm i nodejs-websocket
npm i nodemon -g

我们来介绍一下这个库提供的 API：

createServer：创建一个 WebSocket 服务
on：监听事件
sendText：给订阅端发送信息

代码编写

我们需要创建一个 index.js 并写入以下的代码：

 // 引入nodejs-websocket
const ws = require('nodejs-websocket');
// 定义监听的host地址跟port端口
const HOST = '127.0.0.1'
const PORT = 3001;
// 创建ws服务
const service = ws.createServer((conn) => {
  conn.on('text', (message) => {
    // 当收到消息的时候就开始定时推送
    console.log('message', message);
    setInterval(() => {
      // 随机推送message里的内容
      conn.sendText(`Now is ${new Date().getTime()}`);
    }, 1 * 1000);
  });
}).listen(PORT, HOST, () => {
  console.log('service---success');
});

启动服务

最后使用命令进行启动：

node index.js

我们可以看到这样的界面，说明运行成功：

mac@MacBook-Pro % nodemon index.js
[nodemon] 2.0.12
[nodemon] to restart at any time, enter `rs`
[nodemon] watching path(s): *.*
[nodemon] watching extensions: js,mjs,json
[nodemon] starting `node index.js`
service---success

前端页面连接 WebSocket

页面编写

我们需要创建一个 index.html，并写入以下代码。

<!DOCTYPE html>
<html lang="en">
<head>
  <meta charset="UTF-8">
  <meta http-equiv="X-UA-Compatible" content="IE=edge">
  <meta name="viewport" content="width=device-width, initial-scale=1.0">
  <title>Document</title>
  <script>
    window.onload = () => {
        if ('WebSocket' in window) {
            // 创建websocket连接
            let ws = new WebSocket('ws://127.0.0.1:3001/websocket');
            // 成功连接的时候推送一条消息，此时服务端就可以开始推送数据了
            ws.onopen = () => {
                console.log('websocket success---');
                ws.send('success');
            }
            ws.onmessage = (message) => {
                let data = message.data;
                console.log('get websocket message---', data);
            }
            ws.onerror = () => {
                console.error('websocket fail');
            }
        } else {
            console.error('dont support websocket');
        };
    };
</script>

</head>
<body>
  
</body>
</html>