JPA CriteriaBuilder - Как использовать оператор сравнения "IN" - PullRequest
Новая
       29

JPA CriteriaBuilder - Как использовать оператор сравнения "IN"

48 голосов
/ 17 февраля 2012

Не могли бы вы мне помочь, как преобразовать следующие коды в оператор "in" построителя критериев?Мне нужно отфильтровать, используя список / массив имен пользователей, используя "in".

Я также пытался искать с помощью метода JPA CriteriaBuilder - "in", но не могу найти хороший результат.Поэтому я был бы очень признателен, если бы вы могли дать мне ссылки на эту тему.Спасибо.

Вот мои коды: 

//usersList is a list of User that I need to put inside IN operator 

CriteriaBuilder builder = getJpaTemplate().getEntityManagerFactory().getCriteriaBuilder();
CriteriaQuery<ScheduleRequest> criteria = builder.createQuery(ScheduleRequest.class);

Root<ScheduleRequest> scheduleRequest = criteria.from(ScheduleRequest.class);
criteria = criteria.select(scheduleRequest);

List<Predicate> params = new ArrayList<Predicate>();

List<ParameterExpression<String>> usersIdsParamList = new ArrayList<ParameterExpression<String>>();

for (int i = 0; i < usersList.size(); i++) {
ParameterExpression<String> usersIdsParam = builder.parameter(String.class);
params.add(builder.equal(scheduleRequest.get("createdBy"), usersIdsParam) );
usersIdsParamList.add(usersIdsParam);
}

criteria = criteria.where(params.toArray(new Predicate[0]));

TypedQuery<ScheduleRequest> query = getJpaTemplate().getEntityManagerFactory().createEntityManager().createQuery(criteria);

for (int i = 0; i < usersList.size(); i++) {
query.setParameter(usersIdsParamList.get(i), usersList.get(i).getUsername());
}

List<ScheduleRequest> scheduleRequestList = query.getResultList();

Внутренняя строка запроса преобразуется в приведенную ниже, поэтому я не получаю записи, созданные двумя пользователями, потому что онаиспользует «И». 

select generatedAlias0 from ScheduleRequest as generatedAlias0 where ( generatedAlias0.createdBy=:param0 ) and ( generatedAlias0.createdBy=:param1 ) order by generatedAlias0.trackingId asc

Ответы [ 2 ]

86 голосов
/ 17 февраля 2012

Если я правильно понимаю, вы хотите присоединиться к ScheduleRequest с User и применить условие in к свойству userName сущности User.

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

List<String> myList = new ArrayList<String> ();
for (User u : usersList) {
    myList.add(u.getUsername());
}
Expression<String> exp = scheduleRequest.get("createdBy");
Predicate predicate = exp.in(myList);
criteria.where(predicate);

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

Expression<String> exp = scheduleRequest.get("createdBy");

с этим: 

Expression<String> exp = scheduleRequest.get(ScheduleRequest_.createdBy);

Если это работает, то вы можете попытаться добавить логику Join в Criteria Query. Но сейчас я не могу это проверить, поэтому предпочитаю посмотреть, не хочет ли кто-нибудь еще попробовать.

14 голосов
/ 17 февраля 2012

Не идеальный ответ, хотя могут помочь фрагменты кода. 

public <T> List<T> findListWhereInCondition(Class<T> clazz,
            String conditionColumnName, Serializable... conditionColumnValues) {
        QueryBuilder<T> queryBuilder = new QueryBuilder<T>(clazz);
        addWhereInClause(queryBuilder, conditionColumnName,
                conditionColumnValues);
        queryBuilder.select();
        return queryBuilder.getResultList();

    }


private <T> void addWhereInClause(QueryBuilder<T> queryBuilder,
            String conditionColumnName, Serializable... conditionColumnValues) {

        Path<Object> path = queryBuilder.root.get(conditionColumnName);
        In<Object> in = queryBuilder.criteriaBuilder.in(path);
        for (Serializable conditionColumnValue : conditionColumnValues) {
            in.value(conditionColumnValue);
        }
        queryBuilder.criteriaQuery.where(in);

    }
...