使用带有可空外键的 QueryDSL 在 Spring Data JPA Repository 中过滤

Filtering in Spring Data JPA Repository using QueryDSL with a nullable foreign key

我一直遇到这个问题,我无法使用具有可为空外键的 querydsl 在 table 上正确过滤。我将我的用例简化为一个非常简单的场景。

假设我们有 2 个实体,MyEntity 和 TimeRangeEntity。我的实体只有一个 ID 和一个指向 TimeRangeEntity 的外键。 TimeRangeEntity 只有开始时间和结束时间以及 ID。 BaseEntity,它们都是从中扩展而来的,仅具有使用 @Id 注释设置的 ID。

@Entity
@Table(name = "TEST_OBJECT")
public class MyEntity extends BaseEntity {
    @OneToOne(cascade = { CascadeType.MERGE, CascadeType.PERSIST })
    private TimeRangeEntity actionTime;

    public TimeRangeEntity getActionTime() {
        return actionTime;
    }

    public void setActionTime(TimeRangeEntity actionTime) {
        this.actionTime = actionTime;
    }
}


@Entity
@DiscriminatorValue("static")
public class TimeRangeEntity extends BaseEntity {

    @Column(name = "START_TIME")
    private Instant startTime;

    @Column(name = "END_TIME")
    private Instant endTime;

    public Instant getStartTime() {
        return startTime;
    }

    public void setStartTime(Instant startTime) {
        this.startTime = startTime;
    }

    public Instant getEndTime() {
        return endTime;
    }

    public void setEndTime(Instant endTime) {
        this.endTime = endTime;
    }
}

我在我的存储库中构造了一个默认方法 运行 一个带有谓词的 findAll 使用 querydsl 构建 SQL 语法

@Repository
public interface MyRepository extends JpaRepository<MyEntity, Long>, QueryDslPredicateExecutor<MyEntity> {

    default Page<MyEntity> paginateFilter(PaginationInfo info, String filter){
        int page = info.getOffset() > 0 ? info.getOffset() / info.getLimit() : 0;
        PageRequest pageRequest = new PageRequest(page, info.getLimit(), new Sort(new Sort.Order(info.getSortDirection(), info.getSortProperty())));
        return findAll(createFilterPredicate(filter, myEntity), pageRequest);
    }

    default Predicate createFilterPredicate(String filter, QMyEntity root){
        BooleanBuilder filterBuilder = new BooleanBuilder();
        filterBuilder.or(root.id.stringValue().containsIgnoreCase(filter));
        filterBuilder.or(root.actionTime.startTime.isNotNull());
        return filterBuilder.getValue();
    }
}

我还编写了一个测试,根据所提供的代码应该可以正常工作。我想做的只是根据 ID 进行过滤。需要注意的是,TimeRange 的 FK 可以为空。我会注意到这是一个人为的例子来表达我的观点,解决方案不可能是 "just enforce the FK is not null."

@RunWith(SpringRunner.class)
@DataJpaTest(showSql = false)
@ContextConfiguration(classes = TestConfig.class)
@ActiveProfiles("test")
public class MyRepositoryTest {
    @Autowired
    private MyRepository sut;

    private static final int count = 3;

    @Before
    public void setup(){
        for (int i = 0; i < count; i++){
            sut.save(new MyEntity());
        }
    }

    @Test
    public void testPaginationWithStringFilter(){
        PaginationInfo info = new PaginationInfo();
        info.setOffset(0);
        info.setLimit(10);
        info.setSortDirection(Sort.Direction.ASC);
        info.setSortProperty("id");

        Page<MyEntity> page = sut.paginateFilter(info, "1");
        assertEquals(1, page.getTotalElements());

        page = sut.paginateFilter(info, "10");
        assertEquals(0, page.getTotalElements());
    }
}

我 运行 遇到的问题是,如果 FK 为空,它不会过滤 ID。我保存时所做的只是设置 ID。我知道问题是因为当我注释掉行 filterBuilder.or(root.actionTime.startTime.isNotNull()); 时我可以看到过滤正常工作但是当我有它时它不起作用。

这会生成以下查询。第一个是 "working" 过滤,我可以在其中根据 ID 进行过滤(注释掉的行)。第二个用于包含 actionTime 的过滤。

select myentity0_.id as id2_38_, myentity0_.action_time_id as action_t3_38_ from test_object myentity0_ where lower(cast(myentity0_.id as char)) like ? escape '!' order by myentity0_.id asc limit ?

select myentity0_.id as id2_38_, myentity0_.action_time_id as action_t3_38_ from test_object myentity0_ cross join time_range_entity timerangee1_ where myentity0_.action_time_id=timerangee1_.id and (lower(cast(myentity0_.id as char)) like ? escape '!' or timerangee1_.start_time is not null) order by myentity0_.id asc limit ?

看看这个,我几乎可以肯定这是由于 snipper cross join time_range_entity timerangee1_ where myentity0_.action_time_id=timerangee1_.id 造成的,因为它验证了实体是否匹配,如果范围外键为 null,则它们不能匹配。

我一直在绞尽脑汁试图让这个条件工作只检查时间范围的 table 属性如果 FK 不为空但我找不到使用 querydsl 的方法。任何 advice/guidance/code 片段都会很棒。

编辑:直接翻译成 SQL,我得到了生成的 JPQL 的查询(翻译成这个例子,因为我将它用于真实数据):

select * from test_object cross join time_range range where test_object.action_time_id=range.id and lower(cast(test_object.id as char)) like '%1%';

FK 为空时,return 没有按预期排成一行。将其从交叉连接更改为左连接最终正常工作。

select * from test_object left join time_range on test_object.action_time_id=time_range.id where lower(cast(test_object.id as char)) like '%1%';

那么,有什么方法可以指定与 querydsl 谓词执行器的左连接吗?这似乎可以解决我的问题!

尝试使用 Specification 而不是 Predicate

  private Specification<QMyEntity> createFilterPredicate(final String filter, final QMyEntity root) {
        return new Specification<QMyEntity>() {
            @Nullable
            @Override
            public Predicate toPredicate(Root<QMyEntity> root, CriteriaQuery<?> query,
                    CriteriaBuilder criteriaBuilder) {
                Join<Object, Object> actionTime = root.join("actionTime", JoinType.LEFT);
                return criteriaBuilder.or(criteriaBuilder.like(criteriaBuilder.lower(root.get("id")), "%" + filter + "%"), criteriaBuilder.isNotNull(actionTime.get("startTime")));
            }
        };
    }