org.demoiselle.jpa.template.JPADatabaseAccess Maven / Gradle / Ivy
/*
* Demoiselle Framework
* Copyright (C) 2010 SERPRO
* ----------------------------------------------------------------------------
* This file is part of Demoiselle Framework.
*
* Demoiselle Framework is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public License version 3
* as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License version 3
* along with this program; if not, see implements DatabaseAccess {
private static final long serialVersionUID = -6304662620826264383L;
private Class beanClass;
protected abstract EntityManager getEntityManager();
@SuppressWarnings("WeakerAccess")
public Class getBeanClass() {
if (this.beanClass == null) {
this.beanClass = Reflections.getGenericTypeArgument(this.getClass(), 0);
}
return this.beanClass;
}
@Override
public void persist(T entity) {
getEntityManager().persist(entity);
}
@Override
public void remove(I id) {
T entity = getEntityManager().getReference(getBeanClass(), id);
getEntityManager().remove(entity);
}
@Override
@SuppressWarnings("unchecked")
public int remove(List ids) {
Class idClass = null;
if (!ids.isEmpty()) {
I id = ids.get(0);
idClass = id.getClass();
}
CriteriaBuilder builder = getEntityManager().getCriteriaBuilder();
CriteriaDelete criteriaDelete = builder.createCriteriaDelete(getBeanClass());
Root typeRoot = criteriaDelete.from(getBeanClass());
SingularAttribute idAttribute = typeRoot.getModel().getId(idClass);
Query removeQuery = getEntityManager().createQuery(criteriaDelete.where(typeRoot.get(idAttribute).in(ids)));
return removeQuery.executeUpdate();
}
@Override
@SuppressWarnings(value = "unchecked")
public int remove(I... ids) {
if (ids == null) {
throw new IllegalArgumentException("'ids' can't be null");
}
Class idClass = null;
if (ids.length > 0) {
I id = ids[0];
idClass = id.getClass();
}
CriteriaBuilder builder = getEntityManager().getCriteriaBuilder();
CriteriaDelete criteriaDelete = builder.createCriteriaDelete(getBeanClass());
Root typeRoot = criteriaDelete.from(getBeanClass());
SingularAttribute idAttribute = typeRoot.getModel().getId(idClass);
Query removeQuery = getEntityManager().createQuery(criteriaDelete.where(typeRoot.get(idAttribute).in(ids)));
return removeQuery.executeUpdate();
}
@Override
public T merge(T entity) {
return getEntityManager().merge(entity);
}
@Override
public T load(I id) {
return getEntityManager().find(getBeanClass(), id);
}
@Override
@SuppressWarnings("unchecked")
public List loadList(List ids) {
Class idClass = null;
if (!ids.isEmpty()) {
I id = ids.get(0);
idClass = id.getClass();
}
CriteriaBuilder builder = getEntityManager().getCriteriaBuilder();
CriteriaQuery criteria = builder.createQuery(getBeanClass());
Root typeRoot = criteria.from(getBeanClass());
SingularAttribute idAttribute = typeRoot.getModel().getId(idClass);
TypedQuery listQuery = getEntityManager().createQuery(criteria.where(typeRoot.get(idAttribute).in(ids)));
Pagination pagination = getPagination();
if (pagination != null && pagination.isInitialized()) {
listQuery.setFirstResult(pagination.getFirstResult());
listQuery.setMaxResults(pagination.getPageSize());
}
return listQuery.getResultList();
}
@Override
@SuppressWarnings("unchecked")
public List loadList(I... ids) {
if (ids == null) {
throw new IllegalArgumentException("'ids' can't be null");
}
Class idClass = null;
if (ids.length > 0) {
I id = ids[0];
idClass = id.getClass();
}
CriteriaBuilder builder = getEntityManager().getCriteriaBuilder();
CriteriaQuery criteria = builder.createQuery(getBeanClass());
Root typeRoot = criteria.from(getBeanClass());
SingularAttribute idAttribute = typeRoot.getModel().getId(idClass);
TypedQuery listQuery = getEntityManager().createQuery(criteria.where(typeRoot.get(idAttribute).in(ids)));
Pagination pagination = getPagination();
if (pagination != null && pagination.isInitialized()) {
listQuery.setFirstResult(pagination.getFirstResult());
listQuery.setMaxResults(pagination.getPageSize());
}
return listQuery.getResultList();
}
@Override
public List listAll() {
CriteriaBuilder builder = getEntityManager().getCriteriaBuilder();
CriteriaQuery criteriaQuery = builder.createQuery(getBeanClass());
criteriaQuery.from(getBeanClass());
TypedQuery query = getEntityManager().createQuery( criteriaQuery );
Pagination pagination = getPagination();
if (pagination != null && pagination.isInitialized()) {
pagination.setTotalResults(this.countAll().intValue());
query.setFirstResult(pagination.getFirstResult());
query.setMaxResults(pagination.getPageSize());
}
return query.getResultList();
}
/**
*
* Returns a valid pagination object to control the listing
* of instances of the entity controlled by this class.
*
*
*
* The default implementation always return null. If
* you want to paginate results returned by the methods {@link #loadList(Object[])},
* {@link #loadList(List)} and {@link #listAll()} then overwrite this method
* and return a valid pagination object.
*
*
* @return Pagination object to control the pagination of results for this
* class, or null if no pagination should occurr.
*/
protected Pagination getPagination() {
return null;
}
@SuppressWarnings("WeakerAccess")
protected Number countAll() {
CriteriaBuilder qb = getEntityManager().getCriteriaBuilder();
CriteriaQuery cq = qb.createQuery(Long.class);
cq.select(qb.count(cq.from(getBeanClass())));
return getEntityManager().createQuery(cq).getSingleResult();
}
}