/*
* This file is part of rockframework.
*
* rockframework is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* rockframework 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 General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>;.
*/
package br.net.woodstock.rockframework.domain.persistence.orm.impl;
import java.io.Serializable;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import javax.persistence.EntityManager;
import javax.persistence.NoResultException;
import javax.persistence.Query;
import br.net.woodstock.rockframework.core.RockFrameworkLogger;
import br.net.woodstock.rockframework.core.util.Assert;
import br.net.woodstock.rockframework.core.utils.Conditions;
import br.net.woodstock.rockframework.domain.Entity;
import br.net.woodstock.rockframework.domain.persistence.FilterParameter;
import br.net.woodstock.rockframework.domain.persistence.Page;
import br.net.woodstock.rockframework.domain.persistence.PersistenceException;
import br.net.woodstock.rockframework.domain.persistence.orm.ORMFilter;
import br.net.woodstock.rockframework.domain.persistence.orm.ORMFilterType;
import br.net.woodstock.rockframework.domain.persistence.orm.ORMOptions;
import br.net.woodstock.rockframework.domain.persistence.orm.ORMResult;
@SuppressWarnings("rawtypes")
public abstract class JPARepositoryHelper {
private JPARepositoryHelper() {
//
}
public static void delete(final EntityManager entityManager, final Entity e) {
try {
entityManager.remove(e);
} catch (IllegalArgumentException ex) {
Entity tmp = entityManager.find(e.getClass(), e.getId());
entityManager.remove(tmp);
}
}
public static <E> E get(final EntityManager entityManager, final Class<E> clazz, final Serializable id) {
E e = entityManager.find(clazz, id);
return e;
}
public static void save(final EntityManager entityManager, final Entity<?> e) {
entityManager.persist(e);
}
public static void update(final EntityManager entityManager, final Entity<?> e) {
entityManager.merge(e);
}
public static void executeUpdate(final EntityManager entityManager, final ORMFilter filter) {
Query q = JPARepositoryHelper.getQuery(entityManager, filter);
q.executeUpdate();
}
@SuppressWarnings("unchecked")
public static ORMResult getCollection(final EntityManager entityManager, final ORMFilter filter) {
Query q = JPARepositoryHelper.getQuery(entityManager, filter);
List<Object> list = q.getResultList();
int total = -1;
if (filter.getPage() != null) {
Query qCount = JPARepositoryHelper.getCountQuery(entityManager, filter);
Number number = (Number) qCount.getSingleResult();
total = number.intValue();
} else {
total = list.size();
}
return JPARepositoryHelper.buildResult(filter, list, Integer.valueOf(total));
}
@SuppressWarnings("unchecked")
public static <E> E getSingle(final EntityManager entityManager, final ORMFilter filter) {
try {
Query q = JPARepositoryHelper.getQuery(entityManager, filter);
Object obj = q.getSingleResult();
return (E) obj;
} catch (NoResultException e) {
RockFrameworkLogger.getLogger().debug(e.getMessage(), e);
return null;
}
}
private static ORMResult buildResult(final ORMFilter filter, final Collection<Object> collection, final Integer total) {
return new ORMResult(total, collection, filter.getPage());
}
private static Query getQuery(final EntityManager entityManager, final ORMFilter filter) {
Assert.notNull(entityManager, "entityManager");
Assert.notNull(filter, "filter");
Map<String, Object> options = filter.getOptions();
Query q = null;
if (filter.getType() == ORMFilterType.NAMED) {
q = entityManager.createNamedQuery(filter.getFilter());
} else if (filter.getType() == ORMFilterType.NATIVE) {
if ((Conditions.isContainingKey(options, ORMOptions.OPTION_TARGET_ENTITY))) {
Object targetEntity = options.get(ORMOptions.OPTION_TARGET_ENTITY);
if (targetEntity instanceof Class) {
q = entityManager.createNativeQuery(filter.getFilter(), (Class) targetEntity);
} else if (targetEntity instanceof String) {
try {
q = entityManager.createNativeQuery(filter.getFilter(), Class.forName((String) targetEntity));
} catch (ClassNotFoundException e) {
throw new PersistenceException(e);
}
} else {
throw new PersistenceException("Invalid option[" + ORMOptions.OPTION_TARGET_ENTITY + "] " + targetEntity);
}
} else {
q = entityManager.createNativeQuery(filter.getFilter());
}
} else {
q = entityManager.createQuery(filter.getFilter());
}
JPARepositoryHelper.setParameters(q, filter);
if (filter.getPage() != null) {
Page page = filter.getPage();
int firstResult = ((page.getPageNumber() - 1) * page.getResultsPerPage());
q.setFirstResult(firstResult);
q.setMaxResults(page.getResultsPerPage());
}
if (Conditions.isNotEmpty(options)) {
if (filter.getPage() == null) {
if ((Conditions.isContainingKey(options, ORMOptions.OPTION_FIRST_RESULT))) {
Object obj = options.get(ORMOptions.OPTION_FIRST_RESULT);
int i = -1;
if (obj instanceof Number) {
i = ((Number) obj).intValue();
} else if (obj instanceof String) {
i = Integer.parseInt((String) obj);
}
if (i != -1) {
q.setFirstResult(i);
}
}
if ((Conditions.isContainingKey(options, ORMOptions.OPTION_MAX_RESULT))) {
Object obj = options.get(ORMOptions.OPTION_MAX_RESULT);
int i = -1;
if (obj instanceof Number) {
i = ((Number) obj).intValue();
} else if (obj instanceof String) {
i = Integer.parseInt((String) obj);
}
if (i != -1) {
q.setMaxResults(i);
}
}
}
}
return q;
}
private static Query getCountQuery(final EntityManager entityManager, final ORMFilter filter) {
Assert.notNull(entityManager, "entityManager");
Assert.notNull(filter, "filter");
Query q = null;
if (filter.getType() == ORMFilterType.NAMED) {
q = entityManager.createNamedQuery(filter.getCountQuery());
} else if (filter.getType() == ORMFilterType.NATIVE) {
q = entityManager.createNativeQuery(filter.getCountQuery());
} else {
q = entityManager.createQuery(filter.getCountQuery());
}
JPARepositoryHelper.setParameters(q, filter);
return q;
}
private static void setParameters(final Query query, final ORMFilter filter) {
List<FilterParameter> parameters = filter.getParameters();
if (Conditions.isNotEmpty(parameters)) {
for (FilterParameter parameter : parameters) {
int index = parameter.getIndex();
String name = parameter.getName();
Object value = parameter.getValue();
if (parameter.isIndexed()) {
query.setParameter(index, value);
} else {
query.setParameter(name, value);
}
}
}
}
}