公司项目NODEJS实践0.3[ mongo / session ...]

摘要：使用，可参考执行退出命令，只要设置，即可。下节主要实现注册时的邮件验证保存登录状态异常处理，同步异步操作全栈工程技术新群上一篇公司项目实践下一篇待续

⋅⋅⋅书接上回，我们搭建了WEB服务端路由、模板等功能，完成了register 通过ajax与后端的通信，今天主要完成数据与mongodb的存取，实现注册 / 登录 / 退出功能

⋅⋅⋅DEMO GIT https://github.com/xiaolulu/mynodejs.git

⋅⋅⋅上一节我们已经安装过了mongo，本节主要是对其操作

⋅⋅⋅nodejs对 mongo的操作，我们使用 mongoose库

⋅⋅⋅在package.json添加mongoose，并npm install

⋅⋅⋅使用参考http://www.upopen.cn/article/info?id=559688a7f0e6e0665b000004

javascript        var mongoose = require( "mongoose"  ); //引用模块
        mongoose.connect( "mongodb://127.0.0.1/myDB", function( err ){
            //连接mongoose，连接本地127.0.0.1,mongo的默认端口是 27017
            if( !err ){
                console.log( "DB == connect to mongodb" );
            } else {
                throw err;
            }
        } );

        var Schema = mongoose.Schema;

        var UserSchema = new Schema({ //创建User表模型，数据可据需求增减
            username: String,
            password: String,
            email: String,
            disabled: Boolean, //后面加注册后的邮件验证功能
            date: Date,
            power: Number  //后面会用到权限功能  
        });

        var UserModel = mongoose.model( "User", UserSchema, "User" );

        function initData( data, db ){ //对参数做预处理，以防出现不合要求的参数，后面这块会做扩展
            var query = {};
            for( var key in data ){
                if( db.tree[ key ] ){
                    query[ key ] = data[ key ];
                }
            }
            return query;
        }

        function addUser( data, cb ){ //增加用户
            data = initData( data, UserSchema );
            ( new UserModel( data )).save( function( err, doc  ){
                cb( err, doc );
            })
        }
        function findUser( data, cb ){ //查找用户
            data = initData( data, UserSchema );
            UserModel.findOne( data ).exec( function( err, doc ){
                cb( err, doc );
            })
        }

        module.exports = {
            addUser: addUser,
            findUser: findUser
        }

javascript        var db = require( "../db/sql" ); //添加前面定义的db操作模块

        function addUser( req, res ){ //增加用户
            var data = req.body; //post过来的数据在req.body里，get过来的数据在req.query里
            data.date = new Date(); //数据里增加时间
            db.addUser( data, function( err, doc ){
            if( !err ){
                res.send( { code: 0, msg: "add User Success", data: doc } );
                //对查询结果返回，返回格式统一为 {code: 返回码, msg: 返回描述, data: 返回值}
                }
            })
        }

        function findUser( req, res ){ //查找用户
            var data = req.body;
            db.addUser( data, function( err, doc ){
                if( !err ){
                    res.send(  { code: 0, msg: "find User Success", data: doc }  );
                }
            })
        }

        module.exports = {
            addUser: addUser,
            findUser: findUser
        }

    4、修改上节在/root/web/routes/issue.js定义的register函数改为

javascript
        function registerUser( req, res ){
            user.addUser( req, res );
        }

javascript        var issue = require( "./issue" ),
              session = require( "express-session" ); //添加 express-session引用
        exports.all = function( app ){
            app.use( session({ //配置session
                resave: false,
                saveUninitialized: false,
                secret: "upopen"
            }))
            app.use( function( req, res, next){
                if( req.path != "/login" && !req.session.status ){ //判断session状态是否是true
                    res.redirect("/login"); //不是则跳转到登录页
                } else {
                    next(); //为true，则继续执行其请求
                }
            })
            app.get( "/", function( req, res ){
                issue.index( req, res );
            });
        …

javascript        …
        function findUser( req, res ){
        var data = req.body;
            db.findUser( data, function( err, doc ){
                if( !err ){
                    if( doc ){ //如果登录有查找结果
                        req.session.status = true;        //则session里记录状态为true
                    }
                    res.send(  { code: 0, msg: "find User Success", data: doc }  );
                }
            })
        }
        …

javascript
        module.exports = {
            "/user/center" : 1,
            "/user/info": 1,
            "/user/blog": 1
        }

javascript        …
        var privilege = require( "../config/privilege" );
        …
        app.use( function( req, res, next){
            if( privilege[ req.path ] && req.path != "/login" && !req.session.status ){
                //privilege[ req.path ] 判断该路径是否需要登录权限
                if( req.method == "GET" ){ //如果是get请求
                    res.redirect("/login"); 则执行跳转
                } else { //其它请求，基本都是POST，是不能直接redirect
                    res.send( { code: 1001, msg: "need you to log in"}); //则返回错误码，提示需要登录
                }
            } else {
                next();
            }
        })
        …

javascript         ...
        app.get("/logout", function( req, res ){
            req.session.status = false; //设置session状态为未登录
            res.setHeader("Set-Cookie","username=null;" ); //清除cookie
            res.redirect( "/" ); //跳转到首页
        })
        ...