标签：
原文地址：http://www.moye.me/2015/01/02/node_socket-io/
最近听到这么一个问题：Socket.IO 怎么实现私聊？换个提法：怎么定位到人（端），或者说怎么标识到连接，而不是依赖每个连接的socket.id。好问题。
在 Socket.IO Real-Time Web Application Development 的指引下，形成了如下思路：
socket.request.headers.cookie，从这个cookie中解析出session_id，将socket 连接与 Web框架的context中的session_id 对应上[{name, session_id, socket} , ...]有了思路，就可以动手实践了：
ES6 的生成器太好用，做 Node Web 就从 koa 开始吧。那么，我的 package.json 看起来就会有这些依赖的库：
"co": "^4.0", "koa": "^0.14.0", "koa-mount": "*", "koa-ejs": "*", "koa-static": "*", "koa-router": "*", "koa-session": "*", "co-body": "*"
思路中提到的用户列表，就是一个简单的数组：[{name, session_id, socket} , ...]，围绕它的操作也特别简单（socketHandler：
//暴露给Web的接口 module.exports.addUser = addUser; module.exports.otherUsers = otherUsers; var users = []; function findInUsers(session_id) {//通过session_id查找 var index = -1; for (var j = 0, len = users.length; j < len; j++) { if (users[j].session_id === session_id) index = j; } return index; } function addUser(name, session_id) { var index = findInUsers(session_id); if (index === -1) //不存在则重新添加 users.push({name: name, session_id: session_id, socket: null}); else { //只更新昵称 if (users[index].name !== name) users[index].name = name; } } function setUserSocket(session_id, socket){//更新用户socket var index = findInUsers(session_id); if (index !== -1){ users[index].socket = socket; } } function findUser(session_id) { var index = findInUsers(session_id); return index > -1 ? users[index] : null; } function otherUsers(session_id){//其他人 var results = []; for (var j = 0, len = users.length; j < len; j++) { if (users[j].session_id !== session_id) results.push({session_id: users[j].session_id, name: users[j].name}); } return results; }
koa-session 这个库提供了 session 存储功能，它的使用非常简单：
var koa = require(‘koa‘); var session = require(‘koa-session‘); var app = koa(); app.keys = [config.SECRET]; app.use(session(app));
此外，koa-session会在Web request的cookie中会附上一个 koa:sess的session_id 标识串，那么，在 socket.io 的事件侦听中，我们可以这么用它：
io.on(‘connection‘, function (socket) { var sessionId = getSessionId(socket.request.headers.cookie, ‘koa:sess‘); if(sessionId){ setUserSocket(sessionId, socket); } }); function getSessionId(cookieString, cookieName) { var matches = new RegExp(cookieName + ‘=([^;]+);‘, ‘gmi‘).exec(cookieString); return matches[1] ? matches[1] : null; }
所谓的登录，就是让用户输入一个昵称，将它与session_id对应上，并存储到前述用户数组中。假设我们的路由路径为 /chat，登录action路径为/chat/login，那么这个路由看起来是这样：
var Router = require(‘koa-router‘), router = new Router(); var parse = require(‘co-body‘); var socketHandler = require(‘../../middlewares/socketHandler‘); // GET /chat router.get(‘/‘, function *() { var session_id = this.cookies.get(‘koa:sess‘); var name = this.session.name; if(session_id && name) {//添加到用户列表 socketHandler.addUser(name, session_id); yield this.render(‘../www/views/chat‘); //使用ejs } else { this.redirect(‘/chat/login‘); } }); // GET /chat/login 使用ejs模板 router.get(‘/login‘, function*(){ yield this.render(‘../www/views/login‘) }); // POST /chat/login 接收form提交: <input name=‘name‘> router.post(‘/login‘, function*(){ var body = yield parse(this); this.session.name = body.name || ‘guest‘; this.redirect(‘/chat‘) }); module.exports = router;
io.on(‘connection‘, function (socket) { socket.on(‘broadcast‘, function (data) { //广播 var fromUser = findUser(sessionId); if(fromUser) { socket.broadcast.emit(‘broadcast‘, { name: fromUser.name, msg: data.msg }); } }); socket.on(‘private‘, function (data) { //私聊 {to_session_id, msg} var fromUser = findUser(sessionId); if(fromUser) { var toUser = findUser(data.to_session_id); if (toUser) toUser.socket.emit(‘private‘, { name: fromUser.name, msg: data.msg }); } }); });
在连接到服务端后，客户端会定时拉取其他人的列表：
//定时获取其他人列表 function updateOthers() { $.post(‘/chat/others‘, function (others) { //...若干丑陋的UI DOM操作代码 setTimeout(updateOthers, 1000); }); } setTimeout(updateOthers, 1000);
对应的，服务端会有一个这样的接口：
// POST /chat/others 其他人列表 router.post(‘/others‘, function*(){ var session_id = this.cookies.get(‘koa:sess‘); var name = this.session.name; if(session_id && name) { this.type = ‘application/json‘; this.body = socketHandler.otherUsers(session_id); } else { this.status = 404; } });
在三个不同的浏览器中跑起来，宛如上世纪90年代火得不行的聊天室 :)
完整的源码放在我的Github上：https://github.com/rockdragon/socketchat，想让它跑起来，你需要把 Node 升到 0.11.14（因为用到了 Co V4 ），当然，README.MD里有详细的设置说明。
更多文章请移步我的blog新地址： http://www.moye.me/
标签：
原文地址：http://www.cnblogs.com/moye/p/node_socket-io.html
