Выбор общего первичного ключа с помощью CriteriaQuery

Question

При миграции с Hibernate Criteria api на CriteriaQuery я столкнулся с универсальным DAO для абстрактного класса, у которого есть положение в общем поле, но выполняется выборка по их идентификатору, даже если идентификаторы совершенно разные для каждого класса.

Старая проекция выглядит следующим образом

criteria.setProjection(Projections.id());

Есть ли способ сделать это аналогичным образом с CriteriaQuery?

Редактировать: Полный код критерия

DetachedCriteria detachedCriteria = DetachedCriteria.forClass(MyEntity.class);
detachedCriteria.add(Restrictions.in("accountID", accounts));
detachedCriteria.setProjection(Projections.id());

EntityManager em = ...;
Criteria criteria = detachedCriteria.getExecutableCriteria((Session) em.getDelegate());
List<Integer> list = criteria.list();

Answer 1

Объединение ответов:

https://stackoverflow.com/a/16911313

https://stackoverflow.com/a/47793003

Я создал этот метод:

public String getIdAttribute(EntityManager em, String fullClassName) {
    Class<? extends Object> clazz = null;
    try {
        clazz = Class.forName(fullClassName);
    } catch (ClassNotFoundException e) {
        e.printStackTrace();
    }
    Metamodel m = em.getMetamodel();
    IdentifiableType<T> of = (IdentifiableType<T>) m.managedType(clazz);
    return of.getId(of.getIdType().getJavaType()).getName();
}

, тогда у меня есть менеджер сущностейinjected

@PersistenceContext
private EntityManager em;

Я получаю первичный ключ корневого объекта следующим образом:

String rootFullClassName = root.getModel().getJavaType().getName();
String primaryKeyName = getIdAttribute(em, rootFullClassName);

и получаю первичные ключи, на которые ссылаются такие атрибуты:

return (Specification<T>) (root, query, builder) -> {
    Set<Attribute<? super T, ?>> attributes = root.getModel().getAttributes();
    for (Attribute a: attributes) {
        if(a.isAssociation()) { 
            Path rootJoinGetName = root.join(a.getName());
            String referencedClassName = rootJoinGetName.getJavaType().getName();
            String referencedPrimaryKey = getIdAttribute(em, referencedClassName);
            //then I can use it to see if it is equal to a value (e.g 
            //filtering actors by movies with id = 1 - in 
            //this case referencedPrimaryKey is "id")
            Predicate p = rootJoinGetName.get(referencedPrimaryKey).in(1);
        }
    }
}

Таким образом, мне не нужно заранее знать тип первичного ключа / ссылочного ключа, так как он может быть получен с помощью метамодели Entity Manager.Приведенный выше код можно использовать с CriteriaQuery, а также со спецификациями.

Answer 2

Мне удалось найти его самостоятельно.

criteriaQuery.select(
    root.get(entityRoot.getModel().getDeclaredId(int.class))
);