在记录上使用$ unwind后仅匹配一条记录

时间:2018-03-29 07:19:48

标签: mongodb aggregation-framework aggregation

我的记录如下

[
  {
    _id : 1, nm : 1,
    usr:[
         {id : 1, ty : 'team'}, 
         {id : 2, em : 'a'}, 
         {id : 3, em : 'b'}
       ]
  },
  {
    _id : 2, nm : 2,
    usr:[
         {id : 2, em : 'a'}, 
         {id : 3, em : 'b'}
       ]
  },
  {
    _id : 3, nm : 3,
    usr:[
         {id : 1, ty : 'team'}, 
         {id : 3, em : 'b'}
       ]
  },
  {
    _id : 4, nm : 4,
    usr:[ 
         {id : 3, em : 'b'}
       ]
  }
]
  • 在使用3查询时,我希望计数为userAndTeam = [1, 2],即记录中有usr.id as 1 or 2获取这些记录。
  • 在使用2查询时,我希望计数为userAndTeam4 = [1, 4],即记录中有usr.id as 1 or 4获取这些记录。

我尝试使用$unwind,这使得第一个案例的计数为4,因为$unwind为第一个记录创建了3个记录,而下面的查询与其中2个记录匹配。

我试过的查询:

var userAndTeam = [1, 2] //User id and team id. Record should match one of the ids.
var userAndTeam4 = [1, 4]

[{
   $unwind: '$usr'
 },
 {
   $project:{
     'count-2' : {$cond: {
                if: {$in : ['$usr.id',userAndTeam]},
                then: 1,
                else: 0
            }},
      'count-4' : {$cond: {
                if: {$in : ['$usr.id',userAndTeam4]},
                then: 1,
                else: 0
            }}
   }   
 }

输出

ID为2的用户

预期 - 'count-2':3

ID为2的用户

得到 - 'count-2':4

身份为4的用户

预期 - 'count-4':2

ID为4的用户

得到 - 'count-4':2

有人可以指导我解决这个问题并获得预期的数量吗?

1 个答案:

答案 0 :(得分:1)

您可以尝试以下聚合:

db.col.aggregate([
    {
        $unwind: "$usr"
    },
    {
        $project:{
            "userId": "$usr.id",
            "count-2": {
                $cond: {
                    if: { $in: [ "$usr.id", userAndTeam ] },
                    then: 1,
                    else: 0
                }
            },
            "count-4": {
                $cond: {
                    if: { $in: [ "$usr.id", userAndTeam4] },
                    then: 1,
                    else: 0
                }
            }
        }
    },
    {
        $group: {
            _id: "$_id",
            "count-2": { $max: "$count-2" },
            "count-4": { $max: "$count-4" }
        }
    },
    {
        $group: {
            _id: null,
            "count-2": { $sum: "$count-2" },
            "count-4": { $sum: "$count-4" }
        }
    }
])

您在代码中缺少的是,您希望每个0都有1_id,因此您需要_id分组$max 1}}如果任何元素与您的数组匹配,如果0都不匹配,则每个组的值为1