JPA特定查询WHERE子句中如何强制枚举转换器生效？

我有一个未存储在数据库中的枚举字段TaskStatus，需要用于复杂查询的WHERE子句中。查询代码如下：

@Query("SELECT t FROM Task t " +
       "WHERE " +
       "(:status is null OR GetTaskStatus(t.dueDate, t.completedDate) = :status)")
@Convert(converter = TaskStatus.Converter.class, attributeName = "status")
public Page<Task> findAll(
        @Param("status") TaskStatus status,
        Pageable pageable
);

TaskStatus枚举定义如下：

public enum TaskStatus {
    COMPLETED(2), LATE_COMPLETE(14), SCHEDULED(3), DUE_TODAY(5), LATE(7), OTHER(11);

    private final int code;

    TaskStatus(int i) {
        code = i;
    }

    @javax.persistence.Converter(autoApply = true)
    public static final class Converter implements AttributeConverter<TaskStatus, Integer> {

        @Override
        public Integer convertToDatabaseColumn(TaskStatus attribute) {
            if (attribute == null)
                return null;

            return attribute.code;
        }

        @Override
        public TaskStatus convertToEntityAttribute(Integer dbData) {
            return fromInteger(dbData);
        }

        public static TaskStatus fromInteger(Integer dbData) {
            return dbData == null || dbData == 0 ? null : Arrays.stream(values())
                    .filter(taskStatus -> taskStatus.code == dbData)
                    .findFirst()
                    .orElse(OTHER);
        }
    }
}

该枚举由自定义MySQL函数GetTaskStatus根据dueDate和completedDate实时计算得出，函数已测试可用。当前问题是Hibernate将status参数映射为枚举名称字符串（如"COMPLETED"），而GetTaskStatus返回整数，导致查询条件不匹配。我不想修改GetTaskStatus函数，如何强制Hibernate将status参数映射为整数？

方法一：直接引用枚举的code属性（推荐） #

1. 为TaskStatus枚举添加code的getter方法：

public enum TaskStatus {
    // 原有枚举值定义...

    private final int code;

    TaskStatus(int i) {
        code = i;
    }

    // 添加getter方法
    public int getCode() {
        return code;
    }

    // 原有Converter代码...
}

2. 修改JPQL查询语句，将参数替换为枚举的code属性：

@Query("SELECT t FROM Task t " +
       "WHERE " +
       "(:status is null OR GetTaskStatus(t.dueDate, t.completedDate) = :status.code)")
// 移除无效的@Convert注解，它仅适用于实体属性
public Page<Task> findAll(
        @Param("status") TaskStatus status,
        Pageable pageable
);

Hibernate会自动提取枚举的code整数值作为参数传递，与GetTaskStatus的返回值类型匹配。

方法二：手动转换参数为Integer #

如果不想修改JPQL语句，可以在Repository中封装一层，将TaskStatus参数转换为对应的整数后再传递：

public interface TaskRepository extends JpaRepository<Task, Long> {

    // 对外暴露的方法，接收TaskStatus参数
    default Page<Task> findAll(TaskStatus status, Pageable pageable) {
        Integer statusCode = status != null ? status.getCode() : null;
        return findAllByStatusCode(statusCode, pageable);
    }

    // 内部查询方法，接收Integer类型参数
    @Query("SELECT t FROM Task t " +
           "WHERE " +
           "(:statusCode is null OR GetTaskStatus(t.dueDate, t.completedDate) = :statusCode)")
    Page<Task> findAllByStatusCode(@Param("statusCode") Integer statusCode, Pageable pageable);
}

这种方式对外保持原有的接口签名，内部自动完成参数类型转换，不会影响调用方代码。

注意事项 #

• 原代码中的@Convert注解放在Repository方法上是无效的，该注解仅用于实体类的属性与数据库字段的映射，需要移除。
• 不需要修改GetTaskStatus函数，以上方法均能保证传递的参数为整数类型，与函数返回值匹配。

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