java - 使用 MongoDB java 驱动程序进行聚合

Question

我正在使用 MongoDB java 驱动程序 3.2.2 来执行一些聚合操作，但我不确定是否可以通过它实现某些功能。

MongoDB 中的原始查询是:

db.getCollection('report').aggregate({
$group: {
    _id: "$company_id",
    count: {
            $sum: {
                    $cond: [{
                            $eq: ["$idcard.status", "normal"]
                        },0,1]
                }
            }
    }
})

我不知道如何在下面的代码部分中将“$cond”作为 Java 驱动程序中“$sum”运算符的参数:

        AggregateIterable<Document> res = col.aggregate(Arrays.asList(
            group("$company_id",
                    sum("count", ...)
            )));

我搜索了official document没有结果，有人有这样做的经验吗？谢谢。

Answer 1

对于 3.x 驱动程序

使用 BsonDocument:类型安全版本

BsonArray cond = new BsonArray();
BsonArray eq = new BsonArray();
eq.add(new BsonString("$idcard.status"));
eq.add(new BsonString("normal"));
cond.add(new BsonDocument("$eq", eq));
cond.add(new BsonInt64(0));
cond.add(new BsonInt64(1));

AggregateIterable<BsonDocument> aggregate = dbCollection.aggregate(Arrays.asList(
        group("$company_id",
                sum("count", new BsonDocument("$cond", cond))
    )));

使用文档 - 代码较少但类型不安全

List cond = new ArrayList();
cond.add(new Document("$eq", Arrays.asList("$idcard.status", "normal")));
cond.add(0);
cond.add(1);

AggregateIterable<Document> aggregate = dbCollection.aggregate(Arrays.asList(
        group("$company_id",
                sum("count", new Document("$cond", cond))
        )));