Websocket简介[维基百科]
WebSocket is a computer communications protocol, providing full-duplex communication channels over a single TCP connection. The WebSocket protocol was standardized by the IETF as RFC 6455 in 2011, and the WebSocket API in Web IDL is being standardized by the W3C.
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']);
});
});
