Node.js中使用socket创建私聊和公聊聊天室
作者：想兵
先给大家展示效果图：
在上篇文章给大家介绍使用Angular和Nodejs、socket.io搭建聊天室及多人聊天室，本文继续介绍Node.js中使用socket创建私聊和公聊聊天室，具体详情请看下文吧。
nodejs的应用中，关于socket应该是比较出彩的了，socket.io在github上有几万人的star，它的成功应该是不输于express的，为了方便了解整个socket.io的使用.
例子请点击http://chat.lovewebgames.com/
源码下载https://github.com/tianxiangbing/chat
由于本人太穷，所以服务器和数据库都是使用的国外免费的，访问速度上可以会稍慢。
先说下我对socket.io的理解，websocket更像是开启了一个端口服务，来监视过往的通讯。所以我们可以依赖于当前站点80端口启socket服务，也可以放于其他端口上，比如：
require('socket.io').listen(3000);
这样就是监视3000端口了，由于我用的免费服务器，没有权限打开其他端口，所以，我还是使用80了，由于80已经被express使用了，所以我只好在express使用的时候传进来了。
var server = http.createServer(app);
var socket = require(‘./socket/msg')(server);
然后 我在msg.js里是这样写的
var db = require('../db/mysql');
var sio = require('socket.io');
var IO = function(server) {
var io = sio.listen(server)
这样就和谐了，db是创建mysql连接的方法，不在本节内容里，略。
在socket.io里是这样的，首先创建一个io通道的连接，然后监视里面的socket的事件，nodejs是事件驱动嘛。代码如下：
io.on('connection', function(socket) {
console.log('a user connected.');
socket.on('disconnect', function() {
console.log('user disconnected.');
});
})
这时只要有用户连接上，就会进入connection中了，然后它的参数是个socket,如果是公聊，我们可以直接用
io.emit('chat message', {});
这种形式了。但我们这里是私聊，所以我们要临时的把这个socket对象保存在全局里，供与你私聊的对象使用找到你的socket,很绕口，其实这里的私聊，不算完全的点对点，它还是经过了服务器的，消息传给服务器，服务器再找到你要传达给的那个人的socket对象，发给他。这就是整个的过程了。这里我使用的是一个类数组对象来存储的.
var users = {},
usocket = {};
socket.on('user join', function(data) {
users[username] = username;
usocket[username] = socket;
})
由于我这里需要用户名登录，所以我就把用户名作为了唯一的标识（这只是一个例子，不要跟我谈用户名重复的情况），这里用类数组的形式的好处就是我不用循环也能够很快的找到它。再我给A发送私聊时，我会先在这个uscoket里面找到它，然后调用它的emit。
function sendUserMsg(data) {
if (data.to in usocket) {
console.log('================')
console.log('to' + data.to, data);
usocket[data.to].emit('to' + data.to, data);
usocket[data.user].emit('to' + data.user, data);
console.log('================')
}
}
这里我emit了两次的原因是，我发给对方消息的同时，我自己也要收到这个消息，然后把它显示出来，为什么这样？其一，接口统一了，聊天里的内容全是服务器过来的，其二，证明我发送成功了。
然后我在客户端监听时，也用我自己的用户名起了一个to+用户名的事件监听。
socket.on('to' + user, function(data) {
//console.log(data);
formatMsg(data);
})
这样，不管是我发的消息，还是我收到消息，都会进入这个事件了。最后，在用户离开的时候别忘记delete掉这个对象。
socket.on('disconnect', function() {
console.log('disconnect')
if (username) {
counter--;
delete users[username];
delete usocket[username];
if (home.name == username) {
homeLeave(username);
}
sendmsg({
type: 0,
msg: "用户<b>" + username + "</b>离开聊天室",
counter: counter,
users: users
})
}
});
好了，这样就大功告成了。