koa2中使用mongodb写api接口获取数据，怎么优雅的实现取到数据

Question

const MongoClient = require('mongodb').MongoClient;const url = 'mongodb://localhost:27017';// Database Nameconst dbName = 'youyue';var obj = {};module.exports = async(ctx,next)=>{     const num = ctx.query.num,        page = ctx.query.page,        sort = ctx.query.sort;    num = isNaN(num)? 10 : parseFloat(ctx.query.num)    page = isNaN(page)?0 : parseFloat(ctx.query.page)    sort = isNaN(sort)?-1 : parseFloat(ctx.query.sort)           console.log(num+"*"+sort+"*"+page);    MongoClient.connect(url, function (err, client) {        const db = client.db(dbName);        const collection = db.collection('site');        collection.find().sort({ id: sort  }).skip(num*page).limit(num).toArray(function (err, Database){            console.log(Database);            //目标数据            obj.result = Database;        });    });    await next();     //填充数据到页面    ctx.body = obj.result;}

Answer 1

把 query Mongodb 那段写成一个function， 这个function return promise。 然后调用这个function 就不low了。

function queryToDB(url){

  return new Promise(function(resolve, reject){

    MongoClient.connect(url, function (err, client) {

    const db = client.db(dbName);

    const collection = db.collection('site');

    collection.find().sort({ id: sort  }).skip(num*page).limit(num).toArray(function (err, Database){

        if(err) reject(err);

        else resolve(Database)

    });

});

    

queryToDB(url).then(function(result){

  if(result && result.length > 0){

    ctx.body = result;

  }

}).catch(function(ex){

   // Todo:

});

Answer 2

优雅的建议

1. 把MongoClient.connect用util.promisify把他改成promise用await等待链接成功。

2. 持久化数据链接，不要每次请求链接数据库。

3. 把const collection = db.collection('site');写成类的模式Site.find。

4. 同理把Site.find写成await。