我有一个集合，其中的文档如下所示:

{
  _id: "545b9fa0dd5318a4285f7ce7",
  owner: "admin",  
  messages: [
    {
      id: "100",
      status: "sent",
      note: ""
    },
    {
      id: "100",
      status: "pending",
      note: ""
    },
    {
      id: "101",
      status: "sent",
      note: ""
    },
    {
      id: "102",
      status: "sent",
      note: ""
    },
    {
      id: "101",
      status: "done",
      note: ""
    }
  ]
}

(这只是一个简短的例子，在我的例子中子数组非常大)

我需要查询集合并获取特定文档的一些统计信息。 所以在这个例子中，如果我查询具有 id 的文档:“545b9fa0dd5318a4285f7ce7” 我应该得到这个结果:

{
   sent: 3,
   pending: 1,
   done: 1
}

我如何使用 spring mongotemplate 进行此类聚合？

最佳答案

要执行此类操作，您需要 $cond聚合框架中的运算符。 Spring Data MongoDB 还没有做到这一点，并且公共(public)领域缺少很多东西 $group甚至仅在 $project 下实现的操作.

跟踪任何 $cond 实现的问题支持在这里:

https://jira.spring.io/browse/DATAMONGO-861

对于世界其他地方，它看起来像这样:

db.collection.aggregate([
    { "$match": { "_id": ObjectId("545b9fa0dd5318a4285f7ce7") } },
    { "$unwind": "$messages" },
    { "$group": {
        "_id": "$_id",
        "sent": {
            "$sum": {
                "$cond": [
                    { "$eq": [ "$mesages.status", "sent" ] },
                    1,
                    0
                ]
            }
        },
        "pending": {
            "$sum": {
                "$cond": [
                    { "$eq": [ "$messages.status", "pending" ] },
                    1,
                    0
                ]
            }
        },
        "done": {
            "$sum": {
                "$cond": [
                    { "$eq": [ "$messages.status", "done" ] },
                    1,
                    0
                ]
            }
        }
    }}
])

为了让这种事情在 mongotemplate 聚合下工作，你需要一个扩展聚合操作的类，它可以从 DBObject 构建:

public class CustomGroupOperation implements AggregationOperation {
    private DBObject operation;

    public CustomGroupOperation (DBObject operation) {
        this.operation = operation;
    }

    @Override
    public DBObject toDBObject(AggregationOperationContext context) {
        return context.getMappedObject(operation);
    }
}

然后您可以将“$group”定义为DBObject 并在聚合管道中实现:

   DBObject myGroup = (DBObject)new BasicDBObject(
        "$group", new BasicDBObject(
            "_id","$_id"
        ).append(
            "sent", new BasicDBObject(
                "$sum", new BasicDBObject(
                    "$cond", new Object[]{
                        new BasicDBObject(
                            "$eq", new Object[]{ "$messages.status", "sent"}
                        ),
                        1,
                        0
                    }
                )
            )
        ).append(
            "pending", new BasicDBObject(
                "$sum", new BasicDBObject(
                    "$cond", new Object[]{
                        new BasicDBObject(
                            "$eq", new Object[]{ "$messages.status", "pending"}
                        ),
                        1,
                        0
                    }
                )
             )
        ).append(
            "done", new BasicDBObject(
                "$sum", new BasicDBObject(
                    "$cond", new Object[]{
                         new BasicDBObject(
                            "$eq", new Object[]{ "$messages.status", "done"}
                         ),
                         1,
                         0
                    }
                 )
            )
        )
     );


   ObjectId myId = new ObjectId("545b9fa0dd5318a4285f7ce7");

   Aggregation aggregation = newAggregation(
           match(Criteria.where("_id").is(myId)),
           unwind("messges"),
           new CustomGroupOperation(myGroup)
   );

这使您可以想出一个与上面的 shell 表示基本相同的管道。

所以目前看来，在不支持某些操作和序列的情况下，最好的情况是在 AgggregationOperation 上实现一个类。接口(interface)可以提供一个 DBObject 或以其他方式从您自己的自定义方法内部构造一个接口(interface)。

关于java - 带条件的 mongotemplate 聚合，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/26874376/