欢迎来到 黑吧安全网 聚焦网络安全前沿资讯,精华内容,交流技术心得!

怎么样基于Note.js的websocket实现 通信功能

来源:本站整理 作者:佚名 时间:2017-06-30 TAG: 我要投稿

node的依赖包
node中实现Websocket的依赖包有很多,websocket、ws均可,本文选取ws来实现,首先安装依赖
 npm install ws
聊天室实例
假如A,B,C,D用户均通过客户端连接到Websocket服务,其中每个人发的消息都需要将其通过Websocket转发给其他人,此场景类似于服务端将A的消息广播给组内其他用户。
服务端实现
首先来看服务端程序,具体的工作流程分以下几步:
创建一个WebSocketServer的服务,同时监听8080端口的连接请求。
每当有新的客户端连接该WebSocket成功时,便将该连接push到连接池的数组中。
监听message事件,当该事件发生时,遍历连接池,以连接为单位将该消息转发到对应的客户端
监听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 client
var userConnectionMap = new HashMap();
var connectNum = 0;

// connection
webSocketServer.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']);
 });
});
以上就是本文的全部内容,希望对大家的学习有所帮助

【声明】:黑吧安全网(http://www.myhack58.com)登载此文出于传递更多信息之目的,并不代表本站赞同其观点和对其真实性负责,仅适于网络安全技术爱好者学习研究使用,学习中请遵循国家相关法律法规。如有问题请联系我们,联系邮箱admin@myhack58.com,我们会在最短的时间内进行处理。
  • 最新更新
    • 相关阅读
      • 本类热门
        • 最近下载