如何使用Spring JPA检索带限制条件的实体关联关系？

简述 #

实体A和B为多对多关联，需要检索A的实例及其关联的B实体特定子集（无需获取全部关联B）。

实体定义 #

public class A {
  @Id
  private Long id;

  @ManyToMany
  private List<B> bList;
}

public class B {
  @Id
  private Long id;

  @ManyToMany
  private List<A> aList;

  private Boolean somePropertyToUseWhileFiltering;
}

当前考虑的三种实现方式 #

• 方式1：检索A时获取所有关联B，再剔除不需要的部分（已否定，会加载大量冗余数据）
• 方式2：懒加载关联，分两次调用仓库：先获取不含关联B的A实例，再通过筛选条件获取目标B实例（正在使用，但担心多次数据库往返）
• 方式3：编写自定义JPQL/SQL查询，同时获取A实例及其关联的B子集（认为没必要，不想脱离ORM框架便捷性）

疑问 #

单次请求中多次调用仓库是否有问题？针对这种场景，有没有更合适的解决方案？

可行解决方案 #

1. 用JPA @Filter实现动态筛选关联

利用ORM的过滤器机制，在查询时动态过滤关联的B实体，一次查询即可获取带筛选后子集的A：

• 给B实体定义过滤器：

@Entity
@FilterDef(name = "filterBByProperty", parameters = @ParamDef(name = "propertyValue", type = Boolean.class))
@Filter(name = "filterBByProperty", condition = "some_property_to_use_while_filtering = :propertyValue")
public class B {
    // 原有字段
}

• 在A的关联上绑定过滤器：

@Entity
public class A {
    @Id
    private Long id;

    @ManyToMany
    @Filter(name = "filterBByProperty")
    private List<B> bList;
}

• 查询时启用过滤器：

entityManager.unwrap(Session.class).enableFilter("filterBByProperty")
             .setParameter("propertyValue", true);
A a = entityManager.find(A.class, aId);
// 此时a.getBList()仅包含符合条件的B实例

2. Spring Data JPA简洁自定义查询

用JOIN FETCH一次性加载A和符合条件的B，语法简单且不脱离ORM便捷性：

public interface ARepository extends JpaRepository<A, Long> {
    @Query("SELECT a FROM A a JOIN FETCH a.bList b WHERE a.id = :id AND b.somePropertyToUseWhileFiltering = :filterValue")
    A findByIdWithFilteredBs(@Param("id") Long id, @Param("filterValue") Boolean filterValue);
}

该查询仅需一次数据库请求，直接返回带目标B子集的A实例。

3. DTO投影（按需获取数据）

如果不需要完整实体，用DTO投影减少数据传输，同样一次请求完成：

• 定义DTO类：

public class AWithFilteredBsDto {
    private Long aId;
    private List<BShortDto> filteredBs;

    public AWithFilteredBsDto(Long aId, List<BShortDto> filteredBs) {
        this.aId = aId;
        this.filteredBs = filteredBs;
    }

    // Getters
}

// B的精简DTO
public class BShortDto {
    private Long id;

    public BShortDto(Long id) {
        this.id = id;
    }

    // Getter
}

• Repository查询：

@Query("SELECT new com.example.dto.AWithFilteredBsDto(a.id, " +
       "SELECT new com.example.dto.BShortDto(b.id) FROM a.bList b WHERE b.somePropertyToUseWhileFiltering = :filterValue) " +
       "FROM A a WHERE a.id = :id")
AWithFilteredBsDto findDtoByIdWithFilteredBs(@Param("id") Long id, @Param("filterValue") Boolean filterValue);

关于方式2的顾虑解答 #

单次请求多次调用仓库本身无技术问题，但会增加数据库往返次数，高并发场景下可能影响性能。上述方案均能将多次请求合并为一次，更高效。

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