问Spring规范-谓词的连接
EN

Stack Overflow用户

提问于 2018-01-11 15:08:10

回答 1查看 4.7K关注 0票数 3

我需要一个功能，这将是过滤参数和建立查询。我有4个参数，因此如果我尝试为每个条件实现查询，我将不得不编写16个(2^4)实现-这不是一个好主意。

我试图使用Spring Data JPA中的接口Specification来改进我的代码，但是我不能创建谓词连接。

Specification接口的实现：

public class UserSpecification implements Specification<User> {

    private final UserSearchCriteria criteria;

    public UserSpecification(UserSearchCriteria criteria) {
        this.criteria = criteria;
    }

    @Override
    public Predicate toPredicate(Root<User> root, CriteriaQuery<?> query, CriteriaBuilder builder) {
        final List<Predicate> predicates = new ArrayList<>();
        if (criteria.getName() != null) {
            final Predicate name = builder.equal(root.<String>get("name"), criteria.getName());
            predicates.add(name);
        } else if (criteria.getSurname() != null) {
            final Predicate surname = builder.equal(root.<String>get("surname"), criteria.getSurname());
            predicates.add(surname);
        } else if (criteria.getCity() != null) {
            final Predicate city = builder.equal(root.<String>get("city"), criteria.getCity());
            predicates.add(city);
        } else if (criteria.getCountry() != null) {
            final Predicate country = builder.equal(root.<String>get("country"), criteria.getCountry());
            predicates.add(country);
        }
        return builder.and(predicates.toArray(new Predicate[predicates.size()]));
    }
}

测试不起作用：

@Test
    @Sql(
            scripts = "classpath:sql/specification.sql",
            executionPhase = Sql.ExecutionPhase.BEFORE_TEST_METHOD
    )
    public void specificationTest() {
        // given
        final UserSearchCriteria criteria = UserSearchCriteria.builder()
                .name("john")
                .surname("smith")
                .build();

        final UserSpecification specification = new UserSpecification(criteria);

        // when
        final List<User> result = userRepository.findAll(specification);
        userRepository.flush();

        // then
        assertThat(result).hasSize(3);
    }

在测试之前，我将以下用户插入到数据库：

INSERT INTO users (id, name, surname, city, country) VALUES (1, 'john', 'smith', null, null);
INSERT INTO users (id, name, surname, city, country) VALUES (2, 'john', 'smith', null, null);
INSERT INTO users (id, name, surname, city, country) VALUES (3, 'john', 'smith', null, null);
INSERT INTO users (id, name, surname, city, country) VALUES (4, 'john', 'abc', null, null);
INSERT INTO users (id, name, surname, city, country) VALUES (5, 'abcd', 'abc', null, null);
INSERT INTO users (id, name, surname, city, country) VALUES (6, 'abcd', 'abc', null, null);

结果，我得到了前四行。存储库只匹配第一个参数"name“，而不匹配"surname”。原因是什么？

(UserSearchCriteria和User有相同的字段:姓名、姓氏、城市、国家)。

jpa

spring-data

spring-data-jpa

criteria

spring

CDN&音视频通信出海专场

提供游戏出海、社交娱乐等方案，助力企业抢占出海市场

回答 1

Stack Overflow用户

回答已采纳

发布于 2018-01-11 15:21:47

您使用的是单个if-else，因此每次只匹配一个条件。

尝试将其更改为一系列独立的if：

@Override
public Predicate toPredicate(Root<User> root, CriteriaQuery<?> query, CriteriaBuilder builder) {
    final List<Predicate> predicates = new ArrayList<>();
    if (criteria.getName() != null) {
        final Predicate name = builder.equal(root.<String>get("name"), criteria.getName());
        predicates.add(name);
    }
    if (criteria.getSurname() != null) {
        final Predicate surname = builder.equal(root.<String>get("surname"), criteria.getSurname());
        predicates.add(surname);
    }
    if (criteria.getCity() != null) {
        final Predicate city = builder.equal(root.<String>get("city"), criteria.getCity());
        predicates.add(city);
    }
    if (criteria.getCountry() != null) {
        final Predicate country = builder.equal(root.<String>get("country"), criteria.getCountry());
        predicates.add(country);
    }
    return builder.and(predicates.toArray(new Predicate[predicates.size()]));
}