使用带有可空外键的QueryDSL在Spring Data JPA存储库中进行过滤

时间:2018-12-04 21:48:22

标签: java spring hibernate spring-data-jpa querydsl

我一直遇到这个问题,我无法使用具有可为空的外键的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;
    }
}

我已经在存储库中构造了一个默认方法,以便使用querydsl来运行带谓词的findAll来构建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可以为null。我将注意到,这是一个旨在阐明我的观点的示例,解决方案不能真正是“仅强制执行FK不为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());时,我可以看到过滤器正常工作,但是当我将其插入时,它却无法工作。

这将生成以下查询。第一个是用于“工作”过滤的,其中我可以根据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 ?

对此,我几乎可以肯定这是由于快照程序cross join time_range_entity timerangee1_ where myentity0_.action_time_id=timerangee1_.id所致,因为它验证了实体是否匹配,如果范围外键为null,则它们将无法匹配。

我一直在努力尝试这种有条件的工作,如果FK不为null,但我无法找到一种使用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谓词执行器的左连接吗?这似乎可以解决我的问题!

1 个答案:

答案 0 :(得分:1)

尝试使用规范代替谓词

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