Question

我定义了一种使用聚合框架对数据进行分组的方法。该方法接受可以对数据进行分组的字段列表。我还创建了一个类来映射结果，但问题是我不知道为_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;

}

使用空列表，结果（方法返回的List的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$0, 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参数

1 个答案: