来源:http://blog.csdn.net/koastal/article/details/53677766
socket.io官网api地址
http://socket.io/docs/server-api/
socket.io官方api以及很多教程都知识介绍了广播消息和在当前会话发送消息,在很多时候我们需要实现向某个指定的在线用户发送消息,也就是所谓的“私聊”。
socket.io服务端基本代码结构
var app = require('http').createServer(handler) var io = require('socket.io')(app); var fs = require('fs');
app.listen(3000); function handler (req, res) { fs.readFile(__dirname + '/index.html', function (err, data) { res.writeHead(200);
res.end(data);
});
}
io.on('connection', function(socket){ console.log('connection is established! '+socket.id);
socket.on('disconnect', function(){ console.log('connection is disconnect!');
console.log(socket.id);
});
});
如上述代码所示,每个在线的用户都对应一个socket对象,可以通过socket.id获取其唯一的id。
网上很多人是将所有的socket存储到一个全局数据中,socket.id为键名,socket对象为键值,然后我们可以再建立一个保存用户名和socket.id映射关系的数组;或者直接将用户名作为键名,socket对象作为键值。
这种方式需要维护一个比较大的全局数组,如果用户量很大时,就会比较慢。
事实上,socket.io在运行时,本身就会维护所有的socket对象,这样才能实现消息的广播。
socket.io版本<1,使用下面的方法,将返回所有在线的socket对象数组
io.sockets.clients();
socket.io版本>1,使用下面的属性,将返回所有在线的socket对象集合
io.sockets.sockets;
我们以最新版1.7.2为例,通过上面的属性,就可以获取所有在线的socket集合,没有必要再去存储一个全局数组了。
nodejs的underscore扩展中的findWhere方法,可以是我们很容易在对象集合中,通过对象的属性值找到该对象并返回。
所以,我们现在只需要维护一个socket.id和用户名映射关系的全局数组即可。
建立websocket连接之后,客户端首先设置一下自己的用户名,服务器端将用户名和socket.id的对应关系保存,当客户端A向B发送私聊信息时,需要带上客户端B的用户名。我们通过B的用户名,解析得到客户端B的socket.id,从socket集合中获取响应的socket,然后再发送消息。
var app = require('http').createServer(handler) var io = require('socket.io')(app); var fs = require('fs'); var _ = require('underscore');
app.listen(3000); var hashName = new Array(); function handler (req, res) { fs.readFile(__dirname + '/index.html', function (err, data) { res.writeHead(200);
res.end(data);
});
}
io.on('connection', function(socket){ console.log('connection is established!');
socket.on('setName',function (data) { var name = data;
hashName[name] = socket.id;
console.log(hashName);
});
socket.on('sayTo',function (data) { var toName = data.to; var toId; if(toId = hashName[toName]){ var toSocket = _.findWhere(io.sockets.sockets,{id:toId});
toSocket.emit('message',data.msg);
}
})
socket.on('disconnect', function(){ console.log('connection is disconnect!');
});
});
还有一种方案更加直接,我们将用户名直接设置为socket的name属性,通过name属性来找到socket,并发送消息。
... io.on('connection', function(socket){
console.log('connection is established!');
socket.on('setName',function (data) {
socket.name = data;
});
socket.on('sayTo',function (data) { var toName = data.to; var toSocket; if(toSocket = _.findWhere(io.sockets.sockets,{name:toName})){
toSocket.emit('message',data.msg);
}
})
socket.on('disconnect', function(){
console.log(socket.name+' connection is disconnect!');
});
});
客户端代码:
index.html
<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>socket.io演示客户端</title> <script src="https://cdn.socket.io/socket.io-1.4.5.js"></script> </head> <body> <h1>socket.io演示客户端</h1> </body> <script> var socket = io.connect('127.0.0.1:3000');
socket.on('message',function (data) { console.log(data);
}); </script> </html>
参考资料:
http://bobcaojin198306.blog.163.com/blog/static/9610418201411315012664/