从零开始学习Node.js例子八 使用SQLite3和MongoDB

setup.js：初始化数据库

var util = require('util');
var async = require('async');   //npm install async
var notesdb = require('./nodesdb-sqlite3');
// var notesdb = require('./notesdb-mongoose');

notesdb.connect(function(error){
    if (error) throw error;
});
notesdb.setup(function(error){
    if (error){
        util.log('ERROR ' + error);
        throw error;
    }
    async.series([   //async.series函数可以控制函数按顺序执行，从而保证最后的函数在所有其他函数完成之后执行
        function(cb){
            notesdb.add("test", "testtest",
            function(error){
                if (error) util.log('ERROR ' + error);
                cb(error);
            });
        }
    ],
        function(error, results){
            if (error) util.log('ERROR ' + error);
            notesdb.disconnect(function(err){});
        }
    );
});

nodesdb-sqlite3.js

SQLite3 是一个轻量级的进程内SQL引擎
它是一个无服务器且无需配置的SQL数据库引擎，仅仅是作为一个独立的库被链接到应用程序上

npm install sqlite3 安装此模块之前先在系统上安装sqlite3库 http://www.sqlite.org/download.html 下载

//数据库接口库
var util = require('util');
var sqlite3 = require('sqlite3');

sqlite3.verbose();
var db = undefined;

/*
 数据库名是直接硬编码的，所以当调用connect和setup函数时，当前目录中就会生成chap06.sqlite3文件
 */

exports.connect = function(callback){
    db = new sqlite3.Database("chap06.sqlite3", sqlite3.OPEN_READWRITE | sqlite3.OPEN_CREATE,
        function(err){
            if (err){
                util.log('FAIL on creating database ' + err);
                callback(err);
            } else {
                callback(null);
            }
        });
}

//此处的disconnect函数是空的
exports.disconnect = function(callback){
    callback(null);
}

exports.setup = function(callback){
    db.run("CREATE TABLE IF NOT EXISTS notes " +
        "(ts DATETIME, author VARCHAR(255), note TEXT)",
        function(err){
            if (err){
                util.log('FAIL on creating table ' + err);
                callback(err);
            } else {
                callback(null);
            }
        });
}

exports.emptyNote = {"ts": "", author: "", note: ""};
exports.add = function(author, note, callback){
    db.run("INSERT INTO notes (ts, author, note) " +
        "VALUES (?, ?, ?);",
        [new Date(), author, note],
        function(error){
            if (error){
                util.log('FAIL on add ' + error);
                callback(error);
            } else {
                callback(null);
            }
        });
}
/*
run函数接受一个字符串参数，其中?表示占位符，占位符的值必须通过一个数组传递进来
调用者提供了一个回调函数，然后通过这个回调函数来声明错误
 */

exports.delete = function(ts, callback){
    db.run("DELETE FROM notes WHERE ts = ?;",
        [ts],
        function(err){
            if (err){
                util.log('FAIL to delete ' + err);
                callback(err);
            } else {
                callback(null);
            }
        });
}

exports.edit = function(ts, author, note, callback){
    db.run("UPDATE notes " +
        "SET ts = ?, author = ?, note = ? " +
        "WHERE ts = ?",
        [ts, author, note, ts],
        function(err){
            if (err){
                util.log('FAIL on updating table ' + err);
                callback(err);
            } else {
                callback(null);
            }
        });
}

exports.allNotes = function(callback){
    util.log(' in allnote');
    db.all("SELECT * FROM notes", callback);
}
exports.forAll = function(doEach, done){
    db.each("SELECT * FROM notes", function(err, row){
        if (err){
            util.log('FAIL to retrieve row ' + err);
            done(err, null);
        } else {
            doEach(null, row);
        }
    }, done);
}
/*
allNotes和forAll函数是操作所有数据的两种方法，allNotes把数据库中所有的数据行收集到一个数组里，
而forAll方法可以接受两个回调函数，每当从数据集中拿一行数据，回调函数doEach都会执行一遍，当读完所有数据时，回调函数done就会执行
 */

exports.findNoteById = function(ts, callback){
    var didOne = false;
    db.each("SELECT * FROM notes WHERE ts = ?",
        [ts],
        function(err, row){
            if (err){
                util.log('FAIL to retrieve row ' + err);
                callback(err, null);
            } else {
                if (!didOne){
                    callback(null, row);
                    didOne = true;   //保证回调函数只被执行一次
                }
            }
        });
}

notesdb-mongoose.js

