MongoDB聚合将两个不同的字段合并为一个并获得计数

时间:2015-03-12 19:57:27

标签: javascript node.js mongodb mongoose

我在MongoDB中有以下数据:

[{id:3132, home:'NSH', away:'BOS'}, {id:3112, home:'ANA', away:'CGY'}, {id:3232, home:'MIN', away:'NSH'}]

是否有可能通过汇总管道获得每个团队的总游戏数量?

期望的结果:

[{team: 'NSH', totalGames: 2}, {team:'MIN', totalGames: 1}, ...}]

我可以通过两个聚合调用分别将它们分别放到自己的数组中:

[{$group: {_id: "$home", gamesLeft: {$sum: 1}}}]

[{$group: {_id: "$away", gamesLeft: {$sum: 1}}}]
所得

var homeGames = [ { _id: 'NSH', totalGames: 1 }, { _id: 'SJS', totalGames: 2 }, ...]
var awayGames = [ { _id: 'NSH', totalGames: 1 }, { _id: 'SJS', totalGames: 4 }, ...]

但我真的想让它只使用一个查询。如果不可能,将这两个结果组合成一个使用javascript的最佳方法是什么?

1 个答案:

答案 0 :(得分:2)

经过一番令人费解的事后,我找到了一种方法来使用聚合管道来完成它。结果如下:

db.games.aggregate([{
        $project: {
            isHome: { $literal: [true, false] },
            home: true,
            away: true
        }
    }, {
        $unwind: '$isHome'
    }, {
        $group: {
            _id: { $cond: { if: '$isHome', then: '$home', else: '$away' } },
            totalGames: { $sum: 1 }
        }
    }
]);

正如您所看到的,它包含三个阶段。前两个用于将每个文档复制到一个用于主队,一个用于客队。为此,项目阶段首先在每个包含true和false值的文档上创建一个新的isHome字段,然后展开阶段将拆分为包含true或false值的单独文档。

然后在小组阶段,我们让isHome字段决定是否在homeaway字段进行分组。

如果我们可以在项目步骤中创建一个包含数组team的{​​{1}}字段,那会更好,但是mongo只支持在这里添加数组文字,因此可以解决这个问题。