在使用mongodb的时候有一个问题,在第二次请求时前一次数据库还没有关闭的时候,会导致报错。比较暴力的解决方法是让数据库一直开着,本人自己也做了一个方法来避免此情况发生。将请求数据库的任务放进队列,在数据库打开后完成所有数据库任务后resolve来关闭数据库。
//post.js
var mongodb = require('./db');
function Post(title, post) {
this.pool=[];
this.flag=false;
this.mongodb = mongodb
}
module.exports = Post;
//打开数据库
Post.prototype.open = function () {
this.flag=true; //数据库已开启
return new Promise(function(resolve,reject){
mongodb.open(function (err, db) {
if (err)reject(err)
resolve(db)
})
})
}
//读数据库
Post.prototype.get = function (db) {
return new Promise((resolve,reject)=>{
while (this.pool.length) { //判断是否还有任务池
db.collection('posts', (err, collection)=> {
if (err)reject(err)
let data = this.pool.shift(); //出队列
collection.find(data.type ? {type: data.type} : {}, {_id: 0}) //读数据
.sort({time: -1})
.toArray(function (err, docs) {
if (err)reject(err)
docs.forEach((item, index) => {
item.post = item.post.slice(0, 150) + "..."
})
data.resolve(docs) //成功!resolve出队列的任务
});
});
}
resolve() //清空完任务池resolve来关闭数据库或其他操作
})
};
//index.js
var express = require('express');
var router = express.Router();
var Post = require('../models/post.js');
var post = new Post();
router.get('/', function (req, res, next) {
new Promise(function (resolve, reject) {
post.pool.push({type: req.query.type ? req.query.type : "",resolve}) // 将任务放进队列
}).then(resolve => {
res.send({data: resolve})
})
if (!post.flag) { //判断数据库状态
post.flag = true; //数据库设置为开启状态
post.open().then(resolve => {
return post.get(resolve) //开始查询数据
}).then(resolve=>{
post.mongodb.close();//关闭数据库
post.flag = false; //数据库已关闭
}).catch(err=>{
post.mongodb.close();
post.flag=false;
res.send(400, {error: err})
})
}
});
module.exports = router;