express4下socket.io的验证

标签：node.js   angular   socket   聊天   

node实战里面有socket.io+angular.js的聊天室例子，但是现在express和socket.io的版本都发生了不小的变化，自己写了个express4下的socket.io验证

不废话，贴代码

var express = require('express');
var path = require('path');
var logger = require('morgan');
var cookieParser = require('cookie-parser');
var bodyParser = require('body-parser');
var async = require('async');
var app = express();
var config = require('./config');
var cookie = require('cookie');
var session = require('express-session');
var mongoose = require('mongoose');
var users = require('./routes/users');
var UserModel = require('./Dao/user')();
var MongoStore = require('connect-mongo')(session);
app.use(logger('dev'));
app.use(bodyParser.json());
app.use(bodyParser.urlencoded({ extended: false }));
app.use(cookieParser());
app.use(express.static(path.join(__dirname, 'public')));
app.set('views', path.join(__dirname, 'views'));
app.set('view engine', 'ejs');
//连接数据库
mongoose.connect(config.dbPath, function onMongooseError(err) {
    if (err) throw err;
});
var sessionStore = new MongoStore({
    url: 'mongodb://localhost/technode'
});
app.use(session({
    secret: 'technode',
    cookie: {
        maxAge: 60 * 1000
    },
    store:sessionStore,
    resave: true,
    saveUninitialized: true
}));
app.use('/users', users);

app.get('/',function (req, res) {
    res.sendfile('./static/index.html')
});


var io= require("socket.io").listen(app.listen(8000));
io.set('authorization', function(handshakeData, accept) {
     var cookies = cookie.parse(handshakeData.headers.cookie); //解析cookies
     var connectSid = cookies['connect.sid'];
     if(connectSid){//判断有无session登陆
       var connected = cookieParser.signedCookie(connectSid, 'technode');//验证session的secret
       if(connected){
        sessionStore.get(connected, function (error, session) { //去session里面取当前用户的数据
         if (error) {
           accept(error.message, false)
         } else {
           handshakeData.headers.sessions  = session; //把当前用户数据传给socket.io的<span style="font-family: Arial, Helvetica, sans-serif;">handshakeData</span>
            if (session.user) {
              accept(null, true)<span style="font-family:Helvetica, Tahoma, Arial, sans-serif;"><span style="font-size: 14px; line-height: 25.1875px;">//进行下面的链接</span></span>
            } else {
              accept('No login',false)
            }
        }
        })
      }else {
         accept('No session',false)//没session
      }
 }

 });
var SYSTEM = {
    account: '机器人'
}
var messages = [];
io.sockets.on('connection', function (socket) {
    socket.on('init', function () {
        socket.emit('init', messages);
        if(socket.handshake.headers.sessions!=null){ //如果有用户信息，确认登陆，执行操作
            var user = socket.handshake.headers.sessions.user;
            socket.broadcast.emit('messages.add',{
                content: user.account + '进入了聊天室',
                creator: SYSTEM,
                createAt: new Date()
            });
            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);
                    }
                });
            });

        }
    });
    socket.on('messages.read', function () {
        socket.emit('messages.read', messages);
        console.log("read");
    })
    socket.on('messages.create', function (message) {
        messages.push(message);
        io.sockets.emit('messages.add', message);
    });
    socket.on('disconnect', function() {//socket断开的时候，把用户状态设置为离线
        if(socket.handshake.headers.sessions!=null) {
            var user = socket.handshake.headers.sessions.user;
            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);
        }
    });
});

console.log("TechNode  is on port 8000!")

express4下socket.io的验证

标签：node.js   angular   socket   聊天   

原文地址：http://blog.csdn.net/chengscau/article/details/41808309