文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

怎么用Node.js实现WebSocket通信

2023-07-04 17:04

关注

本篇内容介绍了“怎么用Node.js实现WebSocket通信”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!

node的依赖包

node中实现Websocket的依赖包有很多,websocket、ws均可,本文选取ws来实现,首先安装依赖

npm install ws

聊天室实例

假如A,B,C,D用户均通过客户端连接到Websocket服务,其中每个人发的消息都需要将其通过Websocket转发给其他人,此场景类似于服务端将A的消息广播给组内其他用户。

服务端实现

首先来看服务端程序,具体的工作流程分以下几步:

  1. 创建一个WebSocketServer的服务,同时监听8080端口的连接请求。

  2. 每当有新的客户端连接该WebSocket成功时,便将该连接push到连接池的数组中。

  3. 监听message事件,当该事件发生时,遍历连接池,以连接为单位将该消息转发到对应的客户端

  4. 监听close事件,当该事件发生时,将该连接移出连接池

服务端代码

var WebSocketServer = require('ws').Server,  wss = new WebSocketServer({port: 8080});// 连接池var clients = [];wss.on('connection', function(ws) {  // 将该连接加入连接池  clients.push(ws);  ws.on('message', function(message) {    // 广播消息    clients.forEach(function(ws1){      if(ws1 !== ws) {        ws1.send(message);      }    })  });  ws.on('close', function(message) {    // 连接关闭时,将其移出连接池    clients = clients.filter(function(ws1){      return ws1 !== ws    })  });});

客户端实现

<html><input type="text" id="text"><input type="button" onclick="sendMessage()" value="online"><script>  var ws = new WebSocket("ws://localhost:8080");  ws.onopen = function (e) {    console.log('Connection to server opened');  }  ws.onmessage = function(event) {     console.log('Client received a message', event);   };   ws.onclose = function (e) {    console.log('connection closed.');  }  function sendMessage() {      ws.send(document.getElementById('text').value);  }</script></html>

如何发现用户?

通过上述的demo可以看到,WebSocket都是基于连接的,也就是说我们知道data是从那个connection发过来,但并不知道使用客户端的是李雷或者韩梅梅,这可如何是好?再想另一种场景,李雷只想给韩梅梅发消息,不想将消息广播给其他客户端,此时我们就需要在Server端能够标识用户身份和连接的对应关系。

于是,需要在客户端连接到WebSocket之后,紧接着再发一次请求,告诉Server我的user_id是多少,Server将此user_id与connection之间的关系存储在hashmap中,至此就建立了user_id与connection的对应关系。当需要发送消息给对应的客户端,从此hashmap中取出对应用户的connection信息,调用其send方法发出消息即可。

依赖包

npm install hashmap

服务端实现

var WebSocketServer = require('ws').Server, webSocketServer = new WebSocketServer({port: 8080});var HashMap = require('hashmap');// record the clientvar userConnectionMap = new HashMap();var connectNum = 0;// connectionwebSocketServer.on('connection', function(ws) {  ++ connectNum;  console.log('A client has connected. current connect num is : ' + connectNum);  ws.on('message', function(message) {    var objMessage = JSON.parse(message);    var strType = objMessage['type'];    switch(strType) {      case 'online' :         userConnectionMap.set(objMessage['from'], ws);        break;      default:        var targetConnection = userConnectionMap.get(objMessage['to']);        if (targetConnection) {          targetConnection.send(message);        }    }  });  ws.on('close', function(message) {    var objMessage = JSON.parse(message);    userConnectionMap.remove(objMessage['from']);  });});

“怎么用Node.js实现WebSocket通信”的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识可以关注编程网网站,小编将为大家输出更多高质量的实用文章!

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