如何在Azure Cosmos DB for MongoDB中用Spring Data按DBRef筛选？

问题背景 #

使用Azure Cosmos DB for MongoDB，cars和parkings集合通过Spring Data的@DBRef关联，但调用findByParkingId时，Spring Data生成的原生MongoDB DBRef格式查询（{"parking": {"$ref": "parkings", "$id": "xxx"}}）无法在Cosmos DB中匹配数据，仅当使用{"parking._id": "xxx"}格式的查询才能正常返回结果。

可行解决方案 #

1. 自定义@Query注解生成目标查询条件 #

直接在Repository方法上通过@Query指定符合Cosmos DB要求的查询语句，绕过Spring Data自动生成的DBRef格式查询：

@Repository
interface CarMongoRepository : MongoRepository<CarDocument, String> {
    @Query("{ 'parking._id' : ?0 }")
    fun findByParkingId(id: String): List<CarDocument>
}

这种方式无需修改实体类，直接强制生成parking._id的查询条件，适配Cosmos DB的查询规则。

2. 调整实体类结构，分离关联ID与关联对象 #

为避免重复存储大型ParkingDocument实体，同时兼容Cosmos DB的查询逻辑，可修改实体类：

• 新增parkingId字段存储停车场ID（持久化到数据库）
• 原parking字段标记为@Transient，不持久化到数据库，仅用于业务逻辑中的关联操作

修改后的实体类代码：

@Document("cars")
data class CarDocument(
    @Id
    val id: String,
    val parkingId: String?, // 存储停车场ID，持久化到数据库
    @Transient // 标记为不持久化，避免存储大型对象
    val parking: ParkingDocument?
)

@Document("parkings")
data class ParkingDocument(
    @Id
    val id: String
)

对应的Repository方法可直接使用findByParkingId，此时Spring Data会生成{"parkingId": "xxx"}的查询条件，完全适配Cosmos DB，同时避免了重复存储大型实体的问题。需要注意的是，业务逻辑中需手动维护parkingId和parking对象的关联关系（比如查询车辆后，根据parkingId手动查询停车场信息）。

3. 检查Cosmos DB MongoDB API兼容性设置 #

Azure Cosmos DB的MongoDB API对部分MongoDB原生特性支持有限，可尝试以下操作：

• 确认Cosmos DB账户的MongoDB API版本（建议使用4.0+版本）
• 在Cosmos DB账户的连接字符串或API设置中，检查是否开启了与DBRef相关的兼容选项（注：此选项不一定存在，具体以Azure控制台为准）
  不过此方法可行性较低，因为Cosmos DB本身对DBRef的支持并不完善，优先推荐前两种方案。

内容的提问来源于stack exchange，提问作者Eduard