Spring Boot 数据和 MongoDB - 过滤子文档数组查询

Question

我正在尝试使用 Spring 来查询 Mongo 存储库并过滤数组子文档。我已经引用了how to filter array in subdocument with mongodb，但想知道是否有更合适的或 java 结构化的方法可以使用 Spring 来做到这一点。

我目前正在使用速记存储库接口表示法，但我正在返回未过滤数组的完整文档。

PersonRepository.java

@Repository
public interface PersonRepository extends MongoRepository <Person, String> {
    List<Person> findByAddressZipCode(@Param("zip") int zip);
}

Person.java

@Document
public class Person {
    @Id
    private String id;

    private String firstName;
    private String lastName;
    private List<Address> address;
}

Address.java

public class Address {
    private int zip;
}

示例输入

{
 "firstName":"George",
 "lastName":"Washington",
 "address":[{
     "zip":"12345"
  },{
     "zip":"98765"
  },{
     "zip":"12345"
  }]
}

预期输出

{
 "firstName":"George",
 "lastName":"Washington",
 "address":[{
     "zip":"12345"
  },{
     "zip":"12345"
  }]
}

实际输出

{
 "firstName":"George",
 "lastName":"Washington",
 "address":[{
     "zip":"12345"
  },{
     "zip":"98765"
  },{
     "zip":"12345"
  }]
}

Answer 1

嗯，在 Spring Data 中，这种查询不是trivial。

坏消息：
Spring Data Repository 没有MongoDB Aggregation 的解决方案。因此，您不能在 MongoRepository 中实现任何方法，例如 aggregateBy...

好消息：
Spring Data 提供了MongoTemplate 类，它允许您执行复杂的查询，就像您在标准 MongoDB shell 中所做的那样。

所以，由于您只想exclude 子文档不匹配某些条件，我们需要定义聚合pipelines。

我假设：

zip codes are Numeric (In your example is string)
And, to exclude subdocument, we filter by `zip`
There is no any other filter

MongoDB 聚合将是：

db.person.aggregate([
    {$unwind: "$address"},
    {$match: {"address.zip": 12345}},
    {$group: { _id: { "firstName":"$firstName", "lastName":"$lastName", _id:"$_id" }, address: { $push: "$address" } } },
    {$project: {_id:0, "firstName":"$_id.firstName", "lastName":"$_id.lastName", "address": "$address"}}
])

如果所有过滤器都成功，我们得到：

[ 
    {
        "address" : [ 
            {
                "zip" : 12345
            }, 
            {
                "zip" : 12345
            }
        ],
        "firstName" : "George",
        "lastName" : "Washington"
    }
]

---

现在，以 Spring Data 方式，您需要在项目中添加一些更改：

首先，找到您需要添加的mongo-config.xml：

<!-- Define the mongoDbFactory with your database Name  -->
<mongo:db-factory uri="mongodb://user:pass@localhost:27017/db"/>

<!-- Define the MongoTemplate  -->
<bean id="mongoTemplate" class="org.springframework.data.mongodb.core.MongoTemplate">
    <constructor-arg name="mongoDbFactory" ref="mongoDbFactory" />
</bean>

MongoTemplate 是 Spring 的 MongoDB 支持的中心类，提供与数据库交互的功能集。模板... 提供您的域对象 和MongoDB 文档 之间的映射。 More info

其次，在您的@Service 类中，添加以下代码以在@PostConstruct 中加载

@Autowired
private MongoOperations mongoOperations;

...

public List<Person> findByAddressZipCode(int zip) {

    List<AggregationOperation> list = new ArrayList<AggregationOperation>();
    list.add(Aggregation.unwind("address"));
    list.add(Aggregation.match(Criteria.where("address.zip").is(zip)));
    list.add(Aggregation.group("firstName", "lastName").push("address").as("address"));
    list.add(Aggregation.project("firstName", "lastName", "address"));
    TypedAggregation<Person> agg = Aggregation.newAggregation(Person.class, list);
    return mongoOperations.aggregate(agg, Person.class, Person.class).getMappedResults();
}

注意：Person 和 Address 都应该有默认的空构造函数！