com.bplow.look.bass.dao.HibernateDao.java Source code

Java tutorial

Introduction

Here is the source code for com.bplow.look.bass.dao.HibernateDao.java

Source

/**
 * Copyright (c) 2005-2009 springside.org.cn
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * 
 * $Id: HibernateDao.java,v 1.2 2010/08/16 07:04:19 wxl Exp $
 */
package com.bplow.look.bass.dao;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang.StringUtils;
import org.hibernate.Criteria;
import org.hibernate.Query;
import org.hibernate.SessionFactory;
import org.hibernate.criterion.Criterion;
import org.hibernate.criterion.Disjunction;
import org.hibernate.criterion.Order;
import org.hibernate.criterion.Restrictions;
import org.springframework.dao.DataAccessException;
import org.springframework.util.Assert;

import com.bplow.look.bass.IPagination;
import com.bplow.look.bass.SimplePagination;
import com.bplow.look.bass.dao.PropertyFilter.MatchType;

/**
 * ?SpringSideHibernat DAO.
 * 
 * ,?.
 * ?Service,?DAO?,?.
 * 
 * @param <T> DAO?
 * @param <PK> 
 * 
 * @author calvin
 */
public class HibernateDao<T, PK extends Serializable> extends SimpleHibernateDao<T, PK> {
    /**
     * Dao?.
     * ??Class.
     * eg.
     * public class UserDao extends HibernateDao<User, Long>{
     * }
     */
    public HibernateDao() {
        super();
    }

    /**
     * ?Dao, ServiceHibernateDao.
     * Class.
     * eg.
     * HibernateDao<User, Long> userDao = new HibernateDao<User, Long>(sessionFactory, User.class);
     */
    public HibernateDao(final SessionFactory sessionFactory, final Class<T> entityClass) {
        super(sessionFactory, entityClass);
    }

    //  //

    /**
     * ?.
     */
    public Page<T> getAll(final Page<T> page) {
        return findPage(page);
    }

    /**
     * HQL.
     * 
     * @param page ?.??orderBy?.
     * @param hql hql?.
     * @param values ????,?.
     * 
     * @return , ??.
     */
    @SuppressWarnings("unchecked")
    public Page<T> findPage(final Page<T> page, final String hql, final Object... values) {
        Assert.notNull(page, "page?");

        Query q = createQuery(hql, values);

        if (page.isAutoCount()) {
            long totalCount = countHqlResult(hql, values);
            page.setTotalCount(totalCount);
        }

        setPageParameter(q, page);
        List result = q.list();
        page.setResult(result);
        return page;
    }

    /**
     * HQL.
     * 
     * @param page ?.(??orderBy?)
     * @param hql hql?.
     * @param values ???,??.
     * 
     * @return , ??.
     */
    @SuppressWarnings("unchecked")
    public Page<T> findPage(final Page<T> page, final String hql, final Map<String, Object> values) {
        Assert.notNull(page, "page?");

        Query q = createQuery(hql, values);

        if (page.isAutoCount()) {
            long totalCount = countHqlResult(hql, values);
            page.setTotalCount(totalCount);
        }

        setPageParameter(q, page);

        List result = q.list();
        page.setResult(result);
        return page;
    }

    /**
     * Criteria.
     * 
     * @param page ?.
     * @param criterions ???Criterion.
     * 
     * @return .??.
     */
    @SuppressWarnings("unchecked")
    public Page<T> findPage(final Page<T> page, final Criterion... criterions) {
        Assert.notNull(page, "page?");

        Criteria c = createCriteria(criterions);

        if (page.isAutoCount()) {
            int totalCount = 0/*countCriteriaResult(c)*/;
            page.setTotalCount(totalCount);
        }

        setPageParameter(c, page);
        List result = c.list();
        page.setResult(result);
        return page;
    }

    /**
     * ?Query,.
     */
    protected Query setPageParameter(final Query q, final Page<T> page) {
        //hibernatefirstResult??0
        q.setFirstResult(page.getFirst() - 1);
        q.setMaxResults(page.getPageSize());
        return q;
    }

