当前位置： 移动技术网 > IT编程>开发语言>JavaScript > 基于Node.js的WebSocket通信实现

基于Node.js的WebSocket通信实现

2019年03月22日  | 移动技术网IT编程  | 我要评论

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 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']);
  });
});

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持移动技术网。

您可能感兴趣的文章:

如对本文有疑问， 点击进行留言回复！！

相关文章:

验证码：
最近更新的文章
大家感兴趣的文章
移动技术网