node.js - 在nodejs中为什么mongodb会影响到generator的运行

Question

我在项目中使用nodejs中的generator对mongodb进行操作，但是实验发现只要mongodb的连接不关闭，那么即便generator的done变成true了，程序也不会退出，请各位大侠解惑。

下面的代码中只要把所有的db.close()注释掉，就会出现generator的done为true了，但是程序依然不退出的情况。

var mongodb = require('mongodb');

var db = null;

function* DO(){
    try{
        db = yield connectMongodb();
        var collection = db.collection('userStruct');
        console.log('collection');
        var res = yield findSomething(collection, db);
        console.log('res', res);
    }catch(e){
        console.log('err', e);
        if(db){
            db.close();
        }
    }
}

var a = DO();
console.log('first', a.next());

function connectMongodb(){
    var mongoClient = mongodb.MongoClient;

    var DB_CONN_STR = 'mongodb://127.0.0.1:27017/jue';

    mongoClient.connect(DB_CONN_STR, function(err, db){
        if(err){
            a.throw(err);
        }else{
            console.log('connect', a.next(db));}
        }
    })
}

function findSomething(collection, db){
    collection.find({test:1}).toArray(function(err, retArr){
        if(err){
            a.throw(err);
        }else{
            console.log('in find');
            db.close();
            console.log('find', a.next(retArr));
        }
    })
}

Answer 1

yield 后面你应该接一个 promise 吧