source: tapas/persistence/implementation/com/ether/dao/DomainAccessObjectImpl.java @ 376

package com.ether.dao;

import static com.ether.PersistenceHelper.convert;
import static com.ether.PersistenceHelper.uniqueResult;

import java.io.Serializable;
import java.lang.reflect.ParameterizedType;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

import org.hibernate.Criteria;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.criterion.DetachedCriteria;
import org.hibernate.criterion.Example;
import org.hibernate.criterion.Projections;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.springframework.dao.DataAccessException;

import com.ether.PersistenceException;

public class DomainAccessObjectImpl<DO, PK extends Serializable>
        implements DomainAccessObject<DO, PK>
{
    public DomainAccessObjectImpl(final Class<DO> domainObjectClass) {
        _domainObjectClass = domainObjectClass;
    }

    // ************************************************************************** //
    // ******************************* REQUESTS ********************************* //
    // ************************************************************************** //

    // ******************************* SELECT BY CRITERIA **//    
    @Nullable
    public final DO selectByCriteria(final DetachedCriteria criteria)
            throws PersistenceException
    {
        return selectByCriteria(getDomainObjectClass(), criteria);
    }

    @Nullable
    public final <O> O selectByCriteria(final Class<O> objectClass, final DetachedCriteria criteria) throws PersistenceException
    {
        return (O) criteria.getExecutableCriteria(_sessionFactory.getCurrentSession()).uniqueResult();  
    }

    @NotNull
    public final <O> List<O> selectAllByCriteria(final Class<O> objectClass, final DetachedCriteria criteria) throws PersistenceException
    {
        final List<O> result =  (List<O>) criteria.getExecutableCriteria(_sessionFactory.getCurrentSession()).list();
        return null != result ? result : new ArrayList<O>();
    }

    @NotNull
    public final List<DO> selectAllByCriteria(final DetachedCriteria criteria)
            throws PersistenceException
    {
        return selectAllByCriteria(getDomainObjectClass(), criteria);
    }

    
    // ******************************* INSERT **//    
    @NotNull
    public final PK insert(final DO domainObject) throws PersistenceException
    {
        try
        {
            final Session session = _sessionFactory.getCurrentSession();
            final PK primaryKey = (PK) session.save(domainObject);
            session.flush();
            return primaryKey;
        }
        catch (DataAccessException e)
        {
            throw convert(e);
        }
    }

    public final void insertAll(@NotNull final Collection<DO> domainObjects)
            throws PersistenceException
    {
        try
        {
            final Session session = _sessionFactory.getCurrentSession();
            int i = 0;
            for (final DO domainObject : domainObjects)
            {
                session.save(domainObject);
                // Avoid OutOfMemoryError
                if (i >= 500)
                {
                    session.flush();
                    session.clear();
                    i = 0;
                }
            }
            session.flush();
        }
        catch (DataAccessException e)
        {
            throw convert(e);
        }
    }

    // ******************************* UPDATE **//    
    public void update(@NotNull final DO domainObject)
            throws PersistenceException
    {
        try
        {
            final Session session = _sessionFactory.getCurrentSession();
            session.update(domainObject);
            session.flush();
        }
        catch (DataAccessException e)
        {
            throw convert(e);
        }
    }

    public final void updateAll(final Collection<? extends DO> domainObjects)
            throws PersistenceException
    {
        try
        {
            final Session session = _sessionFactory.getCurrentSession();
            for (final DO domainObject : domainObjects)
                session.saveOrUpdate(domainObject);
            session.flush();
        }
        catch (DataAccessException e)
        {
            throw convert(e);
        }
    }

    // ******************************* UPGRADE **//    
    public final void upgrade(final DO domainObject)
            throws PersistenceException
    {
        try
        {
            final Session session = _sessionFactory.getCurrentSession();
            session.refresh(domainObject);
            session.flush();
        }
        catch (DataAccessException e)
        {
            throw convert(e);
        }
    }

    public final void upgradeAll(final Collection<DO> domainObjects)
            throws PersistenceException
    {
        try
        {
            final Session session = _sessionFactory.getCurrentSession();
            for (final DO domainObject : domainObjects)
            {
                session.refresh(domainObject);
            }
            session.flush();
        }
        catch (DataAccessException e)
        {
            throw convert(e);
        }
    }

    // ******************************* DELETE **//    
    public void deleteByPrimaryKey(final PK primaryKey)
            throws PersistenceException
    {
        try
        {
            final Session session = _sessionFactory.getCurrentSession();
            final Object domainObject = session.get(getDomainObjectClass(), primaryKey);
            if(null != domainObject)
                session.delete(domainObject);
            else
                throw new PersistenceException(PersistenceException.PersistenceCode.OBJECT_TO_DELETE_NOT_FOUND);
            session.flush();
        }
        catch (DataAccessException e)
        {
            throw convert(e);
        }
    }

    public final void delete(@NotNull final DO domainObject)
            throws PersistenceException
    {
        try
        {
            final Session session = _sessionFactory.getCurrentSession();
            session.delete(domainObject);
            session.flush();
        }
        catch (DataAccessException e)
        {
            throw convert(e);
        }
    }

    public final void deleteAll(final Collection<DO> domainObjects)
            throws PersistenceException
    {
        try
        {
            final Session session = _sessionFactory.getCurrentSession();
            for (final DO domainObject : domainObjects)
                session.delete(domainObject);
            session.flush();
        }
        catch (DataAccessException e)
        {
            throw convert(e);
        }
    }

    // ******************************* SELECT **//    
    public final DO selectByPrimaryKey(@NotNull final PK primaryKey)
            throws PersistenceException
    {
        try
        {
            final Session session = _sessionFactory.getCurrentSession();
            return (DO) session.get(getDomainObjectClass(), primaryKey);
        }
        catch (DataAccessException e)
        {
            throw convert(e);
        }
    }

    public final DO selectByExample(final DO domainObject)
            throws PersistenceException
    {
        final Session session = _sessionFactory.getCurrentSession();
        final Criteria executableCriteria = session.createCriteria(domainObject.getClass());
        executableCriteria.add(Example.create(domainObject));
        return uniqueResult((Collection<DO>) executableCriteria.list());
    }

    public final List<DO> selectAllByExample(final DO domainObject)
            throws PersistenceException
    {
        final Session session = _sessionFactory.getCurrentSession();
        final Criteria executableCriteria = session.createCriteria(domainObject.getClass());
        executableCriteria.add(Example.create(domainObject));
        return (List<DO>) executableCriteria.list();
    }

    public final List<DO> selectAll() throws PersistenceException
    {
        try
        {
            final Session session = _sessionFactory.getCurrentSession();
            final Criteria executableCriteria = session.createCriteria(getDomainObjectClass());     
            return (List<DO>) executableCriteria.list();
        }
        catch (DataAccessException e)
        {
            throw convert(e);
        }
    }

    // ******************************* PAGINATE **//
    protected <T> List<T> selectPage(@Nullable final Integer firstResult, @Nullable final Integer maxResults, final DetachedCriteria detachedCriteria)
            throws PersistenceException
    {
        final Session session = _sessionFactory.getCurrentSession();
        final Criteria criteria = detachedCriteria.getExecutableCriteria(session);

        if (null != maxResults)
        {
            criteria.setMaxResults(maxResults);
            if (null != firstResult)
                criteria.setFirstResult(firstResult);
        }
        return (List<T>) criteria.list();
    }

    protected Integer count(final DetachedCriteria detachedCriteria)
            throws PersistenceException
    {
        final Session session = _sessionFactory.getCurrentSession();
        final Criteria executableCriteria = detachedCriteria.getExecutableCriteria(session);
        executableCriteria.setProjection(Projections.rowCount());
        return (Integer) executableCriteria.uniqueResult();
    }
    // ************************************************************************** //
    // ************************************************************************** //
    
    @SuppressWarnings("unchecked")
    public final Class<DO> getDomainObjectClass()
    {
        if (_domainObjectClass == null)
        {
            if (getClass().getGenericSuperclass() instanceof ParameterizedType)
                _domainObjectClass = (Class<DO>) ((ParameterizedType) getClass().getGenericSuperclass()).getActualTypeArguments()[0];
            else
                throw new IllegalArgumentException("domain class is not null and DAO is not parameterized !");
        }

        return _domainObjectClass;
    }

    public final void setDomainObjectClass(final Class<DO> domainObjectClass)
    {
        _domainObjectClass = domainObjectClass;
    }

    public final void setSessionFactory(final SessionFactory sessionFactory)
    {
        _sessionFactory = sessionFactory;
    }

    public final SessionFactory getSessionFactory()
    {
        return _sessionFactory;
    }

    protected static final Integer MAX_RESULTS_FOR_PAGINATION = 100;

    private SessionFactory _sessionFactory;
    private Class<DO> _domainObjectClass;
}