AbstractGenericDao.java

/**
 * Most of the code in the Qalingo project is copyrighted Hoteia and licensed
 * under the Apache License Version 2.0 (release version 0.8.0)
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 *                   Copyright (c) Hoteia, 2012-2014
 * http://www.hoteia.com - http://twitter.com/hoteia - contact@hoteia.com
 *
 */
package org.hoteia.qalingo.core.dao;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.ObjectOutputStream;
import java.sql.Blob;
import java.util.ArrayList;

import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;
import javax.persistence.Query;

import org.hibernate.Criteria;
import org.hibernate.Hibernate;
import org.hibernate.Session;
import org.hoteia.qalingo.core.fetchplan.FetchPlan;
import org.hoteia.qalingo.core.fetchplan.SpecificFetchMode;
import org.hoteia.qalingo.core.util.CoreUtil;

public abstract class AbstractGenericDao {  

	@PersistenceContext
	protected EntityManager em;

    public Session getSession() {
        return (Session) em.getDelegate();
    }
    
    protected String handleCodeValue(String code) {
        return CoreUtil.cleanEntityCode(code);
    }
    
    protected Criteria createDefaultCriteria(Class<?> entityClass) {
        return createDefaultCriteria(entityClass, null);
    }
    
    protected Criteria createDefaultCriteria(Class<?> entityClass, String alias) {
        Criteria criteria = getSession().createCriteria(entityClass, alias);
        criteria.setResultTransformer(Criteria.DISTINCT_ROOT_ENTITY);
        return criteria;
    }
    
    protected Query createSqlQuery(String queryString) {
        return em.createQuery(queryString);
    }
    
    protected Query createSqlQuery(Class<?> entityClass, String queryString) {
        return em.createQuery(queryString, entityClass);
    }
    
    protected Query createNativeQuery(String queryString) {
        return em.createNativeQuery(queryString);
    }
    
    protected Query createNativeQuery(Class<?> entityClass, String queryString) {
        return em.createNativeQuery(queryString, entityClass);
    }
    
    protected Query createNativeQuery(String resultSetMapping, String queryString) {
        return em.createNativeQuery(queryString, resultSetMapping);
    }
    
    protected FetchPlan handleSpecificFetchMode(Criteria criteria, Object... params){
        if (params != null) {
            FetchPlan globalFetchPlan = new FetchPlan(new ArrayList<SpecificFetchMode>());
            for (Object param : params) {
                if (param instanceof FetchPlan) {
                    FetchPlan fetchPlan = (FetchPlan) param;
                    for (SpecificFetchMode specificFetchMode : fetchPlan.getFetchModes()) {
                        if (!globalFetchPlan.getFetchModes().contains(specificFetchMode)) {
                            globalFetchPlan.getFetchModes().add(specificFetchMode);
                        }
                    }
                }
            }
            
            if(globalFetchPlan.getFetchModes() != null){
                for (SpecificFetchMode specificFetchMode : globalFetchPlan.getFetchModes()) {
                    if (specificFetchMode.getRequiredAlias() != null) {
                        // TODO : Denis : check duplicate entry are manage or not
                        criteria.createAlias(specificFetchMode.getRequiredAlias().getAssocationPath(), specificFetchMode.getRequiredAlias().getAlias(), specificFetchMode.getRequiredAlias().getJoinType());
                    }
                    criteria.setFetchMode(specificFetchMode.getAssocationPath(), specificFetchMode.getFetchMode());
                }
                return globalFetchPlan;
            }
            
        }
        return null;
    }
	
    public Blob convertObjectToBlobHibernate(Object object) throws IOException {
        ByteArrayOutputStream bos = new ByteArrayOutputStream();
        ObjectOutputStream oos = new ObjectOutputStream(bos);

        oos.writeObject(object);
        oos.flush();
        oos.close();
        bos.close();

        byte[] data = bos.toByteArray();

        Blob blob = Hibernate.getLobCreator(getSession()).createBlob(data);
        return blob;
    }
    
}