在项目中我们会频繁使用mongodb去做增删改查操作，每次的操作都要写一遍connect、collection，代码量很多，重复代码也很多，这不是一个好的程序员写出的代码。所以我们需要对我们的代码进行封装
未经过封装的mongodb 查询功能代码如下：
MongoClient.connect(dbUrl,function (err,db) {
if(err){
console.log(err);
return;
}
var dbo = db.db("productmanage");
const list = [];
const result = dbo.collection("user").find({"username":param.username,"password":pas});
result.toArray(function (err,data) {
if(err){
return;
}
//拿到数据库数据，进行自定义操作
db.close();
});
})
未经过封装的mongodb 插入功能代码如下：
MongoClient.connect(dbUrl,function (err,db) {
if(err){
console.log(err);
return;
}
var dbo = db.db("productmanage");
dbo.collection(collectionName).insertOne(json,function (error,data) {
if(error){
return;
}
//插入成功之后自定义操作
db.close();
});
})
以上可以看到关于数据库连接一直到dbo.collection操作之前的代码都是重复的，这部分内容我们可以提出作为一个方法，后面的find、inserOne以及deleteOne、updateOne可以写成不同方法去调用，完整的代码如下
const MongoClient = require("mongodb").MongoClient;
const dbUrl = "mongodb://127.0.0.1:27017/";
function _connectDB(callback){
MongoClient.connect(dbUrl,function(err,db){
if(err){
console.log("数据库链接失败");
return;
}
var dbo = db.db("productmanage");
callback(err,dbo);
db.close();
});
}
//查询数据
exports.find = function(collectionName,json,callback){
_connectDB(function(err,dbo){
if(err){
console.log(err);
return;
}
const result = dbo.collection(collectionName).find(json);
result.toArray(function (error,data) {
callback(error,data);
});
});
}
//新增数据
exports.insert = function(collectionName,json,callback){
_connectDB(function(err,dbo){
if(err){
console.log(err);
return;
}
dbo.collection(collectionName).insertOne(json,function (error,data) {
callback(error,data);
});
});
}
//修改数据
exports.update = function(collectionName,json1,json2,callback){
_connectDB(function(err,dbo){
if(err){
console.log(err);
return;
}
dbo.collection(collectionName).updateOne(json1,{$set:json2},function(error,data) {
callback(error,data);
});
});
}
//删除数据.
exports.delete = function(collectionName,json,callback){
_connectDB(function(err,dbo){
if(err){
console.log(err);
return;
}
dbo.collection(collectionName).deleteOne(json,function(error,data) {
callback(error,data);
});
});
}
使用：
导入
const DB = require("./module/db");
调用:
DB.find("user",
{"username":param.username,"password":pas},
function(err,data){
if(err){
return;
}
if(data&&data.length>0){
req.session.userInfo = data[0];
res.redirect("/product");
}else{
res.send("<script>alert('登录失败');location.href='/login';</script>");
}
});
代码量减少，代码也很简洁，而且很易懂，后期维护和别人对接也方便很多。
学会提纯简化代码，封装方法，是我们应该学会的技能。
完整项目git地址：https://github.com/super-jingjing/nodejs-ejs-expressSimpleProject.git
