Java Specification CriteriaBuilder 复杂查询

Posted 2023-04-14

【中文标题】Java Specification CriteriaBuilder 复杂查询

【英文标题】：Java Specification CriteriaBuilder complex query

【发布时间】：2021-04-29 09:48:06

【问题描述】：

我对规范、构建器、查询不是很有经验，我必须做一个相当复杂的查询：

select * from table where
   code in ('code1',  'code2' //codes) and
    (
        (
            date between "2020-03-23 //from" and "2020-03-30 //to"
            and
            status in ('Status1' , 'Status2' //status)
        )
        or
        (
            date between "2021-03-23" and "2021-03-30"
            and
            status in ('Status3' , 'Status4')
        )
    )

我有一个像这样的 DTO：

public class SearchCriteria 

@Embedded
private Filters filters;

@Embeddable
@Getter
@Setter
public static class Filters 
    private List<String> codes;
    private List<TimePeriod> timePeriods;


@Embeddable
@Getter
@Setter
public static class TimePeriod 
    private List<String> status;
    private StartDate startDate;


@Embeddable
@Getter
@Setter
public static class StartDate 
    private LocalDate from;
    private LocalDate to;

这对我来说很难。我正在尝试一切。我宁愿给你看一个具体的案例，以免误会。有人可以帮我吗？非常感谢！

我不需要使用规范，我只需要能够重现那个查询示例，规范似乎是最好的选择。

谢谢大家。

【参考方案1】：

我认为你在正确的轨道上，标准类看起来不错。 以下是您如何在方法中使用它来构建 JPA 标准并使用与您的实体对应的存储库执行查询：

public void query(List<String> codes, List<TimePeriod> timePeriods) 
    // build the code filter
    Specification<Table> codeSpec = (root, query, criteriaBuilder) -> 
        Path<String> codeField = root.get("code");
        var codePredicate = criteriaBuilder.in(codeField);
        codes.forEach(code -> codePredicate.value(code));
        return codePredicate;
    ;
    // iterate over the time periods
    var timePeriodSpec = timePeriods.stream().map(timePeriod -> 
        Specification<Table> dateSpec = (root, query, criteriaBuilder) -> 
            Path<LocalDate> dateField = root.get("date");
            return criteriaBuilder.between(dateField, timePeriod.startDate.from, timePeriod.startDate.to);
        ;
        Specification<Table> statusSpec = (root, query, criteriaBuilder) -> 
            Path<String> statusField = root.get("status");
            var statusPredicate = criteriaBuilder.in(statusField);
            timePeriod.status.forEach(status -> statusPredicate.value(status));
            return statusPredicate;
        ;
        // combine the date and status filter
        return dateSpec.and(statusSpec);
    )
    .reduce(Specification::or).get(); // chain the time period filters together
    
    var fullSpec = codeSpec.and(timePeriodSpec);

    var result = tableRepository.findAll(fullSpec, Pageable.unpaged());

您还需要确保您的存储库实现了 JpaSpecificationExecutor 接口，但您可能已经想通了。

【讨论】：

非常感谢您的提示，即使对我来说真正棘手的部分是通过循环浏览 timePeriods 列表来 OR 日期和状态。这部分：（“2020-03-23 //from”和“2020-03-30 //to”之间的日期和（'Status1'，'Status2' //status）中的状态）或（“2021-之间的日期” 03-23" 和 "2021-03-30" 以及 ('Status3' , 'Status4') 中的状态

我明白了。我已将该部分添加到我的答案中，希望现在更清楚。如果您能够解决您的问题，如果您能接受答案，那就太好了。谢谢！

完美！我想我爱你。非常感谢你的帮助，你很棒。