MongoDB是nosql数据库的领头羊之一，"可扩展、高性能、开源、面向文档的数据库"，它使用JSON风格的文档。
Mongoose是用于访问MongoDB的模块之一，它是一个对象建模工具，意味着你的程序负责定义模式对象来描述数据，
而Mongoose负责数据到MongoDB的存储。
Mongoose对于Node和MongoDB而言是一个非常强大的对象建模工具，使用嵌入式文档，是一个类型灵活的系统，
适用于字段输入、字段验证、虚拟字段等。

MongoDB Windows 下安装部署 http://www.cnblogs.com/EricaMIN1987_IT/p/3571773.html
安装Mongoose模块 npm install mongoose

Mongoose不是唯一一个在node中使用MongoDB的工具。

var util = require('util');
var mongoose = require('mongoose');
var Schema = mongoose.Schema;
var dburl = 'mongodb://localhost/chap06';  //dburl用于连接已运行的MongoDB

exports.connect = function(callback){
    mongoose.connect(dburl);
}
exports.disconnect = function(callback){
    mongoose.disconnect(callback);
}

exports.setup = function(callback){callback(null);}

//定义模式
var NoteSchema = new Schema({
    ts: {type: Date, default: Date.now},  //默认值
    author: String,
    note: String
});
//将NoteSchema作为Mongoose的模型注册进去
mongoose.model('Note', NoteSchema);
var Note = mongoose.model('Note');

exports.emptyNote = {"_id": "", author: "", note: ""};

exports.add = function(author, note, callback){
    var newNote = new Note();
    newNote.author = author;
    newNote.note = note;
    newNote.save(function(err){
        if (err){
            util.log('FATAL ' + err);
            callback(err);
        } else {
            callback(null);
        }
    });
}

exports.delete = function(id, callback){
    exports.findNoteById(id, function(err, doc){
        if (err){
            callback(err);
        } else {
            util.log(util.inspect(doc));
            doc.remove();
            callback(null);
        }
    });
}

exports.edit = function(id, author, note, callback){
    exports.findNoteById(id, function(err, doc){
        if (err){
            callback(err);
        } else {
            doc.ts = new Date();
            doc.author = author;
            doc.note = note;
            doc.save(function(err){
                if (err){
                    util.log('FATAL ' + err);
                    callback(err);
                } else {
                    callback(null);
                }
            });
        }
    });
}

exports.allNotes = function(callback){
    Note.find({}, callback);
}

exports.forAll = function(doEach, done){
    Note.find({}, function(err, docs){
        if (err){
            util.log('FATAL ' + err);
            done(err, null);
        }
        docs.forEach(function(doc){
            doEach(null, doc);
        });
        done(null);
    });
}

/*
_id字段是MongoDB提供的全局唯一的ID，用于标识存储的文档
 */
var findNoteById = exports.findNoteById = function(id, callback){
    Note.findOne({_id: id}, function(err, doc){
        if (err){
            util.log('FATAL ' + err);
            callback(err, null);
        }
        callback(null, doc);
    });
}

app.js

//在数据库需要放置在一台计算机上时，应该考虑使用SQLite3
//控制器，在nodesdb-sqlite3.js和notesdb-mongoose.js模块之间切换
var util = require('util');
var url = require('url');
var express = require('express');
var nmDbEngine = 'sqlite3';  //用于命名数据库引擎、选择合适的notesdb实现和选择合适的views目录
//var nmDbEngine = 'mongoose';
var notesdb = require('./nodesdb-' + nmDbEngine);
var app = express();
app.use(express.logger());
app.use(express.cookieParser()); //添加cookieParser中间件
app.use(express.bodyParser());
app.engine('.html', require('ejs').__express);   //3.X
//app.register('.html', require('ejs'));    //2.X
app.set('views', __dirname + '/views-' + nmDbEngine);
app.set('view engine', 'ejs');

//是一个路由中间件函数，用于在一些路由器函数中解析URL查询参数
var parseUrlParams = function(req, res, next){
    req.urlP = url.parse(req.url, true);
    next();
}

//检查用户是否被允许访问，这里只检查cookie是否等于AOK，这个单词通常意味着一切都没问题
/*
 很多应用都需要用户登录，然后用户才能进行一些特权操作。由于HTTP是一个无状态的协议，
 验证用户的唯一方式就是发送一个cookie到浏览器上，然后验证标识符。cookie包含了应用中用于验证用户的数据。

 cookieParser中间件在这里做了很多工作，查找cookie，解析cookie，然后将解析出来的值让到req对象中。
 当存在cookie时，它的值会被放入req.cookies中。
 */
