Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
io.robe.hibernate.dao.BaseDao Maven / Gradle / Ivy
package io.robe.hibernate.dao;
import com.google.common.base.Optional;
import com.google.common.base.Preconditions;
import io.dropwizard.hibernate.AbstractDAO;
import io.robe.common.service.headers.ResponseHeadersUtil;
import io.robe.common.service.search.SearchFrom;
import io.robe.common.service.search.SearchIgnore;
import io.robe.common.service.search.SearchableEnum;
import io.robe.common.service.search.model.SearchModel;
import io.robe.common.utils.StringsOperations;
import io.robe.hibernate.RobeHibernateBundle;
import io.robe.hibernate.entity.BaseEntity;
import org.hibernate.Criteria;
import org.hibernate.SessionFactory;
import org.hibernate.criterion.*;
import org.hibernate.transform.Transformers;
import org.hibernate.type.StringType;
import org.hibernate.type.Type;
import javax.inject.Inject;
import javax.persistence.Column;
import javax.persistence.Transient;
import java.io.Serializable;
import java.lang.reflect.Field;
import java.math.BigDecimal;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
/**
* Basic Dao Class which limits {@link io.dropwizard.hibernate.AbstractDAO} to take
* type parameters which extends {@link io.robe.hibernate.entity.BaseEntity}
*
* @param Type of the entity parameter.
*/
public class BaseDao extends AbstractDAO {
private static final ConcurrentHashMap fieldCache = new ConcurrentHashMap<>();
@Inject
RobeHibernateBundle bundle;
/**
* Constructor with session factory injection by guice
*
* @param sessionFactory injected session factory
*/
@Inject
public BaseDao(SessionFactory sessionFactory) {
super(sessionFactory);
}
private static String[] parseFilterExp(String filter) {
char[] chars = filter.toCharArray();
char[] name = new char[chars.length];
char[] op = new char[2];
char[] value = new char[chars.length];
int nIndex = 0;
int oIndex = 0;
int vIndex = 0;
short part = 0;
for (int i = 0; i < chars.length; i++) {
switch (part) {
case 0://Filling name
switch (chars[i]) {
case '=':
case '!':
case '<':
case '>':
case '~':
case '|':
//Jump to operation
op[oIndex++] = chars[i];
part = 1;
break;
default:
name[nIndex++] = chars[i];
}
break;
case 1://Filling op
switch (chars[i]) {
case '=':
op[oIndex++] = chars[i];
break;
default:
//Jump to value
value[vIndex++] = chars[i];
part = 2;
}
break;
case 2://Filling value
value[vIndex++] = chars[i];
break;
}
}
return new String[]{
new String(name, 0, nIndex),
new String(op, 0, oIndex),
new String(value, 0, vIndex)};
}
private static List getAllFields(List fields, Class type) {
fields.addAll(Arrays.asList(type.getDeclaredFields()));
if (type.getSuperclass() != null) {
fields = getAllFields(fields, type.getSuperclass());
}
return fields;
}
/**
* Returns modified list of the entities regarding to the search model.
* {@inheritDoc}
*
* @return List of entities.
*/
public List findAll(SearchModel search) {
List list = buildCriteria(search).list();
search.setLimit(null);
search.setOffset(null);
search.setSort(null);
long totalCount = (Long) buildCriteria(search).setProjection(Projections.rowCount()).uniqueResult();
search.setTotalCount(totalCount);
ResponseHeadersUtil.addTotalCount(search);
return list;
}
/**
* Returns modified list of the entities regarding to the search model.
* {@inheritDoc}
*
* @return List of entities.
*/
public List findAllWithSearchFrom(SearchModel search) {
List list = addSearchFromProjection(buildCriteria(search)).list();
search.setLimit(null);
search.setOffset(null);
search.setSort(null);
long totalCount = (Long) buildCriteria(search).setProjection(Projections.rowCount()).uniqueResult();
search.setTotalCount(totalCount);
ResponseHeadersUtil.addTotalCount(search);
return list;
}
public Criteria addSearchFromProjection(Criteria criteria) {
Field[] fields = getCachedFields(getEntityClass());
ProjectionList projectionList = Projections.projectionList();
for (Field field : fields) {
field.setAccessible(true);
SearchFrom searchFrom = field.getAnnotation(SearchFrom.class);
if (searchFrom != null) {
for (String target : searchFrom.target()) {
String alias = field.getName() + StringsOperations.capitalizeFirstChar(target);
StringBuilder sqlBuilder = new StringBuilder("(select ");
if (searchFrom.localId().isEmpty())
sqlBuilder.append(target);
else
sqlBuilder.append("group_concat(").append(target).append(")");
String prefix = bundle.getConfiguration().getProperty("hibernate.prefix");
if (prefix != null)
sqlBuilder.append(" from ").append(prefix).append(searchFrom.entity().getSimpleName());
else
sqlBuilder.append(" from ").append(searchFrom.entity().getSimpleName());
if (searchFrom.localId().isEmpty()) {
sqlBuilder.append(" where ").append(searchFrom.id()).append('=').append(field.getName()).append(") as ").append(alias);
} else {
sqlBuilder.append(" where ").append(searchFrom.id()).append('=').append("this_.").append(searchFrom.localId()).append(") as ").append(alias);
}
projectionList.add(Projections.alias(Projections.sqlProjection(sqlBuilder.toString(),
new String[]{alias}, new Type[]{new StringType()}), alias));
if (searchFrom.localId().isEmpty()) {
projectionList.add(Projections.property(field.getName()), field.getName());
}
}
} else {
if (field.getAnnotation(Column.class) != null)
projectionList.add(Projections.property(field.getName()), field.getName());
}
field.setAccessible(false);
}
criteria.setProjection(projectionList);
criteria.setResultTransformer(Transformers.ALIAS_TO_ENTITY_MAP);
return criteria;
}
/**
* {@inheritDoc}
*
* @return List of entities.
*/
public List findAll() {
Criteria criteria = criteria();
return list(criteria);
}
/**
* {@inheritDoc}
*
* @param oid id of the desired Entity
* @return returns the result.
*/
public T findById(String oid) {
return get(oid);
}
/**
* {@inheritDoc}
*
* @param oid id of the Given Entity
* @return returns the result.
*/
@SuppressWarnings("unchecked")
public T findById(Class clazz, Serializable oid) {
return (T) currentSession().get(clazz, Preconditions.checkNotNull(oid));
}
/**
* Create a record for the given entity instance.
*
* @param entity to record.
* @return updated version of the instance.
*/
public T create(T entity) {
return persist(entity);
}
/**
* Update a record for the given entity instance.
*
* @param entity to record.
* @return updated version of the instance.
*/
public T update(T entity) {
return persist(entity);
}
/**
* Delete a record for the given entity instance.
*
* @param entity to record.
* @return updated version of the instance.
*/
public T delete(T entity) {
currentSession().delete(entity);
return entity;
}
/**
* Flush the session.
*/
public void flush() {
currentSession().flush();
}
/**
* Merges the entity with the session.
*
* @param entity entity to merge
* @return
*/
@SuppressWarnings("unchecked")
public T merge(T entity) {
return (T) currentSession().merge(entity);
}
/**
* Detached the entity from session by evict method.
*
* @param entity entity to detach
* @return
*/
public T detach(T entity) {
currentSession().evict(entity);
return entity;
}
public Object getWithSearchFromData(T entity) throws IllegalAccessException {
Field[] fields = getCachedFields(getEntityClass());
HashMap output = new HashMap<>(fields.length + 5);
for (Field field : fields) {
field.setAccessible(true);
output.put(field.getName(), field.get(entity));
if (field.get(entity) == null)
continue;
SearchFrom searchFrom = field.getAnnotation(SearchFrom.class);
if (searchFrom != null) {
for (String target : searchFrom.target()) {
Object result = getSearchFromData(searchFrom, field.get(entity));
output.put(
(field.getName() + StringsOperations.capitalizeFirstChar(target)),
result);
}
} else if (field.getType().isEnum() && SearchableEnum.class.isAssignableFrom(field.getType())) {
SearchableEnum enumField = (SearchableEnum) field.get(entity);
output.put(field.getName() + "Text", enumField.getText());
}
field.setAccessible(false);
}
return output;
}
private Object getSearchFromData(SearchFrom from, Object id) {
Criteria criteria = currentSession().createCriteria(from.entity());
criteria.add(Restrictions.eq(from.id(), id));
for (String target : from.target()) {
criteria.setProjection(Projections.property(target));
}
return criteria.uniqueResult();
}
/**
* Creates a criteria from the given search model.
*
* @param search model
* @return
*/
protected final Criteria buildCriteria(SearchModel search) {
return this.buildCriteria(search, this.getEntityClass());
}
/**
* Creates a criteria from the given search model.
*
* @param search
* @param clazz of extends {@link BaseEntity}
* @return
*/
protected final Criteria buildCriteria(SearchModel search, Class clazz) {
Criteria criteria = this.currentSession().createCriteria(clazz);
if (search.getFields() != null && search.getFields().length != 0) {
ProjectionList projectionList = Projections.projectionList();
for (String fieldName : search.getFields()) {
projectionList.add(Projections.property(fieldName), fieldName);
}
criteria.setProjection(projectionList);
criteria.setResultTransformer(Transformers.ALIAS_TO_ENTITY_MAP);
}
if (search.getOffset() != null) {
criteria.setFirstResult(search.getOffset());
}
if (search.getLimit() != null) {
criteria.setMaxResults(search.getLimit());
}
if (search.getSort() != null && search.getSort().length != 0) {
for (String fieldName : search.getSort()) {
if (fieldName.startsWith(" ") || fieldName.startsWith("+")) {
criteria.addOrder(Order.asc(fieldName.substring(1)));
} else if (fieldName.startsWith("-")) {
criteria.addOrder(Order.desc(fieldName.substring(1)));
}
}
}
if (search.getQ() != null && !search.getQ().isEmpty()) {
Field[] fields = getCachedFields(clazz);
List fieldLikes = new ArrayList<>(fields.length);
for (Field field : fields) {
SearchFrom searchFrom = field.getAnnotation(SearchFrom.class);
if (searchFrom != null) {
List result = addRemoteMatchCriterias(searchFrom, search.getQ());
for (String id : result) {
if (searchFrom.localId().isEmpty())
fieldLikes.add(Restrictions.eq(field.getName(), id));
else
fieldLikes.add(Restrictions.eq(searchFrom.localId(), id));
}
} else if (field.getType().equals(String.class)) {
if (field.getAnnotation(SearchIgnore.class) == null) {
if (field.isEnumConstant()) {
fieldLikes.add(Restrictions.ilike(field.getName(), search.getQ(), MatchMode.ANYWHERE));
} else {
fieldLikes.add(Restrictions.ilike(field.getName(), search.getQ(), MatchMode.ANYWHERE));
}
}
} else if (field.getType().isEnum() && SearchableEnum.class.isAssignableFrom(field.getType())) {
SearchableEnum[] enums = (SearchableEnum[]) field.getType().getEnumConstants();
for (SearchableEnum anEnum : enums) {
if (anEnum.getText().toLowerCase().contains(search.getQ().toLowerCase())) {
fieldLikes.add(Restrictions.eq(field.getName(), anEnum));
}
}
}
}
criteria.add(Restrictions.or(fieldLikes.toArray(new Criterion[]{})));
}
if (search.getFilter() != null) {
Field[] fields = getCachedFields(clazz);
criteria.add(addFilterCriterias(fields, search.getFilter()));
}
return criteria;
}
private List addRemoteMatchCriterias(SearchFrom from, String searchQ) {
Criteria criteria = currentSession().createCriteria(from.entity());
Field[] fields = getCachedFields(from.entity());
Criterion[] fieldLikes = new Criterion[fields.length];
int i = 0;
for (Field field : fields) {
for (String target : from.target()) {
if (field.getName().equals(target)) {
if (field.getAnnotation(SearchIgnore.class) == null) {
if (SearchableEnum.class.isAssignableFrom(field.getType())) {
try {
Enum anEnum = Enum.valueOf((Class) field.getType(), searchQ);
fieldLikes[i++] = Restrictions.eq(field.getName(), anEnum);
} catch (IllegalArgumentException e) {
continue;
}
} else
fieldLikes[i++] = Restrictions.ilike(field.getName(), searchQ, MatchMode.ANYWHERE);
}
}
}
}
fieldLikes = Arrays.copyOf(fieldLikes, i);
criteria.add(Restrictions.or(fieldLikes));
criteria.setProjection(Projections.property(from.id()));
return criteria.list();
}
public Conjunction addFilterCriterias(Field[] fields, String filterParam) {
String[] filters = filterParam.split(",");
Criterion[] fieldFilters = new Criterion[filters.length];
int i = 0;
for (String filter : filters) {
String[] params = parseFilterExp(filter);
Optional value = null;
fieldsLoop:
for (Field field : fields) {
SearchFrom searchFrom = field.getAnnotation(SearchFrom.class);
if (field.getName().equals(params[0]) && field.getAnnotation(Transient.class) == null) {
if (params[1].equals("|=")) {
String[] svalues = params[2].split("\\|");
LinkedList lvalues = new LinkedList<>();
for (String svalue : svalues)
lvalues.add(castValue(field, svalue));
value = Optional.fromNullable(lvalues);
} else
value = Optional.fromNullable(castValue(field, params[2]));
break;
} else if (searchFrom != null && params[0].startsWith(field.getName())) {
String filterTarget = StringsOperations.unCapitalizeFirstChar(params[0].replace(field.getName(), ""));
for (String target : searchFrom.target()) {
if (filterTarget.equals(target)) {
try {
Field filterField = searchFrom.entity().getDeclaredField(filterTarget);
Criteria criteria = currentSession().createCriteria(searchFrom.entity());
if (searchFrom.localId().isEmpty())
params[0] = field.getName();
else
params[0] = searchFrom.localId();
if (SearchableEnum.class.isAssignableFrom(filterField.getType())) {
Enum anEnum = Enum.valueOf((Class) filterField.getType(), params[2]);
criteria.add(Restrictions.eq(filterTarget, anEnum));
} else if (params[1].equals("=")) {
criteria.add(Restrictions.eq(filterTarget, params[2]));
} else if (params[1].equals("~="))
criteria.add(Restrictions.ilike(filterTarget, params[2], MatchMode.ANYWHERE));
criteria.setProjection(Projections.property(searchFrom.id()));
List list = criteria.list();
if (!list.isEmpty()) {
value = Optional.fromNullable(list);
params[1] = "|=";
break fieldsLoop;
}
value = Optional.of("");
} catch (NoSuchFieldException e) {
continue;
} catch (IllegalArgumentException e) {
continue;
}
}
}
}
}
if (value == null)
continue;
switch (params[1]) {
case "=":
if (value.isPresent())
fieldFilters[i++] = Restrictions.eq(params[0], value.get());
else
fieldFilters[i++] = Restrictions.isNull(params[0]);
break;
case "!=":
if (value.isPresent())
fieldFilters[i++] = Restrictions.ne(params[0], value.get());
else
fieldFilters[i++] = Restrictions.isNotNull(params[0]);
break;
case "<":
fieldFilters[i++] = Restrictions.lt(params[0], value.get());
break;
case "<=":
fieldFilters[i++] = Restrictions.le(params[0], value.get());
break;
case ">":
fieldFilters[i++] = Restrictions.gt(params[0], value.get());
break;
case ">=":
fieldFilters[i++] = Restrictions.ge(params[0], value.get());
break;
case "~=":
fieldFilters[i++] = Restrictions.ilike(params[0], params[2], MatchMode.ANYWHERE);
break;
case "|=":
fieldFilters[i++] = Restrictions.in(params[0], (Collection) value.get());
break;
}
}
fieldFilters = Arrays.copyOf(fieldFilters, i);
return Restrictions.and(fieldFilters);
}
private Object castValue(Field field, String value) {
if ((field.getType() instanceof Class && ((Class) field.getType()).isEnum()))
return Enum.valueOf((Class) field.getType(), value);
if ("null".equals(value))
return null;
switch (field.getType().getName()) {
case "java.math.BigDecimal":
return new BigDecimal(value);
case "java.lang.Boolean":
case "boolean":
return Boolean.parseBoolean(value);
case "java.lang.Double":
case "double":
return Double.parseDouble(value);
case "java.lang.Integer":
case "int":
return Integer.parseInt(value);
case "java.lang.Long":
case "long":
return Long.parseLong(value);
case "java.lang.String":
return value;
case "java.util.Date":
return new Date(Long.parseLong(value));
}
return null;
}
private Field[] getCachedFields(Class entityClass) {
if (!fieldCache.containsKey(entityClass.getName())) {
List fields = new LinkedList<>();
getAllFields(fields, entityClass);
Field[] fieldArr = new Field[fields.size()];
fields.toArray(fieldArr);
fieldCache.put(entityClass.getName(), fieldArr);
}
return fieldCache.get(entityClass.getName());
}
}
