目的
通过nodejs + ws (Websocket) 实现点对点发送及群聊发送
项目搭建直接跳过,这里列举一下用到的插件
yarn add ws
WebSocket
话不多说,直接上码
对比上一篇文章 << Nodejs+MongoDB 搭建服务( demo ) >> server.js 主要是增加了ws的引用,
webSocket 点对点也好,群聊也好,主要是用到一个方法 clients 它的作用就是存储了连接,
我们可以通过对每一个连接增加标识,然后根据接收到的信息进行解析,遍历然后发送正确的响应。
先解析下代码(小声BB), 下面会有HTML页面
nbs.on("connection", (ws, req) => { ws.on("message", message => { // 这里解析数据 const result = JSON.parse(message); // 是否是登陆,登陆后绑定id if (result.login) { ws.socketIdxos = result.id; } else { // 这里只做一个小的场景,不是登陆就是发送信息 // 根据存储的id 是否是要 发送的对象 来单独响应 nbs.clients.forEach(s => { if (s.socketIdxos == result.to && s.readyState == 1) { s.send(JSON.stringify({ data: result.mes })); } }); } }); ws.on("close", message => { console.log(ws.socketIdxos,"退出连接"); }); });
HTML
这里我分别建了三个页面其中两个模拟用户,一个模拟群聊
这是BBB用户,另一个页面是 AAA 只是改了下信息而已,就不都贴上来了,很简单
<!DOCTYPE html> <html> <head> <meta charset="utf-8"> <title>BBB111</title> <script> const ws = new WebSocket("ws://localhost:8181"); ws.onopen = function (e) { ws.send(JSON.stringify({ "id":"BBB111", "login":true, })); } function sendMessage() { ws.send(JSON.stringify({ "id":"BBB111", // 当前ID "login":false, // 是否登陆 "to":"AAA111", // 发送对象 "mes":"我是BBB111,这是发送给AAA111的消息" // 发送内容 })); } function sendYUYUYU() { ws.send(JSON.stringify({ "id":"BBB111", "login":false, "to":"YUYUYU", "mes":"我是BBB111,这是到群里的消息" })); } ws.onmessage = function (e) { console.log(e,"eeee-"); } </script> </head> <body > <button onclick="sendMessage();"> 发送 </button> <button onclick="sendYUYUYU();"> 群聊 </button> </body> </html>
下面的是只用来接收的群页面,更简单
<!DOCTYPE html> <html> <head> <meta charset="utf-8"> <title>群聊</title> <script> const ws = new WebSocket("ws://localhost:8181"); ws.onopen = function (e) { // 这里主要作用就是注册下群成立了 ...... ws.send(JSON.stringify({ "id":"YUYUYU", "login":true, })); } ws.onmessage = function (e) { console.log(e,"eeee-"); } </script> </head> <body > </body> </html>
展示