var checkAccess = function(req, res, next){
    if (!req.cookies || !req.cookies.notesaccess || req.cookies.notesaccess !== "AOK"){
        res.redirect('/login');
    } else {
        next();
    }
}

notesdb.connect(function(error){
    if (error) throw error;
})

app.on('close', function(error){
    notesdb.disconnect(function(err){});
});

app.get('/', function(req, res) {res.redirect('/view');});
app.get('/view', checkAccess, function(req, res){  //可以在每个路由上加checkAccess检查
    notesdb.allNotes(function(err, notes){
        if (err){
            util.log('ERROR ' + err);
            throw err;
        } else {
            res.render('viewnotes.html', {title: "Notes ("+ nmDbEngine +")", notes: notes});
        }
    });
});

/*
当用户点击ADD按钮时app.get('/add', ...)内的函数就会被调用，浏览器会发出一个发往/add的HTTP GET请求。
这个函数使用addedit.html模板来创建一个表单，让用于通过这个表单输入标签，然后通过单击SUBMIT按钮提交，
当用户提交表单，浏览器就会发出一个HTTP POST请求，app.post('/add', ...)内的函数就会被调用，
用户输入的数据会被存放在请求主体中，而请求主体会被bodyParser(app.use(express.bodyParser()))中间件处理并存放在req.body中
 */
app.get('/add', function(req, res){
    res.render('addedit.html', {title: "Notes ("+ nmDbEngine +")", postpath: '/add', note: notesdb.emptyNote});
});
app.post('/add', function(req, res){
    notesdb.add(req.body.author, req.body.note,
        function(error){
            if (error) throw error;
            res.redirect('/view');
        });
});

app.get('/del', parseUrlParams, function(req, res){
    notesdb.delete(req.urlP.query.id,
        function(error){
            if (error) throw error;
            res.redirect('/view');
        });
});

app.get('/edit', parseUrlParams, function(req, res){
    notesdb.findNoteById(req.urlP.query.id,
        function(error, note){
            if (error) throw error;
            res.render('addedit.html',
                {title: "Notes ("+ nmDbEngine +")", postpath: '/edit', note: note});
        });
});
app.post('/edit', function(req, res){
    notesdb.edit(req.body.id, req.body.author, req.body.note,
        function(error){
            if (error) throw error;
            res.redirect('/view');
        });
});

app.get('/login', function(req, res){
    res.render('login.html', {title: "Notes LOGIN ("+ nmDbEngine +")"});
});
app.post('/login', function(req, res){
    //此处可以添加检查用户信息的逻辑
    //...
    res.cookie('notesaccess', 'AOK');
    res.redirect('/view');
});

app.listen(3000);

show.js

//控制台显示
var util = require('util');
var notesdb = require('./notesdb-sqlite3');
// var notesdb = require('./notesdb-mongoose');

notesdb.connect(function(error){
    if (error) throw error;
});
notesdb.forAll(function(error, row){
    util.log('ROW: ' + util.inspect(row));
}, function(error){
    if (error) throw error;
    util.log('ALL DONE');
    notesdb.disconnect(function(err){});
});

前台页面在views-sqlite3目录下

layout.html

<!DOCTYPE html>
<html>
<head>
    <title><%= title%></title>
</head>
<body>
    <h1><%= title%></h1>
    <p><a href='/view'>View</a> | <a href='/add'>Add</a></p>
</body>
</html>

viewnotes.html

<% include layout.html %>
<table><% notes.forEach(function(note){ %>
    <tr>
        <td>
            <p><%=new Date(note.ts).toString()%>: by <b><%=note.author%></b></p>
            <p><%=note.note%></p>
        </td>
        <td>
            <form method="get" action="/del">
                <input type="submit" value="Delete" />
                <input type="hidden" name="id" value="<%=note.ts%>" />
            </form>
            <br/>
            <form method="get" action="/edit">
                <input type="submit" value="Edit" />
                <input type="hidden" name="id" value="<%=note.ts%>" />
            </form>
        </td>
    </tr>
    <% }); %>
</table>

addedit.html

<% include layout.html %>
<form method="post" action="<%=postpath%>">
    <% if (note){ %>
    <input type="hidden" name="id" value="<%=note.ts%>" />
    <% } %>
    <input type="text" name="author" value="<%=note.author%>" />
    <br/>
    <textarea rows="5" cols="40" name="note">
        <%=note.note%>
    </textarea>
    <br/>
    <input type="submit" value="Submit" />
</form>

login.html

<% include layout.html %>
<form method="POST" action="/login">
    <p>Click the <i>Login</i> to log in.</p>
    <input type="submit" value="Login" />
</form>

node setup.js

node app.js