java - JPA 中的动态 COUNT 抛出 NullPointerException

Question

我的目标是创建一种方法，可以从数据库中的任何表中检索总行数。

这就是我所做的:

ENTITY_MANAGER_FACTORY = Persistence.createEntityManagerFactory(PERSISTENCE_UNIT);
EM = ENTITY_MANAGER_FACTORY.createEntityManager();

public static long getRowCount(Root<?> root) {
    CriteriaBuilder cb = EM.getCriteriaBuilder();
    CriteriaQuery<Long> query = cb.createQuery(Long.class);

    root = query.from(root.getClass());
    query.select(cb.count(root));

    return EM.createQuery(query).getSingleResult();
}

这就是我调用该方法的方式:

Root<PlacesEntity> placesRoot = null;
DbTools.getRowCount(placesRoot)

但是它不起作用，该方法在 root = query.from(root.getClass()); 行抛出 NullPointerException。

Answer 1

试试这个(警告未测试):

public static long getRowCount(Class<?> rootClass) {
    CriteriaBuilder cb = EM.getCriteriaBuilder();
    CriteriaQuery<Long> query = cb.createQuery(Long.class);

    Expression<?> root = query.from(rootClass);
    query.select(cb.count(root));

    return EM.createQuery(query).getSingleResult();
}

/*****/
DbTools.getRowCount(PlacesEntity.class);

祝你好运!