如何在聚合中映射 _id 参数

Question

我定义了一种使用聚合框架对数据进行分组的方法。该方法接受一个字段列表，您可以根据这些字段对数据进行分组。我还创建了一个 class 来映射结果，但问题是我不知道为 _id 属性指定什么类型。 _id 可以有不同的类型，具体取决于用于对数据进行分组的字段数。我试过 Object 但它只在用零个或一个字段分组时有效，但在使用多个字段时无效。

public List<Metric> getMetric(List<String> fields) {
    MatchOperation match = match(where("status").in("WIN", "LOSS", "PUSH").and("year").gte(2015));
    GroupOperation group = group(fields.toArray(new String[0]))
            .count().as("count")
            .avg("odd").as("averageOdd")
            .sum("profit").as("profit")
            .push("$$ROOT").as("picks");
    Aggregation aggregation = Aggregation.newAggregation(
            match,
            group
    );
    // TODO: Handle _id
    AggregationResults<Metric> aggregationResults = mongoTemplate.aggregate(aggregation, "pick", Metric.class);
    return aggregationResults.getMappedResults();
}

public class Metric {

    //    private Map<String, Object> _id;
    //    private List<Object> _id;
    private Object _id;

    private Integer count;

    private Double profit;

    private Double averageOdd;

}

使用空列表，结果（JSON 方法返回的列表的表示形式）是：

[
    {
        "_id": null,
        "count": 1997,
        "profit": -0.07237707390649417,
        "averageOdd": 2.4241016559624624,
        "yield": -0.00003624290130520489
    }
]

使用一个字段，部分结果为：

[
    {
        "_id": 2016,
        "count": 751,
        "profit": -34.484,
        "averageOdd": 2.362993342210386,
        "yield": -0.04591744340878828
    }
]

使用多个字段，部分结果为：

[
    {
        "_id": null,
        "count": 211,
        "profit": -18.961,
        "averageOdd": 3.1104597156398106,
        "yield": -0.08986255924170615
    }
]

Answer 1

我找到了一种用多个字段映射 _id 的方法。在映射对象中，您必须声明组成 _id.

的字段

_id: {
  sport: "MLB",
  year: 2016
}

在映射对象中，您必须声明字段 sport 和 year。不太方便的部分是你必须在你的映射对象中保留 _id 来处理单个字段 _id 的情况，即使它是逻辑的。

相关代码来自MongoTemplate：

class UnwrapAndReadDbObjectCallback<T> extends MongoTemplate.ReadDbObjectCallback<T> {
    public UnwrapAndReadDbObjectCallback(EntityReader<? super T, DBObject> this[=11=], Class<T> reader, String type) {
        super(reader, type, collectionName);
    }

    public T doWith(DBObject object) {
        Object idField = object.get("_id");
        if (!(idField instanceof DBObject)) {
            return super.doWith(object);
        } else {
            DBObject toMap = new BasicDBObject();
            DBObject nested = (DBObject)idField;
            toMap.putAll(nested);
            Iterator var5 = object.keySet().iterator();

            while(var5.hasNext()) {
                String key = (String)var5.next();
                if (!"_id".equals(key)) {
                    toMap.put(key, object.get(key));
                }
            }

            return super.doWith(toMap);
        }
    }
}

如何在聚合中映射 _id 参数

How to map _id parameter in aggregation

java

mongodb

spring-data-mongodb