Spring Data JPA：使用可选参数调用存储库方法的有效方法

Question

下面的Java11方法由控制器调用，其中ID是必需的参数，status、version是可选参数。我必须编写多个repository方法来根据这些参数获取记录。我想知道有没有一个更好的/有效的方法来重构这个方法，用if/else梯子？

    @Override
    @Transactional(transactionManager = "customTransactionManager")
    public Optional<String> getInformation(UUID id, Status status, Long version) {
        try {
            Preconditions.checkNotNull(id, ID_MUST_BE_NOT_NULL_MSG);
            if (status != null && version != null) {
                return repository.findByIdAndVersionAndStatus(id, version, status);
            } else if (status != null) {
                return repository.findFirstByIdAndStatus(id, status);
            } else if (version != null) {
                return repository.findFirstByIdAndVersion(id, version);
            } else {
                return repository.findFirstByIdOrderByIdDesc(id);
            }
        } catch (Exception e) {
            log.error(e);
            throw new CustomException(MessageFormat.format(PUBLIC_ERROR_MESSAGE, id));
        }
    }

Answer 1

您可以为此使用规范：

private Specification<YourEntity> toSpecification(UUID id, Status status, Long version) {
    return (root, query, builder) -> {
        Set<Predicate> predicates = new HashSet<>();
        predicates.add(builder.equal(root.get("id"), id));
        if (status != null) predicates.add(builder.equal(root.get("status"), status));
        if (version != null) predicates.add(builder.equal(root.get("version"), version));
        return builder.and(predicates.toArray(Predicate[]::new));
    };
}

如果允许存储库扩展JPaspecificationExecutor，可以使用构建规范对象，如下所示：

Specification<YourEntity> specification = toSpecification(id, status, version);
Optional<YourEntity> result = repository.findOne(specification);

使用Hibernate元模型生成器时，还可以编写builder.equal(yourentity_.id,id)而不是builder.equal(root.get（“id”）,id)。