112

我想删除一些 mongodb 集合,但这是一个异步任务。代码将是:

var mongoose = require('mongoose');

mongoose.connect('mongo://localhost/xxx');

var conn = mongoose.connection;

['aaa','bbb','ccc'].forEach(function(name){
    conn.collection(name).drop(function(err) {
        console.log('dropped');
    });
});
console.log('all dropped');

控制台显示:

all dropped
dropped
dropped
dropped

all dropped在删除所有集合后确保打印的最简单方法是什么?任何第三方都可以用来简化代码。

4

8 回答 8

129

使用承诺

var mongoose = require('mongoose');

mongoose.connect('your MongoDB connection string');
var conn = mongoose.connection;

var promises = ['aaa', 'bbb', 'ccc'].map(function(name) {
  return new Promise(function(resolve, reject) {
    var collection = conn.collection(name);
    collection.drop(function(err) {
      if (err) { return reject(err); }
      console.log('dropped ' + name);
      resolve();
    });
  });
});

Promise.all(promises)
.then(function() { console.log('all dropped)'); })
.catch(console.error);

这会丢弃每个集合,在每个集合之后打印“dropped”,然后在完成时打印“all dropped”。如果发生错误,则显示为stderr


上一个答案(这早于 Node 对 Promises 的原生支持):

使用Q承诺或Bluebird承诺。

var Q = require('q');
var mongoose = require('mongoose');

mongoose.connect('your MongoDB connection string');
var conn = mongoose.connection;

var promises = ['aaa','bbb','ccc'].map(function(name){
    var collection = conn.collection(name);
    return Q.ninvoke(collection, 'drop')
      .then(function() { console.log('dropped ' + name); });
});

Q.all(promises)
.then(function() { console.log('all dropped'); })
.fail(console.error);

蓝鸟

var Promise = require('bluebird');
var mongoose = Promise.promisifyAll(require('mongoose'));

mongoose.connect('your MongoDB connection string');
var conn = mongoose.connection;

var promises = ['aaa', 'bbb', 'ccc'].map(function(name) {
  return conn.collection(name).dropAsync().then(function() {
    console.log('dropped ' + name);
  });
});

Promise.all(promises)
.then(function() { console.log('all dropped'); })
.error(console.error);
于 2012-05-22T21:13:35.163 回答
91

我看到您正在使用mongoose,所以您在谈论服务器端 JavaScript。在这种情况下,我建议查看异步模块并使用async.parallel(...). 你会发现这个模块真的很有帮助——它是为了解决你正在努力解决的问题而开发的。您的代码可能如下所示

var async = require('async');

var calls = [];

['aaa','bbb','ccc'].forEach(function(name){
    calls.push(function(callback) {
        conn.collection(name).drop(function(err) {
            if (err)
                return callback(err);
            console.log('dropped');
            callback(null, name);
        });
    }
)});

async.parallel(calls, function(err, result) {
    /* this code will run after all calls finished the job or
       when any of the calls passes an error */
    if (err)
        return console.log(err);
    console.log(result);
});
于 2012-05-11T13:33:58.553 回答
21

这样做的方法是向任务传递一个更新共享计数器的回调。当共享计数器达到零时,您知道所有任务都已完成,因此您可以继续正常流程。

var ntasks_left_to_go = 4;

var callback = function(){
    ntasks_left_to_go -= 1;
    if(ntasks_left_to_go <= 0){
         console.log('All tasks have completed. Do your stuff');
    }
}

task1(callback);
task2(callback);
task3(callback);
task4(callback);

当然,有很多方法可以使这种代码更通用或可重用,并且许多异步编程库中的任何一个都应该至少有一个函数来做这种事情。

于 2012-05-11T13:40:37.903 回答
8

扩展@freakish 的答案, async 还提供了 each 方法,这似乎特别适合您的情况:

var async = require('async');

async.each(['aaa','bbb','ccc'], function(name, callback) {
    conn.collection(name).drop( callback );
}, function(err) {
    if( err ) { return console.log(err); }
    console.log('all dropped');
});

恕我直言,这使代码更加高效和清晰。我冒昧地删除了console.log('dropped')- 如果你想要它,请改用它:

var async = require('async');

async.each(['aaa','bbb','ccc'], function(name, callback) {
    // if you really want the console.log( 'dropped' ),
    // replace the 'callback' here with an anonymous function
    conn.collection(name).drop( function(err) {
        if( err ) { return callback(err); }
        console.log('dropped');
        callback()
    });
}, function(err) {
    if( err ) { return console.log(err); }
    console.log('all dropped');
});
于 2013-05-28T08:34:42.883 回答
5

我这样做没有外部库:

var yourArray = ['aaa','bbb','ccc'];
var counter = [];

yourArray.forEach(function(name){
    conn.collection(name).drop(function(err) {
        counter.push(true);
        console.log('dropped');
        if(counter.length === yourArray.length){
            console.log('all dropped');
        }
    });                
});
于 2013-12-27T14:19:12.853 回答
4

所有的答案都很老了。自 2013 年初以来,Mongoose 开始逐渐支持所有查询的 Promise,因此我猜这将是按照所需顺序构建多个异步调用的推荐方式。

于 2013-11-06T09:27:03.377 回答
0

使用deferred(另一个承诺/延迟实现),您可以执行以下操作:

// Setup 'pdrop', promise version of 'drop' method
var deferred = require('deferred');
mongoose.Collection.prototype.pdrop =
    deferred.promisify(mongoose.Collection.prototype.drop);

// Drop collections:
deferred.map(['aaa','bbb','ccc'], function(name){
    return conn.collection(name).pdrop()(function () {
      console.log("dropped");
    });
}).end(function () {
    console.log("all dropped");
}, null);
于 2012-07-12T19:32:00.113 回答
0

如果您使用 Babel 或此类转译器并使用 async/await,您可以这样做:

function onDrop() {
   console.log("dropped");
}

async function dropAll( collections ) {
   const drops = collections.map(col => conn.collection(col).drop(onDrop) );
   await drops;
   console.log("all dropped");
}
于 2016-03-12T16:20:19.567 回答