首页 > 解决方案 > 使用带有可为空外键的 QueryDSL 在 Spring Data JPA 存储库中进行过滤

问题描述

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

假设我们有 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 可以为空。我会注意到,这是一个人为的例子来表达我的观点,解决方案不能真的是“只是强制 FK 不为空”。

@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());但是当我输入它时它不起作用。

这会生成以下查询。第一个是“工作”过滤,我可以根据 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,因为它验证实体匹配,如果范围外键为空,它们不能。

我一直在努力让这个有条件的工作只检查时间范围的表属性,如果 FK 不为空,但我找不到使用 querydsl 的方法。任何建议/指导/代码片段都会很棒。

编辑:只是翻译成直接的 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,它没有按预期返回一行。将其从交叉连接更改为左连接最终可以正常工作。

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 谓词执行器指定左连接?这似乎可以解决我的问题!

标签: javaspringhibernatespring-data-jpaquerydsl

解决方案


尝试使用规范而不是谓词

  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")));
            }
        };
    }

推荐阅读