OK，废话少说，直接上代码
首先是app.js,里面很多与书中介绍的不同，不过看起来都很容易懂。首先是各种模块引用，然后就是使用session保存状态。
var express = require('express'); var path = require('path'); var favicon = require('serve-favicon'); var logger = require('morgan'); var cookieParser = require('cookie-parser'); var bodyParser = require('body-parser'); var session = require('express-session');//需要项目目录下npm install 安装模块 var MongoStore = require('connect-mongo')(session);//需要项目目录下npm install connect-mongo 安装模块 var settings = require("./settings"); var flash = require('connect-flash');//需要项目目录下npm install connect-flash 安装模块 var routes = require('./routes/index'); var users = require('./routes/users'); var app = express(); // view engine setup app.set('views', path.join(__dirname, 'views')); app.set('view engine', 'ejs'); // uncomment after placing your favicon in /public //app.use(favicon(__dirname + '/public/favicon.ico')); app.use(logger('dev')); app.use(bodyParser.json()); app.use(bodyParser.urlencoded({ extended: false })); app.use(cookieParser()); app.use(require('less-middleware')(path.join(__dirname, 'public'))); app.use(express.static(path.join(__dirname, 'public'))); app.use(flash()); app.use(session({ secret:settings.coolieSecret,//express.cookieParser() 是 Cookie 解析的中间件 //store参数为MongoStore实例，把会话信息存储到数据库中，以避免丢失。 store:new MongoStore({ db:settings.db }), resave:true, saveUninitialized:true })); //获取状态 app.use(function(req,res,next){ console.log("app.user local"); res.locals.user = req.session.user; res.locals.post = req.session.post; var error = req.flash('error'); res.locals.error = error.length?error:null; var success = req.flash('success'); res.locals.success = success.length?success:null; next(); }); app.use('/', routes); app.use('/users', users); // catch 404 and forward to error handler app.use(function(req, res, next) { var err = new Error('Not Found'); err.status = 404; next(err); }); // error handlers // development error handler // will print stacktrace if (app.get('env') === 'development') { app.use(function(err, req, res, next) { res.status(err.status || 500); res.render('error', { message: err.message, error: err }); }); } // production error handler // no stacktraces leaked to user app.use(function(err, req, res, next) { res.status(err.status || 500); res.render('error', { message: err.message, error: {} }); }); module.exports = app;
下面看看路由控制routes/index.js:代码中都有注释。其中需要注意的是中文参数的处理（先base64编码再解码）
var express = require('express'); var crypto = require('crypto'); var router = express.Router(); var settings = require("../settings"); var User = require("../models/user.js"); var Post = require("../models/post.js"); var iconv = require('iconv-lite'); /* GET home page. */ /* router.get('/', function(req, res, next) { res.render('index', { title: settings.title }); }); */ router.get('/',function(req,res){ Post.get(null,function(err,posts){ if(err){ posts = []; } res.render('index',{title:"首页",posts:posts}); }); }); router.get('/reg',checkNotLogin); router.get('/reg',function(req,res){ res.render('reg',{title:"注册"}); }); //注册事件处理 router.post('/reg',checkNotLogin); router.post('/reg',function(req,res){ console.log(req.body); //检测两次输入的密码是否相同 //req.body 就是 POST 请求信息解析过后的对象，例如我们要访问用户传递的password 域的值，只需访问 req.body['password'] 即可。 if(req.body['password-repeat'] != req.body['password']){ //req.flash 是 Express 提供的一个奇妙的工具，通过它保存的变量只会在用户当前和下一次的请求中被访问，之后会被清除， //通过它我们可以很方便地实现页面的通知和错误信息显示功能 req.flash('error','两次输入的口令不一致'); return res.redirect('/reg'); } //生成口令的散列值 //crypto 是 Node.js 的一个核心模块，功能是加密并生成各种散列， //使用它之前首先要声明 var crypto = require('crypto')。我们代码中使用它计算了密码的散列值。 var md5 = crypto.createHash('md5'); var password = md5.update(req.body.password).digest('base64'); var newUser = new User({ name : req.body.username, password : password }); //检查用户名是否存在 User.get(newUser.name,function(err,user){ if(user){ err = "username is already exists. "; } if(err){ req.flash('error',err); console.log(err); return res.redirect('/reg'); } console.log("save"); //不存在则保存新用户 newUser.save(function(err){ if(err){ req.flash('error',err); console.log(err); console.log("save err"); return res.redirect('/reg'); } req.session.user = newUser; req.flash('success','注册成功'); return res.redirect('/'); }); }); }); //登录 router.get('/login',checkNotLogin); router.get('/login',function(req,res){ res.render('login',{title:'登录'}); }); //响应登录事件 router.post('/login',checkNotLogin); router.post('/login',function(req,res){ var md5 = crypto.createHash('md5'); var password = md5.update(req.body.password).digest('base64'); User.get(req.body.username,function(err,user){ if(!user){ req.flash('error','用户名不存在'); return res.redirect('/login'); } if(user.password != password){ req.flash('error','密码错误'); return res.redirect('/login'); } req.session.user = user; req.flash('success','登录成功'); return res.redirect('/'); }) }); //登出 router.get('/logout',checkLogin); router.get('/logout', function (req,res) { req.session.user = null; req.flash('success','登出成功'); res.redirect('/'); }); //发表处理 router.post('/post',checkLogin); router.post('/post',function(req,res){ var currentUser = req.session.user; var post = new Post(currentUser.name,req.body.post); post.save(function(err){ if(err){ req.flash('error',err); return res.redirect('/'); } req.flash('success','发表成功'); console.log('/u/'+currentUser.name); var pramsUserNnametemp = new Buffer(currentUser.name); var pramsUserNname = pramsUserNnametemp.toString('base64');//为了解决中文参数问题将参数base64编码，然后解码 // res.redirect('/u/'+currentUser.name); res.redirect('/u/'+pramsUserNname); }); }); router.get("/u/:user",function(req,res){ var uNametemp = new Buffer(req.params.user,'base64');//base64解码 var uName = uNametemp.toString(); //User.get(req.params.user, function(err,user){ User.get(uName, function(err,user){ if(!user){ req.flash('error','用户不存在'); res.redirect('/'); } Post.get(user.name,function(err,posts){ if(err){ req.flash('error',err); res.redirect('/'); } res.render('user',{title:user.name,posts:posts}); }); }); }); function checkLogin(req,res,next){ if(!req.session.user){ req.flash('error',"未登录"); return res.redirect('/login'); } next(); } function checkNotLogin(req,res,next){ if(req.session.user){ req.flash('error',"已登录"); return res.redirect('/'); } next(); } module.exports = router;
在看一下models目录下的user.js,就是对mongodb的操作，保存查询
var mongodb = require('./db'); function User(user){ this.name = user.name; this.password = user.password; }; module.exports = User; User.prototype.save = function save(callback){ //存入mongodb的文档 var user = { name:this.name, password:this.password }; mongodb.open(function(err,db){ if(err){ return callback(err); } //读取users集合 db.collection('users',function(err,collection){ if(err){ mongodb.close(); return callback(err); } //为name属性添加索引，新版本的ensureIndex方法需要一个回调函数 collection.ensureIndex('name',{unique:true},function(err){ //写入user文档 collection.insert(user,{safe:true},function(err,user){ mongodb.close(); callback(err,user); }); }); }); }); }; User.get = function get(username,callback){ mongodb.open(function(err,db){ if(err){ return callback(err); } //读取users集合 db.collection('users',function(err,collection){ if(err){ mongodb.close(); return callback(err); } //查找name属性为username的文档 collection.findOne({name:username},function(err,doc){ mongodb.close(); if(doc){ //封装文档为User对象 var user = new User(doc); callback(err,user); } else{ callback(err,null); } }); }); }); };post.js:
var mongodb = require("./db"); function Post(username,post,time){ this.user = username; this.post = post; if(time){ this.time = time; } else{ this.time = new Date(); } }; module.exports = Post; Post.prototype.save = function save(callback){ //存入Mongodb的文档 var post = { username:this.user, post:this.post, time:this.time }; mongodb.open(function(err,db){ if(err){ return callback(err); } //读取posts集合 db.collection('posts',function(err,collection){ if(err){ mongodb.close(); return callback(err); } //为user属性添加索引 collection.ensureIndex('user',function(err){ //写入post文档 collection.insert(post,{safe:true},function(err,post){ mongodb.close(); callback(err,post); }); }); }); }); }; Post.get = function get(username,callback){ mongodb.open(function(err,db){ if(err){ return callback(err); } //读取posts集合 db.collection('posts',function(err,collection){ if(err){ mongodb.close(); return callback(err); } //查找user属性为username的文档，如果sername是null则全部匹配 var query = {}; if(username){ query.username = username; } collection.find(query).sort({time:-1}).toArray(function(err,docs){ mongodb.close(); if(err){ callback(err,null); } //封装posts为Post对象 var posts = []; docs.forEach(function(doc,index){ var post = new Post(doc.username,doc.post,doc.time); posts.push(post); }); callback(null,posts); }); }); }); };
主要的代码就这些了，跟书中的差别还是很大的，主要模块更新太快，书更新的速度跟不上技术的更新。
763
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
