Java中的Mongo聚合：使用多个字段进行分组

Question

我正在尝试使用mongo-java-driver在Java中执行聚合操作。我已经执行了一些其他的查找操作，但是我无法正确地在Java中执行以下聚合操作：

db.I1.aggregate([
    { "$match": { "ci": 862222} },
    { "$match": { "gi": { "$ne": null } }},
    { "$group": {
        "_id": {
            "ci": "$ci",
            "gi": "$gi",
            "gn": "$gn",
            "si": "$si"
        }
    }},
    { "$group": {
        "_id": {
            "ci": "$_id.ci",
            "gi": "$_id.gi",
            "gn": "$_id.gn"
        },
        "sn": { "$sum": 1 }
    }},
    { "$sort" : { "_id.gi" : 1}}
])

我已经尝试了几种方法和技术来在Java中执行聚合操作，但是我无法正确地将组字段“ci”、“gi”、“gn”、“si”包含在coll.aggregate(asList())方法中。到目前为止，我得到的结果如下：

MongoCollection<Document> coll = mongo.getCollection("I1");

Document matchCourse = new Document("$match",
    new Document("ci", Integer.parseInt(courseid)));

Document matchGroupNotNull = new Document("$match",
    new Document("gi", new Document("$ne", null)));

List<Object> list1 = new BasicDBList();
    list1.add(new BasicDBObject("ci", "$ci"));
    list1.add(new BasicDBObject("gi", "$gi"));
    list1.add(new BasicDBObject("gn", "$gn"));
    list1.add(new BasicDBObject("si", "$si"));

Document group1 = new Document(
    "_id", list1).append("count", new Document("$sum", 1));

List<Object> list2 = new BasicDBList();
list2.add(new BasicDBObject("ci", "$_id.ci"));
list2.add(new BasicDBObject("gi", "$_id.gi"));
list2.add(new BasicDBObject("gn", "$_id.gn"));

Document group2 = new Document(
    "_id", list2).append("sn", new Document("$sum", 1));

Document sort = new Document("$sort", 
    new Document("_id.gi", 1));

AggregateIterable<Document> iterable = coll.aggregate(asList(matchCourse,
    matchGroupNotNull, group1, group2, sort));

我知道这并不正确，但我包含它是为了让你了解我正在做什么。我用多种方式搜索过谷歌并阅读了几页内容，但没有找到任何解决方案。MongoDB-Java的可用文档(1, 2) 对我来说太短，并且不包括这种情况。

我怎样才能在Java中执行该查询？任何帮助将不胜感激。

非常感谢！！

- Unai Sanchez

Answer 1

最终我找到了解决办法。我在发帖时有一些错误，因为那是在绝望的边缘进行的最后一次尝试，但最终我找到了最终的解决方案：

MongoDatabase mongo = // initialize your connection;
Document matches = new Document("$match",
    new Document("gi", new Document("$ne", null))
    .append("ci", Integer.parseInt(courseid)));

Document firstGroup = new Document("$group",
    new Document("_id",
        new Document("ci", "$ci")
        .append("gi", "$gi")
        .append("gn", "$gn")
        .append("si", "$si"))
    .append("count", new Document("$sum", 1)));

Document secondGroup = new Document("$group",
    new Document("_id",
        new Document("ci", "$_id.ci")
        .append("gi", "$_id.gi")
        .append("gn", "$_id.gn"))
    .append("ns", new Document("$sum", 1))); 

Document sort = new Document("$sort", 
    new Document("_id.gi", 1));             

List<Document> pipeline = Arrays.asList(matches, firstGroup,
    secondGroup, sort);
AggregateIterable<Document> cursor = mongo.getCollection("I1")
    .aggregate(pipeline);

for(Document doc : cursor) { // do stuff with doc }

我没有试图创建键值列表，而是将元素附加到文档中。希望对某些人有用！

- Unai Sanchez

Answer 2

这个问题很老，但是在我搜索时在谷歌上排名第一，所以如果有人正在寻找解决方案，我成功地通过以下方式完成了它。

Aggregation.group(Fields.fields()
            .and("field1")
            .and("field2"))
            .first("name")
            .`as`("name")
            .count().`as`("count")

这将产生以下的MDB查询：

   { "$group" : 
        { "_id" : 
        { "field1" : "$field1", "field2" : "$field2"}, 
        "name" : { "$first" : "$name"}, 
        "count" : { "$sum" : 1}
   }

- David Billings

---

这个API看起来是专门针对Spring Data的。 - user2521119