    /**
     * ?Criteria,.
     */
    protected Criteria setPageParameter(final Criteria c, final Page<T> page) {
        //hibernatefirstResult??0
        c.setFirstResult(page.getFirst() - 1);
        c.setMaxResults(page.getPageSize());

        if (page.isOrderBySetted()) {
            String[] orderByArray = StringUtils.split(page.getOrderBy(), ',');
            String[] orderArray = StringUtils.split(page.getOrder(), ',');

            Assert.isTrue(orderByArray.length == orderArray.length,
                    "???,????");

            for (int i = 0; i < orderByArray.length; i++) {
                if (Page.ASC.equals(orderArray[i])) {
                    c.addOrder(Order.asc(orderByArray[i]));
                } else {
                    c.addOrder(Order.desc(orderByArray[i]));
                }
            }
        }
        return c;
    }

    /**
     * countHql.
     * 
     * ???hql?,??hql?count?.
     */
    protected Integer countHqlResult(final String hql, final Object... values) {
        Integer count = 0;
        String fromHql = hql;
        //select??order by???count,?.
        fromHql = "from " + StringUtils.substringAfter(fromHql, "from");
        fromHql = StringUtils.substringBefore(fromHql, "order by");

        String countHql = "select count(*) " + fromHql;

        try {
            count = findUnique(countHql, values);
        } catch (Exception e) {
            throw new RuntimeException("hql can't be auto count, hql is:" + countHql, e);
        }
        return count;
    }

    /**
     * countHql.
     * 
     * ???hql?,??hql?count?.
     */
    protected long countHqlResult(final String hql, final Map<String, Object> values) {
        Long count = 0L;
        String fromHql = hql;
        //select??order by???count,?.
        fromHql = "from " + StringUtils.substringAfter(fromHql, "from");
        fromHql = StringUtils.substringBefore(fromHql, "order by");

        String countHql = "select count(*) " + fromHql;

        try {
            count = findUnique(countHql, values);
        } catch (Exception e) {
            throw new RuntimeException("hql can't be auto count, hql is:" + countHql, e);
        }

        return count;
    }

    /**
     * countCriteria.
     */
    /*@SuppressWarnings("unchecked")
    protected int countCriteriaResult(final Criteria c) {
       CriteriaImpl impl = (CriteriaImpl) c;
        
       // Projection?ResultTransformer?OrderBy??,??Count?
       Projection projection = impl.getProjection();
       ResultTransformer transformer = impl.getResultTransformer();
        
       List<CriteriaImpl.OrderEntry> orderEntries = null;
       try {
     orderEntries = (List) ReflectionUtils.getFieldValue(impl, "orderEntries");
     ReflectionUtils.setFieldValue(impl, "orderEntries", new ArrayList());
       } catch (Exception e) {
     logger.error("??:{}", e.getMessage());
       }
        
       // Count
       int totalCount = (Integer) c.setProjection(Projections.rowCount()).uniqueResult();
        
       // ?Projection,ResultTransformerOrderBy??
       c.setProjection(projection);
        
       if (projection == null) {
     c.setResultTransformer(CriteriaSpecification.ROOT_ENTITY);
       }
       if (transformer != null) {
     c.setResultTransformer(transformer);
       }
       try {
     ReflectionUtils.setFieldValue(impl, "orderEntries", orderEntries);
       } catch (Exception e) {
     logger.error("??:{}", e.getMessage());
       }
        
       return totalCount;
    }*/

    // ? //

    /**
     * ,????.
     * 
     * @param matchType ??,????PropertyFilterMatcheType enum.
     */
    public List<T> findBy(final String propertyName, final Object value, final MatchType matchType) {
        Criterion criterion = buildPropertyFilterCriterion(propertyName, value, value.getClass(), matchType);
        return find(criterion);
    }

    /**
     * ?.
     */
    public List<T> find(List<PropertyFilter> filters) {
        Criterion[] criterions = buildPropertyFilterCriterions(filters);
        return find(criterions);
    }

    /**
     * ?.
     */
    public Page<T> findPage(final Page<T> page, final List<PropertyFilter> filters) {
        Criterion[] criterions = buildPropertyFilterCriterions(filters);
        return findPage(page, criterions);
    }

