socket.js

var cookie = require('express/node_modules/cookie');
var Contest = require('./models/contest.js');

module.exports = function(io, sessionStore) {
  //多进程运行需要
  io.adapter(require('socket.io-redis')({
    host: 'localhost',
    port: 6379
  }));

  //socket验证cookie并获取会话
  io.use(function(socket, next){
    var handshakeData = socket.request;
    if (!handshakeData.headers.cookie) {
      return next(new Error('no cookie.'));
    }
    handshakeData.cookie = cookie.parse(handshakeData.headers.cookie);
    var sid = handshakeData.cookie['connect.sid'];
    if (!sid) {
      return next(new Error('no sid.'));
    }
    sessionStore.get(sid.split(':')[1].split('.')[0], function(err, session){
      if (err) {
        return next(err);
      }
      if (!session) {
        return next('no session.');
      }
      socket.session = session;
      next();
    });
  });
  
  //socket连接
  io.on('connection', function(socket){
    var session = socket.session;
    if (session && session.user) {
      var name = session.user.name;
      //监听广播请求
      socket.on('broadcast', function(data, fn){
        if (data) {
          var cid = parseInt(data.room, 10);
          if (!cid) {
            return ; //not allow
          }
          Contest.watch(cid)
          .then(function(contest){
            if (contest && (name === 'admin' || name === contest.userName)) {
              socket.broadcast.to(data.room).emit('broadcast', data.msg);
              if (fn) {
                fn(true);
              }
            }
          })
          .done();
        }
      });
    }
    //加入比赛房间
    socket.on('login', function(room){
      if (room) {
        socket.join(room.toString());
      }
    });
    //discuss实时提醒
    socket.on('addDiscuss', function(room){
      if (room) {
        socket.broadcast.to(room).emit('addDiscuss');
      }
    });
  });
};