package cn.itcast.ems.orm.hibernate; import java.io.Serializable; import java.util.ArrayList; import java.util.LinkedHashSet; import java.util.List; import java.util.Map; import java.util.Set; import org.hibernate.Criteria; import org.hibernate.Hibernate; import org.hibernate.Query; import org.hibernate.Session; import org.hibernate.SessionFactory; import org.hibernate.criterion.CriteriaSpecification; import org.hibernate.criterion.Criterion; import org.hibernate.criterion.Restrictions; import org.hibernate.metadata.ClassMetadata; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.util.Assert; import cn.itcast.ems.utils.ReflectionUtils; /** * 在 Service 层直接使用, 也可以扩展泛型 DAO 子类使用 * T: Dao 操作的对象类型 * PK: 主键类型 * @author Administrator * */ public class SimpleHibernateDao<T, PK extends Serializable> { protected Logger logger = LoggerFactory.getLogger(getClass()); protected SessionFactory sessionFactory; protected Class<T> entityClass; /** * 用于 Dao 层子类使用的构造函数 * 通过子类的泛型定义取得对象类型 Class * * 例如: public class EmployeeDao extends SimpleHibernateDao<User, String> */ public SimpleHibernateDao(){ this.entityClass = ReflectionUtils.getSuperGenericType(getClass()); } /** * 用于省略 Dao 层, 在 Service 层直接使用通用 SimpleHibernateDao 的构造函数 * 在构造函数中定义对象类型 Class * @param sessionFactory * @param entityClass */ public SimpleHibernateDao(SessionFactory sessionFactory,Class<T> entityClass) { this.sessionFactory = sessionFactory; this.entityClass = entityClass; } public SessionFactory getSessionFactory() { return sessionFactory; } public void setSessionFactory(SessionFactory sessionFactory) { this.sessionFactory = sessionFactory; } /** * 获取当前 Session * @return */ public Session getSession(){ return sessionFactory.getCurrentSession(); } /** * 保存新增或修改的对象 * @param entity */ public void save(T entity){ Assert.notNull(entity, "entity 不能为空"); getSession().saveOrUpdate(entity); logger.debug("save entity: {}", entity); } public void delete(PK id){ Assert.notNull(id, "id 不能为空"); delete(get(id)); logger.debug("delete entity {},id is {}", entityClass.getSimpleName(), id); } /** * 删除对象 * @param entity: 持久化对象或"瞬态"对象 */ public void delete(T entity){ Assert.notNull(entity, "entity 不能为空"); getSession().delete(entity); logger.debug("delete entity: {}", entity); } /** * 按 id 获取对象 * @param id * @return */ @SuppressWarnings("unchecked") public T get(PK id){ Assert.notNull(id, "id不能为空"); return (T) getSession().get(entityClass, id); } /** * 获取对象的主键名. * @return */ public String getIdName(){ ClassMetadata meta = getSessionFactory().getClassMetadata(entityClass); return meta.getIdentifierPropertyName(); } /** * 通过 Set 将不唯一的对象列表唯一化 * 主要用于 HQL/Criteria 预加载关联集合形成重复记录, 又不方便使用 distinct 查询语句时 * @param <X> * @param list * @return */ @SuppressWarnings("unchecked") public <X> List<X> distinct(List list){ Set<X> set = new LinkedHashSet<X>(list); return new ArrayList<X>(set); } /** * 为 Criteria 添加 distinct transformer * @param criteria * @return */ public Criteria distinct(Criteria criteria){ criteria.setResultTransformer(CriteriaSpecification.DISTINCT_ROOT_ENTITY); return criteria; } /** * 为 Query 添加 distinct transformer * @param query * @return */ public Query distinct(Query query){ query.setResultTransformer(CriteriaSpecification.DISTINCT_ROOT_ENTITY); return query; } /** * 初始化对象. * 使用 load() 方法得到的仅是对象的代理, 在传到视图层前需要进行初始化 * * 只初始化 entity 的直接属性, 但不会初始化延迟加载的关联集合和属性 * 如需初始化关联属性, 可执行: Hibernate.initialize(user.getRoles()); * @param entity */ public void initEntity(T entity){ Hibernate.initialize(entity); } public void initEntity(List<T> entityList){ for(T entity: entityList){ Hibernate.initialize(entity); } } /** * 根据 Criterion 条件创建 Criteria * @param criterions * @return */ public Criteria createCriteria(Criterion ...criterions){ Criteria criteria = getSession().createCriteria(entityClass); for(Criterion c: criterions){ criteria.add(c); } return criteria; } /** * 按 Criteria 查询唯一对象 * @param criterions: 数量可变的 Criterion * @return */ public T findUnique(Criterion...criterions){ return (T) createCriteria(criterions).uniqueResult(); } /** * 按 Criteria 查询对象列表 * @param criterions: 数量可变的 Criterion * @return */ @SuppressWarnings("unchecked") public List<T> find(Criterion... criterions){ return createCriteria(criterions).list(); } /** * 根据查询 HQL 与参数列表创建 Query 对象 * @param queryString * @param values * @return */ public Query createQuery(String queryString, Map<String, Object> values){ //String hql = "FROM Employee e where e.loginname = :loginname"; Assert.hasText(queryString, "queryString不能为空"); Query query = getSession().createQuery(queryString); if(values != null){ query.setProperties(values); } return query; } /** * 根据查询 HQL 与参数列表创建 Query 对象 * @param queryString * @param values: 数来那个可变的参数, 按顺序绑定 * @return */ public Query createQuery(String queryString, Object... values){ Assert.hasText(queryString, "queryString不能为空"); Query query = getSession().createQuery(queryString); if(values != null){ for(int i = 0; i < values.length; i++){ query.setParameter(i, values[i]); } } return query; } /** * 执行 hql 进行批量修改/删除操作 * @param hql * @param values * @return */ public int batchExecute(String hql, Map<String, Object> values){ return createQuery(hql, values).executeUpdate(); } /** * 执行 hql 进行批量修改/删除操作 * @param hql * @param values * @return */ public int batchExecute(String hql, Object... values){ return createQuery(hql, values).executeUpdate(); } /** * 按 HQL 查询唯一对象 * @param <X> * @param hql * @param values * @return */ @SuppressWarnings("unchecked") public <X> X findUnique(String hql, Map<String, Object> values){ return (X) createQuery(hql, values).uniqueResult(); } /** * 按 HQL 查询唯一对象 * @param <X> * @param hql * @param values * @return */ @SuppressWarnings("unchecked") public <X> X findUnique(String hql, Object... values){ return (X) createQuery(hql, values).uniqueResult(); } /** * 按 HQL 查询对象列表 * @param <X> * @param hql * @param values * @return */ @SuppressWarnings("unchecked") public <X> List<X> find(String hql, Map<String, Object> values){ return createQuery(hql, values).list(); } /** * 按 HQL 查询对象列表 * @param <X> * @param hql * @param values * @return */ @SuppressWarnings("unchecked") public <X> List<X> find(String hql, Object... values){ return createQuery(hql, values).list(); } /** * 按 id 列表获取对象列表 * @param ids * @return */ public List<T> findByIds(List<PK> ids){ return find(Restrictions.in(getIdName(), ids)); } /** * 按属性查找唯一对象, 匹配方式为相等 * @param propertyName * @param value * @return */ @SuppressWarnings("unchecked") public T findUniqueBy(String propertyName, Object value){ Assert.hasText(propertyName, "propertyName不能为空"); Criterion criterion = Restrictions.eq(propertyName, value); return (T) createCriteria(criterion).uniqueResult(); } /** * 按属性查找对象列表, 匹配方式为相等 * @param propertyName * @param value * @return */ public List<T> findBy(String propertyName, Object value){ Assert.hasText(propertyName, "propertyName不能为空"); Criterion criterion = Restrictions.eq(propertyName, value); return find(criterion); } /** * 获取全部对象 * @return */ public List<T> getAll(){ return find(); } } |