Действительно динамический JPA CriteriaBuilder

Мне нужно создать "реальный" динамический JPA CriteriaBuilder. Я получаю Map<String, String> с инструкциями. Это выглядит так:

name : John
surname : Smith
email : [email protected]

...more pairs possible

Вот что я реализую:

CriteriaBuilder cb = em.getCriteriaBuilder();
CriteriaQuery<User> query = cb.createQuery(User.class);
Root<User> userRoot = query.from(User.class);
query.select(userRoot);

List<Predicate> predicates = new ArrayList<Predicate>();
Iterator<String> column = statements.keySet().iterator();
while (column.hasNext()) {

    // get the pairs
    String colIndex = column.next();
    String colValue = statements.get(colIndex);

    // create the statement
    Predicate pAnd = cb.conjunction();
    pAnd = cb.and(pAnd, cb.equal(userRoot.get(colIndex), colValue));
    predicates.add(pAnd);
}

// doesn't work, i don't know how many predicates i have -> can not address them
query.where(predicates.get(0), predicates.get(1), ...);

// doesn't work, because it is a list of predicates
query.where(predicates);

// doesn't work, because the actual predicate overwrites the old predicate
for (Predicate pre : predicates) {
     query.where(pre)
}

Я попытался построить большой Predicate, который содержит все остальные предикаты и добавит его в query.where(), но снова предикаты перезаписывают старые значения. Похоже, нет возможности добавить Predicate вместо изменения предиката: - (

Реальный проект еще сложнее, потому что для некоторых пар требуется equal и еще один a like. И этого недостаточно: может быть добавлен дополнительный оператор с or, похожим на type : 1;4;7. Здесь значение должно разделиться и создать оператор вроде:

<rest of statement> AND (type = 1 OR type = 4 OR type = 7)

ОБНОВЛЕНИЕ И РЕШЕНИЕ Получил два списка, первый список для AND работает хорошо. Второй список содержит операторы OR, такие как exspected:

final List<Predicate> andPredicates = new ArrayList<Predicate>();
final List<Predicate> orPredicates = new ArrayList<Predicate>();
for (final Entry<String, String> entry : statements.entrySet()) {
    final String colIndex = entry.getKey();
    final String colValue = entry.getValue();
    if (colIndex != null && colValue != null) {

        if (!colValue.contains(";")) {
            if (equals) {
                andPredicates.add(cb.equal(userRoot.get(colIndex), colValue));
            } else {
                andPredicates.add(cb.like(userRoot.<String> get(colIndex), "%" + colValue + "%"));
            }
        } else {
            String[] values = colValue.split(";");
            for (String value : values) {
                orPredicates.add(cb.or(cb.equal(userRoot.get(colIndex), value)));
            }
        }       
    }
}

// Here goes the magic to combine both lists
if (andPredicates.size() > 0 && orPredicates.size() == 0) {
    // no need to make new predicate, it is already a conjunction
    query.where(andPredicates.toArray(new Predicate[andPredicates.size()]));
} else if (andPredicates.size() == 0 && orPredicates.size() > 0) {
    // make a disjunction, this part is missing above
    Predicate p = cb.disjunction();
    p = cb.or(orPredicates.toArray(new Predicate[orPredicates.size()]));
    query.where(p);
} else {
    // both types of statements combined
    Predicate o = cb.and(andPredicates.toArray(new Predicate[andPredicates.size()]));
    Predicate p = cb.or(orPredicates.toArray(new Predicate[orPredicates.size()]));
    query.where(o, p);
}

query.where(predicates.toArray(new Predicate[predicates.size()]));
users = em.createQuery(query).getResultList();

Ответы

Ответ 1

Вы можете передать массив предикатов в CriteriaBuilder, выбрав equal или like по мере продвижения. Для этого создайте список и упакуйте содержимое списка в массив в одном выражении and. Вот так:

final List<Predicate> predicates = new ArrayList<Predicate>();

for (final Entry<String, String> e : myPredicateMap.entrySet()) {

    final String key = e.getKey();
    final String value = e.getValue();

    if ((key != null) && (value != null)) {

        if (value.contains("%")) {
            predicates.add(criteriaBuilder.like(root.<String> get(key), value));
        } else {
            predicates.add(criteriaBuilder.equal(root.get(key), value));
        }
    }
}

query.where(criteriaBuilder.and(predicates.toArray(new Predicate[predicates.size()])));
query.select(count);

Если вам нужно distingiush между and и or, используйте два списка.

Ответ 2

Один из вариантов заключается в использовании того факта, что метод с переменным числом аргументов может принимать массив:

query.where(predicates.toArray(new Predicate[predicates.size()]));

В качестве альтернативы вы можете объединить их в один предикат (обратите внимание, что если вы этого не сделаете, вам не нужно создавать конъюнкцию, как в вашем примере);

Predicate where = cb.conjunction();
while (column.hasNext()) {
    ...
    where = cb.and(where, cb.equal(userRoot.get(colIndex), colValue));
}

query.where(where);

Ответ 3

Я всегда думал, что создание такого решения похоже на повторное использование велосипеда. Здесь https://github.com/sasa7812/jfbdemo - мое решение. Он был протестирован на EclipseLink и Hibernate (EclipseLink в производстве, мы использовали его в нескольких проектах для простых случаев). Иногда вам просто нужно быстрое решение для создания dataTable с сортировкой и фильтрацией, ничего необычного. Он способен фильтровать и сортировать по объединенным полям и даже по коллекциям. Проект содержит демонстрацию на Primefaces, показывающую возможности FilterCriteriaBuilder. В основе всего этого вам просто нужно:

 public List<T> loadFilterBuilder(int first, int pageSize, Map<String, Boolean> sorts, List<FieldFilter> argumentFilters, Class<? extends AbstractEntity> entityClass) {
    FilterCriteriaBuilder<T> fcb = new FilterCriteriaBuilder<>(getEntityManager(), (Class<T>) entityClass);
    fcb.addFilters(argumentFilters).addOrders(sorts);
    Query q = getEntityManager().createQuery(fcb.getQuery());
    q.setFirstResult(first);
    q.setMaxResults(pageSize);
    return q.getResultList();
}

чтобы получить результаты из базы данных.

Мне действительно нужно, чтобы кто-то сказал мне, что он полезен и используется где-то, чтобы продолжить мою работу над этой библиотекой.

Ответ 4

Для Predicate вы можете создать новый ArrayList, чтобы добавить весь предикат в список, а затем в конце вы можете создать предикатный массив и преобразовать список в предикатный массив и напрямую назначить его критерию. where (new Pridicate [])

Ответ 5

Вы можете преобразовать в Предикат... следующим образом

query.where( predicates.stream().toArray( Predicate[]::new ) );