/*
* 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 org.hibernate.NonUniqueObjectException;
import org.hibernate.PropertyValueException;
import org.hibernate.Query;
import org.hibernate.SQLQuery;
import org.hibernate.Session;
import br.net.woodstock.rockframework.core.RockFrameworkLogger;
import br.net.woodstock.rockframework.core.util.Assert;
import br.net.woodstock.rockframework.core.utils.Arrays;
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;
public abstract class HibernateRepositoryHelper {
private HibernateRepositoryHelper() {
//
}
@SuppressWarnings("rawtypes")
public static void delete(final Session session, final Entity e) {
try {
session.delete(e);
} catch (PropertyValueException pve) {
session.refresh(e);
session.delete(e);
} catch (NonUniqueObjectException nuoe) {
Class clazz = e.getClass();
Serializable id = (Serializable) e.getId();
Entity tmp = (Entity) session.get(clazz, id);
session.delete(tmp);
}
}
@SuppressWarnings("unchecked")
public static <E> E get(final Session session, final Class<E> clazz, final Serializable id) {
E e = (E) session.get(clazz, id);
return e;
}
public static void save(final Session session, final Entity<?> e) {
session.save(e);
}
public static void update(final Session session, final Entity<?> e) {
try {
session.update(e);
} catch (NonUniqueObjectException nuoe) {
session.merge(e);
}
}
public static void executeUpdate(final Session session, final ORMFilter filter) {
Query q = HibernateRepositoryHelper.getQuery(session, filter);
q.executeUpdate();
}
@SuppressWarnings("unchecked")
public static ORMResult getCollection(final Session session, final ORMFilter filter) {
Query q = HibernateRepositoryHelper.getQuery(session, filter);
List<Object> list = q.list();
int total = -1;
if (filter.getPage() != null) {
Query qCount = HibernateRepositoryHelper.getCountQuery(session, filter);
Number number = (Number) qCount.uniqueResult();
total = number.intValue();
} else {
total = list.size();
}
return HibernateRepositoryHelper.buildResult(filter, list, Integer.valueOf(total));
}
@SuppressWarnings("unchecked")
public static <E> E getSingle(final Session session, final ORMFilter filter) {
Query q = HibernateRepositoryHelper.getQuery(session, filter);
Object obj = q.uniqueResult();
return (E) obj;
}
private static ORMResult buildResult(final ORMFilter filter, final Collection<Object> collection, final Integer total) {
return new ORMResult(total, collection, filter.getPage());
}
@SuppressWarnings("rawtypes")
private static Query getQuery(final Session session, final ORMFilter filter) {
Assert.notNull(session, "session");
Assert.notNull(filter, "filter");
Map<String, Object> options = filter.getOptions();
Query q = null;
if (filter.getType() == ORMFilterType.NAMED) {
throw new PersistenceException("Hibernate dont support named queries");
} 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) {
SQLQuery sq = session.createSQLQuery(filter.getFilter());
sq.addEntity((Class) targetEntity);
q = sq;
} else if (targetEntity instanceof String) {
try {
SQLQuery sq = session.createSQLQuery(filter.getFilter());
sq.addEntity(Class.forName((String) targetEntity));
q = sq;
} catch (ClassNotFoundException e) {
throw new PersistenceException(e);
}
} else {
throw new PersistenceException("Invalid option[" + ORMOptions.OPTION_TARGET_ENTITY + "] " + targetEntity);
}
} else {
q = session.createSQLQuery(filter.getFilter());
}
} else {
q = session.createQuery(filter.getFilter());
}
HibernateRepositoryHelper.setParameters(q, filter);
if (filter.getPage() != null) {
Page page = filter.getPage();
int firstResult = ((page.getPageNumber() - 1) * page.getResultsPerPage()) + 1;
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 Session session, final ORMFilter filter) {
Assert.notNull(session, "session");
Assert.notNull(filter, "query");
Query q = null;
if (filter.getType() == ORMFilterType.NAMED) {
throw new PersistenceException("Hibernate dont support named queries");
} else if (filter.getType() == ORMFilterType.NATIVE) {
q = session.createSQLQuery(filter.getCountQuery());
} else {
q = session.createQuery(filter.getCountQuery());
}
HibernateRepositoryHelper.setParameters(q, filter);
return q;
}
@SuppressWarnings("rawtypes")
private static void setParameters(final Query query, final ORMFilter filter) {
List<FilterParameter> parameters = filter.getParameters();
if (Conditions.isNotEmpty(parameters)) {
for (FilterParameter parameter : parameters) {
if (parameter.isIndexed()) {
int index = parameter.getIndex();
Object value = parameter.getValue();
if ((value instanceof Collection) || (Arrays.isArray(value))) {
RockFrameworkLogger.getLogger().warn("Setting Collection/Array may cause unexpected errors");
}
query.setParameter(index, value);
} else {
String name = parameter.getName();
Object value = parameter.getValue();
if (value instanceof Collection) {
query.setParameterList(name, (Collection) value);
} else if (Arrays.isArray(value)) {
query.setParameterList(name, Arrays.toObjectArray(value));
} else {
query.setParameter(name, value);
}
}
}
}
}
}