使用criteria简单查询工具类
以前用jpa写了一个条件筛选的查询数据如下，才知道那么渣渣，就是一个表，根据前端来筛选数据，写的如下
首先就是判断前端传来的参数就写了那么多，现在才发现是渣渣中的渣渣，而且还费时，用criteria很快就搞定
首先创建类并实现Specification<T>接口
import java.util.ArrayList;
import java.util.List;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import org.springframework.data.jpa.domain.Specification;
public class ExpandCriteria<T> implements Specification<T>{
private List<ExpandCriterion> criterions = new ArrayList<ExpandCriterion>();
public Predicate toPredicate(Root<T> root, CriteriaQuery<?> query,
CriteriaBuilder builder) {
if (!criterions.isEmpty()) {
List<Predicate> predicates = new ArrayList<Predicate>();
for(ExpandCriterion c : criterions){
predicates.add(c.toPredicate(root, query,builder));
}
// 将所有条件用 and 联合起来
if (predicates.size() > 0) {
return builder.and(predicates.toArray(new Predicate[predicates.size()]));
}
}
return builder.conjunction();
}
public void add(ExpandCriterion criterion){
if(criterion!=null){
criterions.add(criterion);
}
}
public static void main(String[] args) {
//使用示例Demo
// Criteria<Entity> c = new Criteria<Entity>();
// c.add(Restrictions.like("code", searchParam.getCode(), true));
// c.add(Restrictions.eq("level", searchParam.getLevel(), false));
// c.add(Restrictions.eq("mainStatus", searchParam.getMainStatus(), true));
// c.add(Restrictions.eq("flowStatus", searchParam.getFlowStatus(), true));
// c.add(Restrictions.eq("createUser.userName", searchParam.getCreateUser(), true));
// c.add(Restrictions.lte("submitTime", searchParam.getStartSubmitTime(), true));
// c.add(Restrictions.gte("submitTime", searchParam.getEndSubmitTime(), true));
// c.add(Restrictions.eq("needFollow", searchParam.getIsfollow(), true));
// c.add(Restrictions.ne("flowStatus", searchParam.getMainStatus() true));
// c.add(Restrictions.in("solveTeam.code",teamCodes, true));
// repository.findAll(c);
}
} 新建ExpandCriterion接口
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
public interface ExpandCriterion {
public enum Operator {
EQ, NE, LIKE, GT, LT, GTE, LTE, AND, OR
}
public Predicate toPredicate(Root<?> root, CriteriaQuery<?> query,
CriteriaBuilder builder);
}新建Restrictions.java
import java.util.Collection;
import org.springframework.util.StringUtils;
import com.sll.iot.dao.base.criteria.ExpandCriterion.Operator;
public class Restrictions {
public static SimpleExpression eq(String fieldName, Object value, boolean ignoreNull) {
if(StringUtils.isEmpty(value))return null;
return new SimpleExpression (fieldName, value, Operator.EQ);
}
public static SimpleExpression ne(String fieldName, Object value, boolean ignoreNull) {
if(StringUtils.isEmpty(value))return null;
return new SimpleExpression (fieldName, value, Operator.NE);
}
public static SimpleExpression like(String fieldName, String value, boolean ignoreNull) {
if(StringUtils.isEmpty(value))return null;
return new SimpleExpression (fieldName, value, Operator.LIKE);
}
public static SimpleExpression gt(String fieldName, Object value, boolean ignoreNull) {
if(StringUtils.isEmpty(value))return null;
return new SimpleExpression (fieldName, value, Operator.GT);
}
public static SimpleExpression lt(String fieldName, Object value, boolean ignoreNull) {
if(StringUtils.isEmpty(value))return null;
return new SimpleExpression (fieldName, value, Operator.LT);
}
public static SimpleExpression lte(String fieldName, Object value, boolean ignoreNull) {
if(StringUtils.isEmpty(value))return null;
return new SimpleExpression (fieldName, value, Operator.GTE);
}
public static SimpleExpression gte(String fieldName, Object value, boolean ignoreNull) {
if(StringUtils.isEmpty(value))return null;
return new SimpleExpression (fieldName, value, Operator.LTE);
}
public static LogicalExpression and(ExpandCriterion... criterions){
return new LogicalExpression(criterions, Operator.AND);
}
public static LogicalExpression or(ExpandCriterion... criterions){
return new LogicalExpression(criterions, Operator.OR);
}
@SuppressWarnings("rawtypes")
public static LogicalExpression in(String fieldName, Collection value, boolean ignoreNull) {
if(ignoreNull&&(value==null||value.isEmpty())){
return null;
}
SimpleExpression[] ses = new SimpleExpression[value.size()];
int i=0;
for(Object obj : value){
ses[i]=new SimpleExpression(fieldName,obj,Operator.EQ);
i++;
}
return new LogicalExpression(ses,Operator.OR);
} 新建SimpleExpression.java
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Expression;
import javax.persistence.criteria.Path;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
public class SimpleExpression implements ExpandCriterion{
private String fieldName; //属性名
private Object value; //对应值
private Operator operator; //计算符
protected SimpleExpression(String fieldName, Object value, Operator operator) {
this.fieldName = fieldName;
this.value = value;
this.operator = operator;
}
public String getFieldName() {
return fieldName;
}
public Object getValue() {
return value;
}
public Operator getOperator() {
return operator;
}
@SuppressWarnings({ "rawtypes", "unchecked" })
public Predicate toPredicate(Root<?> root, CriteriaQuery<?> query,
CriteriaBuilder builder) {
Path expression = null;
if(fieldName.contains(".")){
String[] names = fieldName.split(".");
expression = root.get(names[0]);
for (int i = 1; i < names.length; i++) {
expression = expression.get(names[i]);
}
}else{
expression = root.get(fieldName);
}
switch (operator) {
case EQ:
return builder.equal(expression, value);
case NE:
return builder.notEqual(expression, value);
case LIKE:
return builder.like((Expression<String>) expression, "%" + value + "%");
case LT:
return builder.lessThan(expression, (Comparable) value);
case GT:
return builder.greaterThan(expression, (Comparable) value);
case LTE:
return builder.lessThanOrEqualTo(expression, (Comparable) value);
case GTE:
return builder.greaterThanOrEqualTo(expression, (Comparable) value);
default:
return null;
}
}
} LogicalExpression.java
import java.util.ArrayList;
import java.util.List;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
public class LogicalExpression implements ExpandCriterion {
private ExpandCriterion[] criterion; // 逻辑表达式中包含的表达式
private Operator operator; //计算符
public LogicalExpression(ExpandCriterion[] criterions, Operator operator) {
this.criterion = criterions;
this.operator = operator;
}
public Predicate toPredicate(Root<?> root, CriteriaQuery<?> query,
CriteriaBuilder builder) {
List<Predicate> predicates = new ArrayList<Predicate>();
for(int i=0;i<this.criterion.length;i++){
predicates.add(this.criterion[i].toPredicate(root, query, builder));
}
switch (operator) {
case OR:
return builder.or(predicates.toArray(new Predicate[predicates.size()]));
default:
return null;
}
}
}使用criteria前提是dao接口必须实现JpaSpecificationExecutor<T>接口
改造如下
//条件查询
@Override
public Paging<Channel> query(Paging<Channel> paging,String channelName,String operator) {
Pageable pageReq = new PageRequest(paging.getCurrentPage()-1, paging.getPageSize());
Page<Channel> pageChanel=null;
ExpandCriteria<Channel> criteria = new ExpandCriteria<Channel>();
if(StringUtil.isNotEmpty(channelName)){
criteria.add(Restrictions.like("name", channelName, false));
}
if(StringUtil.isNotEmpty(operator)){
criteria.add(Restrictions.eq("operator",Operator.valueOf(operator), false));
}
pageChanel=channelRepository.findAll(criteria, pageReq);
if(pageChanel!=null){
paging.setTotalCount((int)pageChanel.getTotalElements());
paging.setData(pageChanel.getContent());
paging.setTotalPage(pageChanel.getTotalPages());
}
return paging;
}都不用在dao接口写什么东西
使用方法就是demo
public static void main(String[] args) {
//使用示例Demo
// Criteria<Entity> c = new Criteria<Entity>();
// c.add(Restrictions.like("code", searchParam.getCode(), true));
// c.add(Restrictions.eq("level", searchParam.getLevel(), false));
// c.add(Restrictions.eq("mainStatus", searchParam.getMainStatus(), true));
// c.add(Restrictions.eq("flowStatus", searchParam.getFlowStatus(), true));
// c.add(Restrictions.eq("createUser.userName", searchParam.getCreateUser(), true));
// c.add(Restrictions.lte("submitTime", searchParam.getStartSubmitTime(), true));
// c.add(Restrictions.gte("submitTime", searchParam.getEndSubmitTime(), true));
// c.add(Restrictions.eq("needFollow", searchParam.getIsfollow(), true));
// c.add(Restrictions.ne("flowStatus", searchParam.getMainStatus() true));
// c.add(Restrictions.in("solveTeam.code",teamCodes, true));
// repository.findAll(c);
}打包JPA动态查询(CriteriaQuery) eq、ge、gt
封装JPA动态查询（CriteriaQuery）
JPA动态查询（CriteriaQuery）封装的一段代码：
package com.platform.framework.dao.jpa;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Date;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import javax.persistence.EntityManager;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaBuilder.In;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Order;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import org.apache.log4j.Logger;
@SuppressWarnings({ "unused", "unchecked", "rawtypes", "null", "hiding" })
public class Query implements Serializable {
private static final long serialVersionUID = 5064932771068929342L;
private static Logger log = Logger.getLogger(Query.class);
private EntityManager entityManager;
private Class clazz;
private Root from;
private List<Predicate> predicates;
private CriteriaQuery criteriaQuery;
private CriteriaBuilder criteriaBuilder;
private List<Order> orders;
private Map<String, Query> subQuery;
private Map<String, Query> linkQuery;
private String projection;
private List<Query> orQuery;
private String groupBy;
private Query() {
}
private Query(Class clazz, EntityManager entityManager) {
this.clazz = clazz;
this.entityManager = entityManager;
this.criteriaBuilder = this.entityManager.getCriteriaBuilder();
this.criteriaQuery = criteriaBuilder.createQuery(this.clazz);
this.from = criteriaQuery.from(this.clazz);
this.predicates = new ArrayList();
this.orders = new ArrayList();
}
public static Query forClass(Class clazz, EntityManager entityManager) {
return new Query(clazz, entityManager);
}
private void addSubQuery(String propertyName, Query query) {
if (this.subQuery == null)
this.subQuery = new HashMap();
if (query.projection == null)
throw new RuntimeException("子查询字段未设置");
this.subQuery.put(propertyName, query);
}
private void addSubQuery(Query query) {
addSubQuery(query.projection, query);
}
public void addLinkQuery(String propertyName, Query query) {
if (this.linkQuery == null)
this.linkQuery = new HashMap();
this.linkQuery.put(propertyName, query);
}
public void eq(String propertyName, Object value) {
if (isNullOrEmpty(value))
return;
this.predicates.add(criteriaBuilder.equal(from.get(propertyName), value));
}
private boolean isNullOrEmpty(Object value) {
if (value instanceof String) {
return value == null || "".equals(value);
}
return value == null;
}
public void or(List<String> propertyName, Object value) {
if (isNullOrEmpty(value))
return;
if ((propertyName == null) || (propertyName.size() == 0))
return;
Predicate predicate = criteriaBuilder.or(criteriaBuilder.equal(from.get(propertyName.get(0)), value));
for (int i = 1; i < propertyName.size(); ++i)
predicate = criteriaBuilder.or(predicate, criteriaBuilder.equal(from.get(propertyName.get(i)), value));
this.predicates.add(predicate);
}
public void orLike(List<String> propertyName, String value) {
if (isNullOrEmpty(value) || (propertyName.size() == 0))
return;
if (value.indexOf("%") < 0)
value = "%" + value + "%";
Predicate predicate = criteriaBuilder.or(criteriaBuilder.like(from.get(propertyName.get(0)), value.toString()));
for (int i = 1; i < propertyName.size(); ++i)
predicate = criteriaBuilder.or(predicate, criteriaBuilder.like(from.get(propertyName.get(i)), value));
this.predicates.add(predicate);
}
public void isNull(String propertyName) {
this.predicates.add(criteriaBuilder.isNull(from.get(propertyName)));
}
public void isNotNull(String propertyName) {
this.predicates.add(criteriaBuilder.isNotNull(from.get(propertyName)));
}
public void notEq(String propertyName, Object value) {
if (isNullOrEmpty(value)) {
return;
}
this.predicates.add(criteriaBuilder.notEqual(from.get(propertyName), value));
}
public void notIn(String propertyName, Collection value) {
if ((value == null) || (value.size() == 0)) {
return;
}
Iterator iterator = value.iterator();
In in = criteriaBuilder.in(from.get(propertyName));
while (iterator.hasNext()) {
in.value(iterator.next());
}
this.predicates.add(criteriaBuilder.not(in));
}
public void like(String propertyName, String value) {
if (isNullOrEmpty(value))
return;
if (value.indexOf("%") < 0)
value = "%" + value + "%";
this.predicates.add(criteriaBuilder.like(from.get(propertyName), value));
}
public void between(String propertyName, Date lo, Date go) {
if (!isNullOrEmpty(lo) && !isNullOrEmpty(go)) {
this.predicates.add(criteriaBuilder.between(from.get(propertyName), lo, go));
}
// if (!isNullOrEmpty(lo) && !isNullOrEmpty(go)) {
// this.predicates.add(criteriaBuilder.lessThan(from.get(propertyName),
// new DateTime(lo).toString()));
// }
// if (!isNullOrEmpty(go)) {
// this.predicates.add(criteriaBuilder.greaterThan(from.get(propertyName),
// new DateTime(go).toString()));
// }
}
public void between(String propertyName, Number lo, Number go) {
if (!(isNullOrEmpty(lo)))
ge(propertyName, lo);
if (!(isNullOrEmpty(go)))
le(propertyName, go);
}
public void le(String propertyName, Number value) {
if (isNullOrEmpty(value)) {
return;
}
this.predicates.add(criteriaBuilder.le(from.get(propertyName), value));
}
public void lt(String propertyName, Number value) {
if (isNullOrEmpty(value)) {
return;
}
this.predicates.add(criteriaBuilder.lt(from.get(propertyName), value));
}
public void ge(String propertyName, Number value) {
if (isNullOrEmpty(value)) {
return;
}
this.predicates.add(criteriaBuilder.ge(from.get(propertyName), value));
}
public void gt(String propertyName, Number value) {
if (isNullOrEmpty(value)) {
return;
}
this.predicates.add(criteriaBuilder.gt(from.get(propertyName), value));
}
public void in(String propertyName, Collection value) {
if ((value == null) || (value.size() == 0)) {
return;
}
Iterator iterator = value.iterator();
In in = criteriaBuilder.in(from.get(propertyName));
while (iterator.hasNext()) {
in.value(iterator.next());
}
this.predicates.add(in);
}
public void addCriterions(Predicate predicate) {
this.predicates.add(predicate);
}
public CriteriaQuery newCriteriaQuery() {
criteriaQuery.where(predicates.toArray(new Predicate[0]));
if (!isNullOrEmpty(groupBy)) {
criteriaQuery.groupBy(from.get(groupBy));
}
if (this.orders != null) {
criteriaQuery.orderBy(orders);
}
addLinkCondition(this);
return criteriaQuery;
}
private void addLinkCondition(Query query) {
Map subQuery = query.linkQuery;
if (subQuery == null)
return;
for (Iterator queryIterator = subQuery.keySet().iterator(); queryIterator.hasNext();) {
String key = (String) queryIterator.next();
Query sub = (Query) subQuery.get(key);
from.join(key);
criteriaQuery.where(sub.predicates.toArray(new Predicate[0]));
addLinkCondition(sub);
}
}
public void addOrder(String propertyName, String order) {
if (order == null || propertyName == null)
return;
if (this.orders == null)
this.orders = new ArrayList();
if (order.equalsIgnoreCase("asc"))
this.orders.add(criteriaBuilder.asc(from.get(propertyName)));
else if (order.equalsIgnoreCase("desc"))
this.orders.add(criteriaBuilder.desc(from.get(propertyName)));
}
public void setOrder(String propertyName, String order) {
this.orders = null;
addOrder(propertyName, order);
}
public Class getModleClass() {
return this.clazz;
}
public String getProjection() {
return this.projection;
}
public void setProjection(String projection) {
this.projection = projection;
}
public Class getClazz() {
return this.clazz;
}
public List<Order> getOrders() {
return orders;
}
public void setOrders(List<Order> orders) {
this.orders = orders;
}
public EntityManager getEntityManager() {
return this.entityManager;
}
public void setEntityManager(EntityManager em) {
this.entityManager = em;
}
public Root getFrom() {
return from;
}
public List<Predicate> getPredicates() {
return predicates;
}
public void setPredicates(List<Predicate> predicates) {
this.predicates = predicates;
}
public CriteriaQuery getCriteriaQuery() {
return criteriaQuery;
}
public CriteriaBuilder getCriteriaBuilder() {
return criteriaBuilder;
}
public void setFetchModes(List<String> fetchField, List<String> fetchMode) {
}
public String getGroupBy() {
return groupBy;
}
public void setGroupBy(String groupBy) {
this.groupBy = groupBy;
}
}
<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:p="http://www.springframework.org/schema/p"
xmlns:tx="http://www.springframework.org/schema/tx" xmlns:context="http://www.springframework.org/schema/context"
xmlns:util="http://www.springframework.org/schema/util"
xmlns:aop="http://www.springframework.org/schema/aop"
xsi:schemaLocation="
http://www.springframework.org/schema/beans
http://www.springframework.org/schema/beans/spring-beans-3.1.xsd
http://www.springframework.org/schema/tx
http://www.springframework.org/schema/tx/spring-tx-3.1.xsd
http://www.springframework.org/schema/context
http://www.springframework.org/schema/context/spring-context-3.1.xsd
http://www.springframework.org/schema/aop
http://www.springframework.org/schema/aop/spring-aop.xsd
http://www.springframework.org/schema/util
http://www.springframework.org/schema/util/spring-util-3.1.xsd">
<!-- JPA Entity Manager Factory -->
<bean id="entityManagerFactory"
class="org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean"
p:packagesToScan="com.**.model" p:dataSource-ref="dataSource"
p:jpaVendorAdapter-ref="hibernateVendor" p:jpaPropertyMap-ref="jpaPropertyMap"/>
<util:map id="jpaPropertyMap">
<entry key="hibernate.hbm2ddl.auto" value="update" /><!-- create,update,none -->
<entry key="hibernate.format_sql" value="false" />
<entry key="hibernate.show_sql" value="false" />
<entry key="hibernate.current_session_context_class" value="org.hibernate.context.internal.ThreadLocalSessionContext"/>
<entry key="hibernate.dialect" value="org.hibernate.dialect.MySQLDialect" />
<!-- To enable Hibernate's second level cache and query cache settings -->
<entry key="hibernate.max_fetch_depth" value="4" />
<entry key="hibernate.cache.use_second_level_cache" value="true" />
<entry key="hibernate.cache.use_query_cache" value="true" />
<!-- <entry key="hibernate.cache.region.factory_class" value="org.hibernate.cache.ehcache.EhCacheRegionFactory" /> -->
<entry key="hibernate.cache.region.factory_class" value="org.hibernate.cache.SingletonEhCacheRegionFactory" />
</util:map>
<bean id="hibernateVendor"
class="org.springframework.orm.jpa.vendor.HibernateJpaVendorAdapter"
p:database="MYSQL" p:showSql="true" p:generateDdl="true"
p:databasePlatform="org.hibernate.dialect.MySQLDialect" />
<bean id="transactionHandler" class="com.platform.framework.dao.jpa.TransactionHandler" >
<property name="txmethod">
<list>
<value>insert</value>
<value>update</value>
<value>delete</value>
</list>
</property>
<property name="entityManagerFactory" ref="entityManagerFactory"/>
</bean>
<aop:config>
<aop:aspect id="tran" ref="transactionHandler">
<aop:pointcut id="tranMethod"
expression="
execution(* com.*.dao.*.*(..))||
execution(* com.*.service.impl.*.*(..))||
execution(* com.*.*.dao.*.*(..))||
execution(* com.*.*.service.impl.*.*(..))||
execution(* com.*.*.*.dao.*.*(..))||
execution(* com.*.*.*.service.impl.*.*(..))||
execution(* com.*.*.*.*.dao.*.*(..))||
execution(* com.*.*.*.*.service.impl.*.*(..))||
execution(* com.*.*.*.*.*.dao.*.*(..))||
execution(* com.*.*.*.*.*.service.impl.*.*(..))||
execution(* com.*.*.*.*.*.*.dao.*.*(..))||
execution(* com.*.*.*.*.*.*.service.impl.*.*(..))||
execution(* com.platform.framework.dao.jpa.BaseDaoImpl.*(..))"/>
<aop:around method="exec" pointcut-ref="tranMethod" />
</aop:aspect>
</aop:config>
<bean id="baseDao" class="com.platform.framework.dao.jpa.BaseDaoImpl">
<property name="emf" ref="entityManagerFactory"/>
</bean>
</beans>
package com.platform.framework.dao.jpa;
import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.persistence.EntityTransaction;
import org.apache.log4j.Logger;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.Signature;
public class TransactionHandler {
private static final Logger log = Logger
.getLogger(TransactionHandler.class);
private String[] txmethod;// 配置事务的传播特性方法
private EntityManagerFactory entityManagerFactory;// JPA工厂
public Object exec(ProceedingJoinPoint point) throws Throwable {
Signature signature = point.getSignature();
log.debug(point.getTarget().getClass().getName() + "."
+ signature.getName() + "()");
Boolean isTransaction = false;
for (String method : txmethod) {
if (signature.getName().startsWith(method)) {// 以method开头的方法打开事务
isTransaction = true;
break;
}
}
// JPA->Hibernate
if (point.getTarget() instanceof EntityManagerFactoryProxy) {
// 获得被代理对象
EntityManagerFactoryProxy emfp = (EntityManagerFactoryProxy) point
.getTarget();
EntityManager em = emfp.getEntityManager();
if (em != null) {// 如果对象已经有em了就不管
return point.proceed();
} else {
em = entityManagerFactory.createEntityManager();
}
log.debug("JPA->Hibernate open connection...");
if (isTransaction) {
EntityTransaction t = null;
try {
// 打开连接并开启事务
log.debug("JPA->Hibernate begin transaction...");
t = em.getTransaction();
if (!t.isActive())
t.begin();
emfp.setEntityManager(em);
Object obj = point.proceed();
// 提交事务
log.debug("JPA->Hibernate commit...");
t.commit();
return obj;
} catch (Exception e) {
if (t != null) {
log.debug("JPA->Hibernate error...,rollback..."
+ e.getMessage());
t.rollback();
}
e.printStackTrace();
throw e;
} finally {
if (em != null && em.isOpen()) {// 关闭连接
em.close();
log.debug("JPA->Hibernate close connection...");
}
emfp.setEntityManager(null);
}
} else {
try {
emfp.setEntityManager(em);
return point.proceed();
} catch (Exception e) {
log.debug("JPA->Hibernate error..." + e.getMessage());
e.printStackTrace();
throw e;
} finally {
if (em != null && em.isOpen()) {// 关闭连接
em.close();
log.debug("JPA->Hibernate close connection...");
}
emfp.setEntityManager(null);
}
}
} else {
return point.proceed();
}
}
public String[] getTxmethod() {
return txmethod;
}
public void setTxmethod(String[] txmethod) {
this.txmethod = txmethod;
}
public void setEntityManagerFactory(
EntityManagerFactory entityManagerFactory) {
this.entityManagerFactory = entityManagerFactory;
}
} EntityManager管理器，通过spring管理
package com.platform.framework.dao.jpa;
import java.util.Collection;
import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
public class EntityManagerFactoryProxy {
private static ThreadLocal<EntityManager> emThreadLocal = new ThreadLocal<EntityManager>();
private static EntityManagerFactory emf;
public void setEmf(EntityManagerFactory emf) {
EntityManagerFactoryProxy.emf = emf;
}
public static EntityManagerFactory getEmf() {
return emf;
}
public EntityManager getEntityManager() {
return emThreadLocal.get();
}
public void setEntityManager(EntityManager em) {
emThreadLocal.set(em);
}
public String createInCondition(String name, Collection<String> values) {
if (values == null || values.size() == 0) {
return "1<>1";
}
StringBuffer sb = new StringBuffer();
sb.append(name + " in(");
for (String id : values) {
sb.append("'" + id + "',");
}
String hsqlCondition = sb.substring(0, sb.length() - 1) + ")";
return hsqlCondition;
}
} Page分页和结果封装类
package com.platform.framework.dao.jpa;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
public class Page<T> implements Serializable {
private static final long serialVersionUID = 665620345605746930L;
private int count;
private int pageNo;
private int rowsPerPage;
private int totalPageCount;
private int firstRow;
private int lastRow;
private List<T> result;
public Object obj;
public Integer code; // 返回码
private boolean success = true;
private String message;
public Page() {
}
public Page(List<T> list) {
this(list.size(), 1, list.size(), list);
}
public Page(int count, int pageNo, int rowsPerPage, List<T> result) {
if (rowsPerPage < 1) {
rowsPerPage = 1;
}
this.count = count;
this.pageNo = pageNo;
this.result = result;
this.rowsPerPage = rowsPerPage;
if (this.result == null)
this.result = new ArrayList<T>();
totalPageCount = count / rowsPerPage;
if (count - (count / rowsPerPage) * rowsPerPage > 0)
totalPageCount++;
if (count == 0) {
totalPageCount = 0;
pageNo = 0;
}
firstRow = (pageNo - 1) * rowsPerPage + 1;
if (count == 0) {
firstRow = 0;
}
lastRow = (pageNo) * rowsPerPage;
if (lastRow > count) {
lastRow = count;
}
}
public int getCount() {
return count;
}
public List<T> getResult() {
return result;
}
public int getPageNo() {
return pageNo;
}
public int getRowsPerPage() {
return rowsPerPage;
}
public int getTotalPageCount() {
return totalPageCount;
}
public void setPageNo(int pageNo) {
this.pageNo = pageNo;
}
public void setRowsPerPage(int rowsPerPage) {
this.rowsPerPage = rowsPerPage;
}
public int getFirstRow() {
return firstRow;
}
public int getLastRow() {
return lastRow;
}
public void setFirstRow(int firstRow) {
this.firstRow = firstRow;
}
public void setLastRow(int lastRow) {
this.lastRow = lastRow;
}
public void setCount(int count) {
this.count = count;
}
public void setTotalPageCount(int totalPageCount) {
this.totalPageCount = totalPageCount;
}
public void setResult(List<T> result) {
this.result = result;
}
public Object getObj() {
return obj;
}
public void setObj(Object obj) {
this.obj = obj;
}
public boolean isSuccess() {
return success;
}
public void setSuccess(boolean success) {
this.success = success;
}
public String getMessage() {
return message;
}
public void setMessage(String message) {
this.message = message;
}
public static int calc(int pageNo, int rowsPerPage, int count) {
if (pageNo <= 0)
pageNo = 1;
if (rowsPerPage <= 0)
rowsPerPage = 10;
// 当把最后一页数据删除以后,页码会停留在最后一个上必须减一
int totalPageCount = count / rowsPerPage;
if (pageNo > totalPageCount && (count % rowsPerPage == 0)) {
pageNo = totalPageCount;
}
if (pageNo - totalPageCount > 2) {
pageNo = totalPageCount + 1;
}
int firstRow = (pageNo - 1) * rowsPerPage;
if (firstRow < 0) {
firstRow = 0;
}
return firstRow;
}
}
IBaseDao接口实现了BaseDaoImpl
package com.platform.framework.dao.jpa;
import java.io.Serializable;
import java.util.List;
import javax.persistence.EntityManager;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Selection;
import javax.persistence.metamodel.EntityType;
import org.apache.log4j.Logger;
import com.google.common.base.Strings;
@SuppressWarnings({ "unchecked", "rawtypes" })
public class BaseDaoImpl<T> extends EntityManagerFactoryProxy implements IBaseDao {
private static Logger log = Logger.getLogger(BaseDaoImpl.class);
private int batchSize = 50;
public void setBatchSize(int batchSize) {
this.batchSize = batchSize;
}
public <E> E get(Class clazz, Serializable id) {
return (E) getEntityManager().find(clazz, id);
}
public void insert(Object entity) {
if (entity instanceof List) {
insertList((List) entity);
return;
} else if (entity instanceof Object[]) {
return;
}
try {
getEntityManager().persist(entity);
} catch (Exception e) {
e.printStackTrace();
}
}
public void insertList(List list) {
EntityManager entityManager = getEntityManager();
if (list == null || list.size() == 0) {
return;
}
int i = 0;
for (Object o : list) {
insert(o);
if (i % batchSize == 0) {
entityManager.flush();
}
i++;
}
log.debug(list.get(0).getClass() + "批量增加数据" + i + "条");
}
public void update(Object entity) {
if (entity instanceof List) {
this.updateList((List) entity);
return;
}
getEntityManager().merge(entity);
}
public void updateList(List list) {
for (Object entity : list) {
this.update(entity);
}
}
public void delete(Object entity) {
if (entity instanceof List) {
List list = (List) entity;
for (Object o : list) {
getEntityManager().remove(o);
}
} else {
getEntityManager().remove(entity);
}
}
public <E extends Serializable> List<E> query(String jpql) {
return getEntityManager().createQuery(jpql).getResultList();
}
public Integer updateJpql(String jpql) {
return getEntityManager().createQuery(jpql).executeUpdate();
}
public Integer updateSql(String sql) {
return getEntityManager().createNativeQuery(sql).executeUpdate();
}
public <E extends Serializable> List<E> queryBySql(String sql) {
return getEntityManager().createNativeQuery(sql).getResultList();
}
public <E extends Serializable> List<E> query(Class clazz, String hqlCondition) {
return getEntityManager().createQuery("select t from " + clazz.getName() + " as t where " + hqlCondition)
.getResultList();
}
public void delete(Class entity, String jpqlCondition) {
if (Strings.isNullOrEmpty(jpqlCondition)) {
jpqlCondition = "1=1";
}
int no = updateJpql("delete " + entity.getName() + " where " + jpqlCondition);
log.debug(entity.getName() + "删除" + no + "条数据");
}
public void delete(Class entity, List ids) {
String idName = getIdName(entity, getEntityManager());
StringBuffer sb = new StringBuffer();
sb.append(idName + " in(");
for (int i = 0; i < ids.size(); i++) {
sb.append("'" + ids.get(i) + "',");
}
String jpqlCondition = sb.substring(0, sb.length() - 1) + ")";
delete(entity, jpqlCondition);
}
public <E extends Serializable> List<E> query(String jpql, int firstResult, int maxResults) {
List result = getEntityManager().createQuery(jpql).setFirstResult(firstResult).setMaxResults(maxResults)
.getResultList();
return result;
}
public <E extends Serializable> List<E> queryBySql(String sql, int firstResult, int maxResults) {
return getEntityManager().createNativeQuery(sql).setFirstResult(firstResult).setMaxResults(maxResults)
.getResultList();
}
public <E extends Serializable> List<E> queryAll(Class clazz) {
CriteriaQuery criteriaQuery = getEntityManager().getCriteriaBuilder().createQuery(clazz);
criteriaQuery.from(clazz);
return getEntityManager().createQuery(criteriaQuery).getResultList();
}
public Page queryPageByJpql(String jpql, int pageNo, int rowsPerPage) {
if (pageNo <= 0)
pageNo = 1;
if (rowsPerPage <= 0)
rowsPerPage = 7;
log.debug("-----开始查询,页码:" + pageNo + ",每页显示:" + rowsPerPage + "----");
String countJpql = "select count(*) from (" + jpql + ")";
int count = getCount(countJpql).intValue();
// 当把最后一页数据删除以后,页码会停留在最后一个上必须减一
int totalPageCount = count / rowsPerPage;
if (pageNo > totalPageCount && (count % rowsPerPage == 0)) {
pageNo = totalPageCount;
}
if (pageNo - totalPageCount > 2) {
pageNo = totalPageCount + 1;
}
int firstResult = (pageNo - 1) * rowsPerPage;
if (firstResult < 0) {
firstResult = 0;
}
List result = getEntityManager().createQuery(jpql).setFirstResult(firstResult).setMaxResults(rowsPerPage)
.getResultList();
return new Page(count, pageNo, rowsPerPage, result);
}
public Long getCount(String jpql) {
return (Long) getEntityManager().createQuery(jpql).getResultList().get(0);
}
public void updateJpql(String jpql, List paramList) {
javax.persistence.Query query = getEntityManager().createQuery(jpql);
for (int i = 0; i < paramList.size(); i++) {
query.setParameter(i + 1, paramList.get(i));
}
query.executeUpdate();
}
public Long getCount(Query query) {
Selection selection = query.getCriteriaQuery().getSelection();
query.getCriteriaQuery().select(query.getCriteriaBuilder().count(query.getFrom()));
Long count = (Long) getEntityManager().createQuery(query.newCriteriaQuery()).getResultList().get(0);
query.getCriteriaQuery().select(selection);
return count;
}
public Page queryPage(Query query, int pageNo, int rowsPerPage) {
if (pageNo <= 0)
pageNo = 1;
if (rowsPerPage <= 0)
rowsPerPage = 7;
log.debug(query.getClazz() + "-----开始查询,页码:" + pageNo + ",每页显示:" + rowsPerPage + "----");
log.debug("查询条件:");
for (Predicate cri : query.getPredicates())
log.debug(cri);
int count = getCount(query).intValue();
// 当把最后一页数据删除以后,页码会停留在最后一个上必须减一
int totalPageCount = count / rowsPerPage;
if (pageNo > totalPageCount && (count % rowsPerPage == 0)) {
pageNo = totalPageCount;
}
if (pageNo - totalPageCount > 2) {
pageNo = totalPageCount + 1;
}
int firstResult = (pageNo - 1) * rowsPerPage;
if (firstResult < 0) {
firstResult = 0;
}
List result = getEntityManager().createQuery(query.newCriteriaQuery()).setFirstResult(firstResult)
.setMaxResults(rowsPerPage).getResultList();
return new Page(count, pageNo, rowsPerPage, result);
}
public <E extends Serializable> List<E> query(Query query, int firstResult, int maxResults) {
List result = getEntityManager().createQuery(query.newCriteriaQuery()).setFirstResult(firstResult)
.setMaxResults(maxResults).getResultList();
return result;
}
public <E extends Serializable> List<E> query(Query query) {
return getEntityManager().createQuery(query.newCriteriaQuery()).getResultList();
}
public static String getIdName(Class clazz, EntityManager entityManager) {
EntityType entityType = entityManager.getMetamodel().entity(clazz);
return entityType.getId(entityType.getIdType().getJavaType()).getName();
}
} IBaseDao接口
package com.platform.framework.dao.jpa;
import java.io.Serializable;
import java.util.List;
import javax.persistence.EntityManager;
@SuppressWarnings({ "rawtypes" })
public interface IBaseDao {
public EntityManager getEntityManager();
public <E> E get(Class clazz, Serializable id);
public void insert(Object entity);
public void update(Object entity);
public void updateList(List list);
public void delete(Object entity);
public void delete(Class entity, List ids);
public void delete(Class entity, String jpqlCondition);
public Long getCount(Query query);
public Long getCount(String jpql);
public Page queryPage(Query query, int pageNo, int rowsPerPage);
public <E extends Serializable> List<E> query(Query query, int firstResult, int maxResults);
public <E extends Serializable> List<E> query(Query query);
public <E extends Serializable> List<E> query(String jpql);
public <E extends Serializable> List<E> queryAll(Class clazz);
public <E extends Serializable> List<E> query(String jpql, int firstResult, int maxResults);
public <E extends Serializable> List<E> queryBySql(String sql);
public <E extends Serializable> List<E> queryBySql(String sql, int firstResult, int maxResults);
public <E extends Serializable> List<E> query(Class clazz, String hqlCondition);
public Integer updateSql(String sql);
public Integer updateJpql(String jpql);
public Page queryPageByJpql(String hql, int pageNo, int rowsPerPage);
public void updateJpql(String jpql, List paramList);
} 以上为个人经验，希望能给大家一个参考，也希望大家多多支持编程网。
