Mongo:如何通过java的mongoTemplate在一个聚合中计算几个数组

时间:2016-05-24 09:13:25

标签: java mongodb aggregation-framework mongotemplate

我必须关注Db数据:

{user : Tom, CORRECT: {q1, q3}, WRONG : {q2, q4} },
{user : jim, CORRECT: {q1}, WRONG : {q2, q3, q4} },
{user : Tom, CORRECT: {q6}, WRONG : {7} },

我想使用聚合来计算每个用户每个CORRECT \ WRONG的数量,即

{user : Tom, correctCount : 3, wrongCount : 3},
{user : jim, correctCount : 1, wrongCount : 3},

我尝试过的是:

    Aggregation agg = newAggregation(
            group("name").
                    addToSet(correct).as(correct).
                    addToSet(wrong).as(wrong).
                    addToSet(partial).as(partial)
    );

但是对于每个用户,我得到了完整的数据列表(即q1,q2,q3 ......),我总是可以在该列表上执行size() - 但它不合适。我怎样才能获得计数值?

谢谢

1 个答案:

答案 0 :(得分:0)

您可以采用的一种方法是使用 $size中的 $project 运算符创建一个具有这些数组大小的额外字段管道步骤,然后将文档分组到 $group 管道中,以获得累积总和作为新大小字段的计数:

Mongo shell:

db.collection.aggregate([
    {
        "$project": {
            "user": 1,
            "correctSize": { "$size": "$CORRECT" },
            "wrongSize": { "$size": "$WRONG" }
        }
    },
    {
        "$group": {
            "_id": "$user",
            "correctCount": { "$sum": "$correctSize" },
            "wrongCount": { "$sum": "$wrongSize" }
        }
    }
])

Java:在项目步骤中使用 SpEL andExpression 来使用 $size expression

import static org.springframework.data.mongodb.core.aggregation.Expressions.*; //new
...
Aggregation agg = newAggregation(
    project("user")
        .andExpression(expression("size", field("CORRECT"))).as("correctSize");
        .andExpression(expression("size", field("WRONG"))).as("wrongSize");
    group("user")
        .sum("correctSize").as("correctCount")
        .sum("wrongSize").as("wrongCount")
);