    /**
     * ?Criterion,.
     */
    protected Criterion[] buildPropertyFilterCriterions(final List<PropertyFilter> filters) {
        List<Criterion> criterionList = new ArrayList<Criterion>();
        for (PropertyFilter filter : filters) {
            if (!filter.isMultiProperty()) { //??.
                Criterion criterion = buildPropertyFilterCriterion(filter.getPropertyName(),
                        filter.getPropertyValue(), filter.getPropertyType(), filter.getMatchType());
                criterionList.add(criterion);
            } else {//??,or?.
                Disjunction disjunction = Restrictions.disjunction();
                for (String param : filter.getPropertyNames()) {
                    Criterion criterion = buildPropertyFilterCriterion(param, filter.getPropertyValue(),
                            filter.getPropertyType(), filter.getMatchType());
                    disjunction.add(criterion);
                }
                criterionList.add(disjunction);
            }
        }
        return criterionList.toArray(new Criterion[criterionList.size()]);
    }

    /**
     * ??Criterion,.
     */
    protected Criterion buildPropertyFilterCriterion(final String propertyName, final Object propertyValue,
            final Class<?> propertyType, final MatchType matchType) {
        Assert.hasText(propertyName, "propertyName?");
        Criterion criterion = null;
        //      try {
        //         //entity property.
        //         Object realValue = ReflectionUtils.convertValue(propertyValue, propertyType);
        //
        //         //?MatchTypecriterion
        //         if (MatchType.EQ.equals(matchType)) {
        //            criterion = Restrictions.eq(propertyName, realValue);
        //         }
        //         if (MatchType.LIKE.equals(matchType)) {
        //            criterion = Restrictions.like(propertyName, (String) realValue, MatchMode.ANYWHERE);
        //         }
        //         if (MatchType.LE.equals(matchType)) {
        //            criterion = Restrictions.le(propertyName, realValue);
        //         }
        //         if (MatchType.LT.equals(matchType)) {
        //            criterion = Restrictions.lt(propertyName, realValue);
        //         }
        //         if (MatchType.GE.equals(matchType)) {
        //            criterion = Restrictions.ge(propertyName, realValue);
        //         }
        //         if (MatchType.GT.equals(matchType)) {
        //            criterion = Restrictions.gt(propertyName, realValue);
        //         }
        //      } catch (Exception e) {
        //         throw ReflectionUtils.convertToUncheckedException(e);
        //      }
        return criterion;
    }

    /**
     * ??.
     * 
     * ,(value)?(orgValue)?.
     */
    public boolean isPropertyUnique(final String propertyName, final Object newValue, final Object oldValue) {
        if (newValue == null || newValue.equals(oldValue))
            return true;
        Object object = findUniqueBy(propertyName, newValue);
        return (object == null);
    }

    /**
     * hibernate 
     * 
     */
    public IPagination queryForPagination(String hql, String hqlCount, int firstResult, int maxResults)
            throws DataAccessException {
        IPagination pagination = new SimplePagination(firstResult, maxResults);
        String fromHql = hql;
        //select??order by???count,?.
        fromHql = "from " + StringUtils.substringAfter(fromHql, "from");
        fromHql = StringUtils.substringBefore(fromHql, "order by");

        Query q = createQuery(hql);
        Long totalCount = 0L;
        if (StringUtils.isNotEmpty(hqlCount)) {
            //totalCount =countHqlResult(hqlCount);
            totalCount = (Long) this.getSession().createQuery(hqlCount).uniqueResult();
        } else
            totalCount = (Long) this.getSession().createQuery(" select count(*) " + fromHql).uniqueResult();
        ;

        q.setFirstResult(firstResult);
        q.setMaxResults(maxResults);
        List result = q.list();
        pagination.setAllCount(Integer.parseInt(totalCount.toString()));
        pagination.setResults(result);

        return pagination;
    }

    /**
     * ??
     */
    public Integer getSeqNextValue(String sequencesName) {

        String seqstr = "select " + sequencesName + ".Nextval from dual ";
        Integer val = Integer.valueOf(this.getSession().createSQLQuery(seqstr).uniqueResult().toString());

        return val;
    }

}