socket.io一对一通信的实现

socket.io网上有很多例子，但是大多数都是聊天室的，没有找到一对一通信的例子

百度搜索了一下，在oschina里面看到了一篇文章http://www.oschina.net/question/190778_164263?sort=time

给了一点思路，就是每次用户连上socket的时候，保存一个id映射到socket对象的键值对

然后在每次emit新消息的时候，根据客户端传来的id，获取对应的socket对象，用这个对象emit信息，就可以实现一对一通信了

下面贴一下我的关键实现代码，代码基于上次发的文章的代码https://github.com/xyc-cn/socket-angular/tree/xyc_3

1.首先得在socket验证登陆的时候，把user对象传进来，前提是你在req.session.user里面添加了你的user对象

var io= require("socket.io").listen(app.listen(8000));
io.set('authorization', function(handshakeData, accept) {
var cookies = cookie.parse(handshakeData.headers.cookie);
var connectSid = cookies['connect.sid'];
if(connectSid){
var connected = cookieParser.signedCookie(connectSid, 'technode');
if(connected){
sessionStore.get(connected, function (error, session) {
if (error) {
accept(error.message, false)
} else {
handshakeData.headers.sessions  = session;//根据connectSid找session如果session对象不为空，把session对象添加到socket的headers里面
if (session.user) {
accept(null, true)
} else {
accept(null,false)
}
}
})
}else {
accept(null,false)
}
}

});

2.在socket链接的时候，我在客户端emit了init，在服务端保存id映射到socket对象的键值对

var loginUser = new Object();
socket.on('init', function () {
socket.emit('init', messages);
if(socket.handshake.headers.sessions!=null){//如果sessions不为空，取出里面保存的user对象
var user = socket.handshake.headers.sessions.user;
loginUser[user._id] = socket;//建立id映射到socket对象的键值对
UserModel.modify({account:user.account},{status:1},function(err,doc){//这部分是我获取在线用户的代码
UserModel.User.find({status:1},{account:1}, function (err,doc) {
if(doc){
io.sockets.emit('users.init', doc);
}
});
});

}
});

3.新建消息的时候，客户端传来一个belong，告诉服务端，这条消息是属于哪两个人的，这个值必须是唯一的，我是把两个通信的人的ID加起来生成这个值，然后还要传来to告诉服务器这个消息是发给谁的，from告诉服务器这个信息是谁发出来的

socket.on('messages.create', function (message) {
var data ={};
data.content = message.content;
data.account = message.account;
data.belong = message.belong;
var to= message.to;

MessageModel.create(data,function (err,doc) {//新建数据库条目
if(!err){
socket.emit('messages.add',doc,to,message.from);//emit通知自己，在发送方的窗口新增他自己发送的消息
if(loginUser[to]!=null){//根据to值，选择对应的socket对象
loginUser[to].emit('messages.add', doc,to,message.from);//如果找到了这个对象，就emit信息给这个socket对象的链接客户端，传递信息
}
}
});
});

4.客户端读取信息的时候，要传来一个belong值，告诉服务端读取哪些聊天记录

socket.on('messages.read', function (belong) {
MessageModel.getMessage(new Date(),belong,function(err,doc){//根据传来的belong值，获取对应的聊天记录
socket.emit('messages.read', doc);
});
});

5.链接断开的时候，清空数据

socket.on('disconnect', function() {
if(socket.handshake.headers.sessions!=null) {
var user = socket.handshake.headers.sessions.user;
loginUser[user.id]=null;//清空对应的键值对
UserModel.modify({account: user.account}, {status: 0}, function (err, doc) {//用户状态设置为下线
if (doc != null) {
console.log(doc.account + " has disconnect");
}
});
socket.broadcast.emit('users.remove', user);
}
});

整个项目的代码在https://github.com/xyc-cn/socket-angular/tree/xyc_4

有问题或者建议欢迎Q我278255596