使用Node.js进行回调递归插入Mongo

debugcn 发表于 Dev

手册

我正在使用Node.js和MongoDB（Mongoose.js）做API。我制作了一个“安装”脚本，用一些示例数据（例如用户和其他内容）填充数据库。

问题是当我尝试添加递归插入时。

我将999个文档添加到一个集合（父亲）中，然后将12个文档添加到另一个集合（儿子）中的每个父亲文档中。

父亲收藏有999份文件。但是，当我期望Sons集合中有11988个文档时，儿子的集合只有1188个。

模型是正确的。我有很多单元测试可以验证这些操作。

这是我的代码：

var mongoose = require('mongoose');
var conf = require('../config/dbConfig');
var fatherDb = require('../model/father');
var sonsDb = require('../model/son');
var db_lnk = conf.mongoSrv();
global.db = (global.db ? global.db : mongoose.createConnection(db_lnk));

function addFather(idSome, idOther){
    console.log("Adding Fathers");
    fatherDb.count({active:true}, function(err,count){
        if(!err){
            var max = (999-count);
            for(i=0; i<max; i++){
                var father = new fatherDb({
                    "name":"Father "  + (i+1),
                    "createdBy" : "55ad5f224f350c123b8e2bda",
                    "modifyBy" : "55ad5f224f350c123b8e2bda",
                    "active" : true,
                    "_some" : idSome,
                    "_other" : idOther,
                    "__v" : 0
                });
                father.save(function(err, data){
                    console.log("Adding 12 Sons to Father # " + i);
                    var idFather = data._id;
                    for (var x = 1; x < 13; x++) {
                        var son = new sonsDb({
                            "name": "Son " + x,
                            "position": x,
                            "createdBy": "55ad5f442f350c431b8e2bda",
                            "modifyBy": "55ad5f442f350c431b8e2bda",
                            "active": true,
                            "_father": idFather,
                            "__v": 0
                        });
                        son.save(function (err) {
                            console.log("Adding Son " + x);
                            if(i == max) {
                                endInstall();
                            }
                        });
                    }
                });
            }
        }else{
            console.log("Error");
            endInstall();
        }
    });
}

function endInstall(){
    console.log("Ending Installation");
    process.exit();
}

addFather('SOMEid','OtherId');

伊格纳西奥·拉戈（Ignacio Lago）

.save()是一个异步函数，因此for循环将在它们完成之前继续运行它们。这样i可以max比您预期的更快。

换句话说，i到达目标max将father.save被解雇，但尚未完成，并且son.save会endInstall()过早执行。

异步是一个库来处理这种情况。如果要继续使用计数器，请查看times方法。

就像是：

// generate max fathers
async.times(max, function(n, next_father){
  var father = ...
  father.save(function(err){
    // generate 12 sons
    async.times(12, function(m, next_son){
      var son = ...
      son.save(function(err){
        next_son(err);
      });
    }, function(err, sons){
      next_father(err);
    });
  });
}, function(err, fathers) {
  // it's done
});

这样，您可以确定所有步骤都已完成。

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。