跨文档分组和计数？

Question

我有一个包含类似于以下格式的文档的集合：

{
  departure:{name: "abe"},
  arrival:{name: "tom"}
},
{
  departure:{name: "bob"},
  arrival:{name: "abe"}
}

并得到这样的输出：

{ 
  name: "abe",
  departureCount: 1,
  arrivalCount: 1
},
{
  name: "bob",
  departureCount: 1,
  arrivalCount: 0
},
{
  name: "tom",
  departureCount: 0,
  arrivalCount: 1
}

我可以通过查询特定数据来单独获取计数，如下所示：

db.sched.aggregate([
  {
    "$group":{
      _id: "$departure.name",
      departureCount: {$sum: 1}
    }
  }
])

但我还没有弄清楚如何将 arrival 和 departure 名称合并到一个文档中以及两者的计数。关于如何完成此任务的任何建议？

Answer 1

您应该使用$map 将您的文档分成两份，然后是$unwind 和$group..

[
{
    $project: {
        dep: '$departure.name',
        arr: '$arrival.name'

    }
},

{
    $project: {
        f: {
            $map: {
                input: {
                    $literal: ['dep', 'arr']
                },
                as: 'el',
                in : {
                    type: '$$el',
                    name: {
                        $cond: [{
                            $eq: ['$$el', 'dep']
                        }, '$dep', '$arr']
                    }
                }
            }
        }
    }
},

{
    $unwind: '$f'
}, {
    $group: {
        _id: {
            'name': '$f.name'
        },
        departureCount: {
            $sum: {
                $cond: [{
                    $eq: ['$f.type', 'dep']
                }, 1, 0]
            }
        },
        arrivalCount: {
            $sum: {
                $cond: [{
                    $eq: ['$f.type', 'arr']
                }, 1, 0]
            }
        }
    }
}, {
    $project: {
        _id: 0,
        name: '$_id.name',
        departureCount: 1,
        arrivalCount: 1
    }
}
]