无法将聚合查询从 mongodb 转换为 Spring Data MongoDb

Question

我一直在尝试使用聚合对象将mongodb中的查询转换为Spring Data MongoDb时遇到麻烦。我在 mongo 中有以下文档：

{
"_id" : ObjectId("596ce468798b61179c6442bb"),
"_class" : "com.test.model.User",
"name" : "Oi",
"surName" : "Alo",
"workLogs" : [
{
"_id" : ObjectId("596ce468798b61179c6442bc"),
"day" : 1,
"month" : 1,
"year" : 2017,
"timeEntrance" : "8:00",
"lunchLeave" : "12:00",
"lunchBack" : "13:00",
"timeLeave" : "18:00"
},
{
"_id" : ObjectId("596ce468798b61179c6442bd"),
"day" : 2,
"month" : 1,
"year" : 2017,
"timeEntrance" : "8:00",
"lunchLeave" : "12:00",
"lunchBack" : "13:00",
"timeLeave" : "18:00"
}
]
}

我想查询同一年和同一月的所有工作日志，之后我只想获取工作日志数组作为结果。我设法使用此查询使用 mongo 做到这一点：

db.user.aggregate([
{$unwind: '$workLogs'}, 
{$match: {'workLogs.month':2, 'workLogs.year':2017}}, 
{$group: {_id:'$_id', workLogs:{$push: '$workLogs'}}}, 
{$project: {'_id':0, 'workLogs': 1}}
]).pretty()

但是我找不到如何将此查询转换为Spring Data MongoDb，我想我快到了，如果有人可以帮助我，我将不胜感激。这是我在java中使用的代码。

Aggregation agg = Aggregation.newAggregation(
unwind("workLogs"),
match(Criteria
.where("_id").is(userId)
.and("workLogs.month").is(1)
.and("workLogs.year").is(2017)
),
group("_id"),
group("horarios")
.push(new BasicDBObject("workLogs", "workLogs")).as("workLogs"),
project("workLogs")
);
AggregationResults<WorkLog> results = mongoTemplate.aggregate(agg, "workLogs", WorkLog.class);

提前谢谢大家！

Answer 1

不知道为什么你的java代码中有所有额外的字段。

用于 shell 查询的 java 等效代码是

Aggregation agg = Aggregation.newAggregation(
unwind("workLogs"),
match(Criteria
.where("workLogs.month").is(1)
.and("workLogs.year").is(2017)
),
group("_id").push("workLogs").as("workLogs"),
project("workLogs").andExclude("_id")
);

或者，您可以简化代码以使用$filter。

import static org.springframework.data.mongodb.core.aggregation.Aggregation.newAggregation;
import static org.springframework.data.mongodb.core.aggregation.Aggregation.project;
import static org.springframework.data.mongodb.core.aggregation.ArrayOperators.Filter.filter;
import static org.springframework.data.mongodb.core.aggregation.BooleanOperators.And.and;
import static org.springframework.data.mongodb.core.aggregation.ComparisonOperators.Eq;
Aggregation agg = newAggregation(project().
and(
filter("workLogs").
as("workLog").
by(
and(
Eq.valueOf("workLog.month").equalToValue(1), 
Eq.valueOf("workLog.year").equalToValue(2017)
)
)
).as("workLogs").
andExclude("_id")
);