JPQL与JPA关联查询：仅返回符合条件的TimeSheetRecords子记录

你的核心问题是：JPA默认会加载实体的完整关联集合，即使你在查询中做了JOIN过滤，也只是筛选出符合条件的根实体（Project），但不会自动过滤关联的Tasks和TimeSheetRecords集合。以下是几种可行的解决方案：

方案一：修正JPQL，使用FETCH JOIN过滤关联集合 #

这是最直接的解决方案，通过JOIN FETCH明确加载关联的子集合，并在WHERE子句中添加过滤条件，同时用DISTINCT避免笛卡尔积导致的重复实体。

首先修正你的JPQL查询（原查询存在两处错误：误用不存在的task.date字段、未使用consultantId参数）：

@Query("SELECT DISTINCT project FROM ProjectEntity project " +
    "JOIN FETCH project.tasks task " +
    "JOIN FETCH task.timeSheetRecords timeSheetRecord " +
    "WHERE project.customerId = :customerId " +
    "AND timeSheetRecord.consultantId = :consultantId " +
    "AND timeSheetRecord.date BETWEEN :dateFrom AND :dateTo")
fun findByCustomerIdAndTimeRecordDateBetween(
        @Param("customerId") customerId: UUID,
        @Param("consultantId") consultantId: UUID,
        @Param("dateFrom") dateFrom: LocalDate,
        @Param("dateTo") dateTo: LocalDate,
    ): List<ProjectEntity>

关键说明： #

• JOIN FETCH：区别于普通JOIN，它会将关联的tasks和timeSheetRecords直接加载到查询结果的ProjectEntity中，而非懒加载。
• DISTINCT：多表JOIN会产生笛卡尔积（一个Project对应多个Task，每个Task对应多个符合条件的TimeSheetRecord），用DISTINCT可确保返回的Project实体不重复。
• 修正后的条件：使用timeSheetRecord.date替代错误的task.date，并加入consultantId的过滤逻辑。

方案二：使用DTO投影精确控制返回数据 #

如果你不想修改实体的关联加载行为，或者需要更灵活的字段控制，可以创建DTO类封装需要返回的数据，通过JPQL的构造函数调用实现投影：

1. 创建DTO类 #

data class ProjectDTO(
    val id: UUID,
    val name: String,
    val customerId: UUID?,
    val tasks: List<TaskDTO>
)

data class TaskDTO(
    val id: UUID,
    val name: String,
    val timeSheetRecords: List<TimeSheetRecordDTO>
)

data class TimeSheetRecordDTO(
    val id: UUID,
    val date: LocalDate,
    val hours: Int,
    val description: String?
)

2. 修改JPQL查询 #

@Query("SELECT DISTINCT new com.yourpackage.ProjectDTO(" +
    "project.id, project.name, project.customerId, " +
    "COLLECT(new com.yourpackage.TaskDTO(" +
        "task.id, task.name, " +
        "COLLECT(new com.yourpackage.TimeSheetRecordDTO(" +
            "timeSheetRecord.id, timeSheetRecord.date, timeSheetRecord.hours, timeSheetRecord.description" +
        "))" +
    "))" +
") FROM ProjectEntity project " +
"JOIN project.tasks task " +
"JOIN task.timeSheetRecords timeSheetRecord " +
"WHERE project.customerId = :customerId " +
"AND timeSheetRecord.consultantId = :consultantId " +
"AND timeSheetRecord.date BETWEEN :dateFrom AND :dateTo " +
"GROUP BY project.id, task.id")
fun findProjectsWithFilteredTimeRecords(
        @Param("customerId") customerId: UUID,
        @Param("consultantId") consultantId: UUID,
        @Param("dateFrom") dateFrom: LocalDate,
        @Param("dateTo") dateTo: LocalDate,
    ): List<ProjectDTO>

这种方式完全脱离实体的关联约束，可以精确返回你需要的数据结构，避免加载不必要的字段。

方案三：使用@Filter注解实现全局动态过滤 #

如果你的系统中多个地方需要对TimeSheetRecords进行相同条件的过滤，可以使用JPA的@Filter注解实现全局动态过滤：

1. 在实体上添加Filter配置 #

在TaskEntity的timeSheetRecords关联字段上添加注解：

@OneToMany(fetch = FetchType.LAZY)
@JoinColumn(name = "task_id", referencedColumnName = "id")
@Filter(
    name = "consultantDateFilter",
    condition = "consultant_id = :consultantId AND date BETWEEN :dateFrom AND :dateTo"
)
var timeSheetRecords: List<TimeSheetRecordEntity> = emptyList()

2. 在查询时启用Filter #

// 假设你持有EntityManager实例
entityManager.enableFilter("consultantDateFilter")
    .setParameter("consultantId", consultantId)
    .setParameter("dateFrom", dateFrom)
    .setParameter("dateTo", dateTo)

// 执行普通的按customerId查询
val projects = projectRepository.findByCustomerId(customerId)

// 查询完成后禁用Filter，避免影响其他查询
entityManager.disableFilter("consultantDateFilter")

这种方式适合需要全局复用过滤规则的场景，但需要手动管理Filter的启用和禁用。

内容的提问来源于stack exchange，提问作者Paweł Sosnowski