如何在spring mongodb中检索数组中的匹配元素？

Question

我试图检索具有特定“_id”的文档和具有另一个特定“_id”的单个嵌入文档。

我的文档代表一个目录，它包含一系列课程。

示例数据：

'_id': ObjectId('1111'),
'name': 'example catalog',
...
...
'courses': [
     { 
         '_id': ObjectId('2222'),
         'name': 'my course',
         ...
     },
     {
         ....
     }

在 mongod 我运行这个聚合查询，并得到我想要的：

db.getCollection('catalogs').aggregate(
{ $match: { '_id': ObjectId('58e8da206ca4f710bab6ef74') } },
{ $unwind: '$courses' },
{ $match: { 'courses._id': ObjectId('58d65541495c851c1703c57f') } })

正如我之前提到的，我已经恢复了单个目录实例，其中包含单个课程实例。

在我的 java repo 中，我试图做同样的事情：

    Aggregation aggregation = Aggregation.newAggregation(
            Aggregation.match(Criteria.where(Catalog.ID_FIELD).is(catalogId)),
            Aggregation.unwind(Catalog.COURSES_FIELD, true),
            Aggregation.match(Criteria.where(Catalog.COURSES_FIELD + '.' + Course.ID_FIELD).is(embeddedCourseId))
    );
    AggregationResults<Catalog> results = mongoTemplate.aggregate(aggregation,
            Catalog.class, Catalog.class);

    List<Catalog> catalog  = results.getMappedResults();

但不幸的是，我有一个“示例目录”的实例，其中包含空的课程数组。

在调试时，我发现里面results有两个返回的 props。第一个是我使用过的，称为mappedResults（代表从 mongoDB 返回的转换对象） - 包含一个空的课程数组。另一个是rawResults,（将数据表示为DBObject） -包含我查询的特定课程

我的 Catalog 类包含一个 ArrayList（如果有什么不同的话）。

请帮助并让我知道我应该怎么做才能正确转换结果，或者我在代码中做错了什么。

Answer 1

您可以尝试以下选项。关键是在映射响应时保留结构。

常规查询：

使用$positional投影

Query query = new Query();
query.addCriteria(Criteria.where("id").is(new ObjectId("58e8da206ca4f710bab6ef74")).and("courses.id").is(new ObjectId("58d65541495c851c1703c57f")));
query.fields().include("name").position("courses", 1);
List<Course> courses = mongoTemplate.find(query, Course.class);

使用$elemMatch投影

Query query = new Query();
query.addCriteria(Criteria.where("id").is(new ObjectId("58e8da206ca4f710bab6ef74")));
query.fields().include("name").elemMatch("courses", Criteria.where("_id").is(new ObjectId("58d65541495c851c1703c57f") ) );
List<Course> Course = mongoTemplate.find(query, Course.class);

聚合

Mongo 版本 >= 3.4 & Spring 1.5.2 引导/Spring 1.10.1 Mongo。

您可以使用$addFields阶段，它将courses用$filter值覆盖该字段，同时保留所有现有属性。我addFields在当前的 spring 版本中找不到任何构建器。所以我必须用它AggregationOperation来创建一个新的。

AggregationOperation addFields = new AggregationOperation() {
    @Override
    public DBObject toDBObject(AggregationOperationContext aggregationOperationContext) {
        DBObject dbObject =
                new BasicDBObject("courses",
                        new BasicDBObject("$filter",
                                new BasicDBObject("input", "$$courses").
                                        append("as", "course").
                                        append("cond",
                                            new BasicDBObject("$eq", Arrays.<Object>asList("$$course._id", new ObjectId("58d65541495c851c1703c57f"))))));
        return new BasicDBObject("$addFields", dbObject);
    }
};

Aggregation aggregation = Aggregation.newAggregation(
            Aggregation.match(Criteria.where("_id").is(new ObjectId("58e8da206ca4f710bab6ef74"))),
            addFields
 );

Mongo 版本 = 3.2 & Spring 1.5.2 Boot / Spring 1.10.1 Mongo。.

这个想法仍然与上面相同，但此管道使用$project因此您必须添加要保留在最终响应中的所有字段。还使用 spring 辅助方法来创建$filter管道。

Aggregation aggregation = newAggregation(
     Aggregation.match(Criteria.where("id").is(new ObjectId("58e8da206ca4f710bab6ef74"))),
     Aggregation.project("name")
                 .and(ArrayOperators.Filter.filter("courses").as("course")                          
                 .by(ComparisonOperators.Eq.valueOf("course._id").equalToValue(new ObjectId("58d65541495c851c1703c57f")))
                    ).as("courses")
 );

Mongo 版本 <= 2.6

您必须使用$unwind并添加一个course字段才能让 spring 正确映射它。