source: ether_statistics/persistence/implementation/com/ether/dao/simulation/SimulationDAOImpl.java @ 569

package com.ether.dao.simulation;

import com.ether.PersistenceException;
import com.ether.dao.SearchableAbstractHibernateMappingDAO;
import com.ether.simulation.Simulation;
import com.ether.simulation.SimulationFilter;
import org.hibernate.criterion.DetachedCriteria;
import org.hibernate.criterion.Order;
import org.hibernate.criterion.Projections;
import org.hibernate.criterion.Restrictions;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.springframework.util.StringUtils;

import java.util.ArrayList;
import java.util.List;

/**
 * @author vmipsl
 * @date 13 juillet 2012
 */
public class SimulationDAOImpl
        extends SearchableAbstractHibernateMappingDAO<Simulation, Integer, SimulationFilter>
        implements SimulationDAO
{
    protected SimulationDAOImpl()
    {
        super( Simulation.class, Integer.class );
    }

    @Nullable
    public Simulation getSimulationByLocation( @NotNull final String location )
            throws PersistenceException
    {
        final DetachedCriteria criteria = DetachedCriteria.forClass( Simulation.class )
                .add( Restrictions.eq( "location", location ) );
        return selectByCriteria( Simulation.class, criteria );
    }

    @Nullable
    public List<List<String>> getAllModels()
            throws PersistenceException
    {
        final DetachedCriteria criteriaTitles = createCriteriaForModels( "Title", true );
        final List<Object[]> listTitles = selectAllByCriteria( Object[].class, criteriaTitles );

        final DetachedCriteria criteriaVersions = createCriteriaForModels( "Chimere_version", true );
        final List<Object[]> listVersions = selectAllByCriteria( Object[].class, criteriaVersions );

        final DetachedCriteria criteriaLevels = createCriteriaForModels( "bottom_top", false );
        final List<Object[]> listLevels = selectAllByCriteria( Object[].class, criteriaLevels );

        final List<List<String>> modelsList = new ArrayList<List<String>>();
        int i = 0;
        for( final Object[] title : listTitles )
        {
            final String version = (String) listVersions.get( i )[1];
            final String level = (String) listLevels.get( i )[1];
            final List<String> values = new ArrayList<String>( 3 );
            values.add( title[0].toString() );
            values.add( (String) title[1] );
            values.add( version );
            values.add( (String) title[2] );
            values.add( level );
            i++;
            modelsList.add( values );
        }

        return modelsList;
    }

    @Nullable
    public String getLocationByModelId( @NotNull final Integer modelId )
            throws PersistenceException
    {
        final DetachedCriteria criteria = DetachedCriteria.forClass( Simulation.class )
                .add( Restrictions.idEq( modelId ) )
                .setProjection( Projections.property( "location" ) );

        return selectByCriteria( String.class, criteria );
    }

    @Override
    protected DetachedCriteria searchCriteria( @NotNull final SimulationFilter filter, final boolean isForCount )
    {
        final DetachedCriteria criteria = DetachedCriteria.forClass( Simulation.class );

        if( StringUtils.hasText( filter.getSearchText() ) )
            criteria.add( Restrictions.or( escapedLike( "location", filter.getSearchText() ), escapedLike( "xmlnsHeader", filter.getSearchText() ) ) );

        if( !isForCount )
            criteria.addOrder( Order.asc( filter.getSort() ) );
        return criteria;
    }

    @NotNull
    private DetachedCriteria createCriteriaForModels( @NotNull final String attributeName, final boolean isAttribute )
    {
        final DetachedCriteria criteria = DetachedCriteria.forClass( Simulation.class, "simulation" );
        if( isAttribute )
            criteria.createCriteria( "attribute", "attribute" )
                    .add( Restrictions.eq( "attribute.name", attributeName ) )
                    .setProjection( Projections.projectionList()
                            .add( Projections.property( "simulation.id" ) )
                            .add( Projections.property( "attribute.value" ) )
                            .add( Projections.property( "simulation.dates" ) )
                    );
        else
            criteria.createCriteria( "dimension", "dimension" )
                    .add( Restrictions.eq( "dimension.name", attributeName ) )
                    .setProjection( Projections.projectionList()
                            .add( Projections.property( "simulation.id" ) )
                            .add( Projections.property( "dimension.length" ) )
                            .add( Projections.property( "simulation.dates" ) )
                    );

        return criteria.addOrder( Order.asc( "simulation.id" ) );
    }
}