|
@ -0,0 +1,205 @@
|
|
|
package com.yihu.wlyy.util.query;
|
|
|
|
|
|
import org.apache.commons.lang3.StringUtils;
|
|
|
import org.hibernate.Query;
|
|
|
import org.hibernate.Session;
|
|
|
import org.springframework.data.domain.Sort;
|
|
|
import org.springframework.data.jpa.repository.JpaRepository;
|
|
|
import org.springframework.data.repository.PagingAndSortingRepository;
|
|
|
import org.springframework.transaction.annotation.Propagation;
|
|
|
import org.springframework.transaction.annotation.Transactional;
|
|
|
|
|
|
import javax.persistence.EntityManager;
|
|
|
import javax.persistence.PersistenceContext;
|
|
|
import javax.persistence.criteria.CriteriaBuilder;
|
|
|
import javax.persistence.criteria.CriteriaQuery;
|
|
|
import javax.persistence.criteria.Predicate;
|
|
|
import javax.persistence.criteria.Root;
|
|
|
import javax.persistence.metamodel.EntityType;
|
|
|
import java.io.Serializable;
|
|
|
import java.lang.reflect.ParameterizedType;
|
|
|
import java.lang.reflect.Type;
|
|
|
import java.text.ParseException;
|
|
|
import java.util.ArrayList;
|
|
|
import java.util.Arrays;
|
|
|
import java.util.List;
|
|
|
|
|
|
/**
|
|
|
* Created by yww on 2017/08/14.
|
|
|
* medicine数据库通用查询
|
|
|
*/
|
|
|
@Transactional(propagation = Propagation.SUPPORTS)
|
|
|
public class BasemedicineJpaService<T,R> {
|
|
|
Class<R> repoClass;
|
|
|
|
|
|
@PersistenceContext(
|
|
|
name = "entityManagerMedicine",
|
|
|
unitName = "threeTimePersistenceUnit"
|
|
|
)
|
|
|
protected EntityManager entityManager;
|
|
|
|
|
|
public BasemedicineJpaService(){
|
|
|
Type genType = getClass().getGenericSuperclass();
|
|
|
if ((genType instanceof ParameterizedType)) {
|
|
|
Type[] params = ((ParameterizedType) genType).getActualTypeArguments();
|
|
|
if (params.length==2) {
|
|
|
repoClass = (Class) params[1];
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
public T save(T entity) {
|
|
|
return (T) getRepository().save(entity);
|
|
|
}
|
|
|
|
|
|
public T retrieve(Serializable id) {
|
|
|
return (T) getRepository().findOne(id);
|
|
|
}
|
|
|
|
|
|
public void delete(Serializable id) {
|
|
|
getRepository().delete(id);
|
|
|
}
|
|
|
|
|
|
public void delete(T entity) {
|
|
|
getRepository().delete(entity);
|
|
|
}
|
|
|
|
|
|
public void delete(Iterable ids) {
|
|
|
Iterable list = getRepository().findAll(ids);
|
|
|
getRepository().delete(list);
|
|
|
}
|
|
|
|
|
|
public Class<T> getEntityClass() {
|
|
|
Type genType = this.getClass().getGenericSuperclass();
|
|
|
Type[] parameters = ((ParameterizedType) genType).getActualTypeArguments();
|
|
|
return (Class) parameters[0];
|
|
|
}
|
|
|
|
|
|
public List search(String fields, String filters, String sorts, Integer page, Integer size) throws ParseException {
|
|
|
URLQueryParser queryParser = createQueryParser(fields, filters, sorts);
|
|
|
CriteriaQuery query = queryParser.makeCriteriaQuery();
|
|
|
|
|
|
if (page == null || page <= 0) page = 1;
|
|
|
if (size == null || size <= 0 || size > 10000) size = 15;
|
|
|
|
|
|
return entityManager
|
|
|
.createQuery(query)
|
|
|
.setFirstResult((page - 1) * size)
|
|
|
.setMaxResults(size)
|
|
|
.getResultList();
|
|
|
}
|
|
|
|
|
|
public List search(String filters) throws ParseException {
|
|
|
URLQueryParser queryParser = createQueryParser("", filters, "");
|
|
|
CriteriaQuery query = queryParser.makeCriteriaQuery();
|
|
|
|
|
|
return entityManager
|
|
|
.createQuery(query)
|
|
|
.getResultList();
|
|
|
}
|
|
|
|
|
|
public long getCount(String filters) throws ParseException {
|
|
|
URLQueryParser queryParser = createQueryParser(filters);
|
|
|
CriteriaQuery query = queryParser.makeCriteriaCountQuery();
|
|
|
|
|
|
return (long) entityManager.createQuery(query).getSingleResult();
|
|
|
}
|
|
|
|
|
|
protected <T> URLQueryParser createQueryParser(String fields, String filters, String orders) {
|
|
|
URLQueryParser queryParser = new URLQueryParser<T>(fields, filters, orders)
|
|
|
.setEntityManager(entityManager)
|
|
|
.setEntityClass(getEntityClass());
|
|
|
|
|
|
return queryParser;
|
|
|
}
|
|
|
|
|
|
protected <T> URLQueryParser createQueryParser(String filters) {
|
|
|
URLQueryParser queryParser = new URLQueryParser<T>(filters)
|
|
|
.setEntityManager(entityManager)
|
|
|
.setEntityClass(getEntityClass());
|
|
|
|
|
|
return queryParser;
|
|
|
}
|
|
|
|
|
|
protected Sort parseSorts(String sorter){
|
|
|
if (StringUtils.isNotEmpty(sorter)) {
|
|
|
String[] orderArray = sorter.split(",");
|
|
|
|
|
|
List<Sort.Order> orderList = new ArrayList<Sort.Order>(orderArray.length);
|
|
|
Arrays.stream(orderArray).forEach(
|
|
|
elem -> orderList.add(
|
|
|
elem.startsWith("+") ? new Sort.Order(Sort.Direction.ASC, elem.substring(1)):
|
|
|
new Sort.Order(Sort.Direction.DESC, elem.substring(1))));
|
|
|
|
|
|
return new Sort(orderList);
|
|
|
}
|
|
|
|
|
|
return null;
|
|
|
}
|
|
|
|
|
|
protected Session currentSession() {
|
|
|
return entityManager.unwrap(Session.class);
|
|
|
}
|
|
|
|
|
|
public PagingAndSortingRepository getRepository() {
|
|
|
return (PagingAndSortingRepository) SpringContext.getService(repoClass);
|
|
|
}
|
|
|
|
|
|
public JpaRepository getJpaRepository(){
|
|
|
return (JpaRepository) SpringContext.getService(repoClass);
|
|
|
}
|
|
|
|
|
|
public List<T> findByField(String field, Object value){
|
|
|
|
|
|
return findByFields(
|
|
|
new String[]{field},
|
|
|
new Object[]{value}
|
|
|
);
|
|
|
}
|
|
|
|
|
|
public List<T> findByFields(String[] fields, Object[] values){
|
|
|
CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder();
|
|
|
CriteriaQuery query = criteriaBuilder.createQuery(getEntityClass());
|
|
|
Root<T> root = query.from(getEntityClass());
|
|
|
List<Predicate> ls = new ArrayList<Predicate>();
|
|
|
for(int i=0; i< fields.length; i++){
|
|
|
if(values[i].getClass().isArray())
|
|
|
ls.add(criteriaBuilder.in(root.get(fields[i]).in((Object[])values[i])));
|
|
|
else
|
|
|
ls.add(criteriaBuilder.equal(root.get(fields[i]), values[i]));
|
|
|
}
|
|
|
query.where(ls.toArray(new Predicate[ls.size()]));
|
|
|
return entityManager
|
|
|
.createQuery(query)
|
|
|
.getResultList() ;
|
|
|
}
|
|
|
|
|
|
public String getClzName(){
|
|
|
return getEntityClass().getName();
|
|
|
}
|
|
|
|
|
|
public String getEntityIdFiled(){
|
|
|
EntityType entityType = entityManager.getMetamodel().entity(getEntityClass());
|
|
|
javax.persistence.metamodel.Type type = entityType.getIdType();
|
|
|
String s = entityType.getId(type.getJavaType()).getName();
|
|
|
return s;
|
|
|
}
|
|
|
|
|
|
|
|
|
public int delete(Object[] ids){
|
|
|
String hql = " DELETE FROM "+getEntityClass().getName()+" WHERE "+getEntityIdFiled()+" in(:ids)";
|
|
|
Query query = currentSession().createQuery(hql);
|
|
|
query.setParameterList("ids", ids);
|
|
|
return query.executeUpdate();
|
|
|
}
|
|
|
|
|
|
public void batchInsert(List list) {
|
|
|
for (int i = 0; i < list.size(); i++) {
|
|
|
entityManager.persist(list.get(i));
|
|
|
if (i % 30 == 0) {
|
|
|
entityManager.flush();
|
|
|
entityManager.clear();
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|