Nodejs + Mongodb:聚合后查找数据

时间:2017-02-28 11:07:41

标签: node.js mongodb mongodb-query aggregation-framework asynchronous-javascript

我是Nodejs和MongoDB的新手 以下是我的数据集示例:

{ 
  'name': ABC,
  'age':24,
  'gender':male,
  ...
}

一般来说,我想要做的是在使用数据来查找不同的数据集之前聚合数据 具体来说,我想知道不同年龄段有多少人。然后,找到每个年龄段的人(文件)并存储它们。

这是我的代码:

MongoClient.connect(url, function(err, db) {
    if(err) { 
        console.log('Unable to connect to the mongoDB server. Error:', err); 
    } else { 
        db.collection('test').aggregate(
        [
          { $group: { _id: "$age" , total: { $sum: 1 } } },
          { $sort: { total: -1 } } 
        ]).toArray(function(err, result) {
            assert.equal(err, null);
            age = [];
            for(var i in result) {
                age.push(result[i]['_id'])
            };
            ageNodes = {};
            for(var i in age) {
                 nodes = [];
                 var cursor = db.collection('test').find({'age':age[i]});
                 // query based on aggregated data
                 cursor.each(function(err,doc){
                    if(doc!=null){
                        nodes.push(doc);
                    } else {
                        console.log(age[i]);
                        ageNodes[age[i]] = nodes;
                    }
                })
            }
            res.json(ageNodes);
        });
    };
});

我期望的JSON格式:

{
  age:[different documents]
}

示例:

{
  20:[{name:A,gender:male,...},{},...],
  30:[{name:B,gender:male,...},{},...],
  ...
}

然而,我得到的是一个空洞的结果,所以我想也许它是由for循环引起的。
我不知道如何处理异步回调。

1 个答案:

答案 0 :(得分:4)

您只需运行以下管道,该管道使用 $push 在管道中添加根文档(由 $$ROOT 系统变量表示)每个年龄组的数组:

使用MongoDB 3.4.4及更高版本:

MongoClient.connect(url, function(err, db) {
    if(err) { 
        console.log('Unable to connect to the mongoDB server. Error:', err); 
    } else { 
        db.collection('test').aggregate([
            { '$group': { 
                '_id': '$age', 
                'total': { '$sum': 1 }, 
                'docs': { '$push': '$$ROOT' }
            } },
            { '$sort': { 'total': -1 } },
            { '$group': {
                '_id': null,
                'data': {
                    '$push': {
                        'k': '$_id',
                        'v': '$docs'
                    }
                }
            } },
            { '$replaceRoot': {
                'newRoot': { '$arrayToObject': '$data' }
            } }    
        ]).toArray(function(err, results) {
            console.log(results);
            res.json(results);
        });
    };
});

使用MongoDB 3.2及以下版本:

MongoClient.connect(url, function(err, db) {
    if(err) { 
        console.log('Unable to connect to the mongoDB server. Error:', err); 
    } else { 
        db.collection('test').aggregate([
            { '$group': { 
                '_id': '$age', 
                'total': { '$sum': 1 }, 
                'docs': { '$push': '$$ROOT' }
            } },
            { '$sort': { 'total': -1 } } 
        ]).toArray(function(err, results) {
            console.log(results);
            var ageNodes = results.reduce(function(obj, doc) { 
                obj[doc._id] = doc.docs
                return obj;
            }, {});
            console.log(ageNodes);
            res.json(ageNodes);
        });
    };
});