您可以将管道简化为
db.scenarios.aggregate([
{ $match: { bid: "build_1481711758" } },
{
$group: {
_id: "$bid",
nb: { $sum: 1 },
nbS: {
"$sum": {
"$cond": [ "$scst", 1, 0 ]
}
},
nbE: {
"$sum": {
"$cond": [ "$scst", 0, 1 ]
}
}
}
}
])
因为$cond
运算符计算一个布尔表达式以返回两个指定的返回表达式之一,并且该scst
字段默认返回一个布尔值。
$cond
如果使用通过管道支持操作员的当前 Spring Data 版本$project
,则可以将其转换为(未经测试):
import static org.springframework.data.mongodb.core.aggregation.Aggregation.*;
import static org.springframework.data.mongodb.core.aggregation.ConditionalOperators.Cond.*;
import org.springframework.data.mongodb.core.query.Criteria;
Cond operatorNbS = ConditionalOperators.when("scst").thenValueOf(1).otherwise(0);
Cond operatorNbE = ConditionalOperators.when("scst").thenValueOf(0).otherwise(1);
Aggregation agg = newAggregation(
match(Criteria.where("bid").is("build_1481711758"),
project("bid")
.and("scst")
.applyCondition(operatorNbE, field("nbE"))
.applyCondition(operatorNbS, field("nbS"))
group("bid")
.count().as("nb")
.sum("nbE").as("nbS")
.sum("nbE").as("nbE")
);
AggregationResults<Scenarios> results = mongoTemplate.aggregate(agg, Scenarios.class);
List<Scenarios> scenarios = results.getMappedResults();
如果您的 Spring Data 版本不支持此功能,解决方法是实现AggregationOperation接口以接收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 operation = (DBObject)new BasicDBObject(
"$group", new BasicDBObject(
"_id", "$bid"
)
.append( "nb", new BasicDBObject("$sum", 1) )
.append(
"nbS", new BasicDBObject(
"$sum", new BasicDBObject(
"$cond", new Object[]{ "$scst", 1, 0 }
)
)
).append(
"nbE", new BasicDBObject(
"$sum", new BasicDBObject(
"$cond", new Object[]{ "$scst", 0, 1 }
)
)
)
);
然后您可以将其用作:
Aggregation agg = newAggregation(
match(Criteria.where("bid").is("build_1481711758"),
new CustomGroupOperation(operation)
);
对于比上述执行速度更快的更灵活和性能更好的方法,请考虑运行替代管道,如下所示
db.scenarios.aggregate([
{ $match: { bid: "build_1481711758" } },
{
"$group": {
"_id": {
"bid": "$bid",
"scst": "$scst"
},
"count": { "$sum": 1 }
}
},
{
"$group": {
"_id": "$_id.bid",
"counts": {
"$push": {
"scst": "$_id.scst",
"count": "$count"
}
}
}
}
])