封装mongodb模块

在项目中我们会频繁使用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