gvSIG projects pool

package org.gvsig.vcsgis.swing.impl.importhistory.process;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.expressionevaluator.Expression;

import org.gvsig.expressionevaluator.ExpressionBuilder;

import org.gvsig.expressionevaluator.ExpressionUtils;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.FeatureQuery;

import org.gvsig.fmap.dal.feature.FeatureSet;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.task.SimpleTaskStatus;

import org.gvsig.vcsgis.lib.VCSGisEntity;

import static org.gvsig.vcsgis.lib.VCSGisManager.ERR_EXCEPTION;

import static org.gvsig.vcsgis.lib.VCSGisManager.ERR_OK;

import org.gvsig.vcsgis.lib.workspace.VCSGisWorkspace;

import org.gvsig.vcsgis.swing.impl.importhistory.AbstractImportProcess;

/**

 *

 * @author gvSIG Team

 */

@SuppressWarnings("UseSpecificCatch")

public class ImportHistoryWithRevision_minimize_memory extends AbstractImportProcess {

    protected String fieldEfectiveDateName;

    protected String fieldRevisionName;

    private String sourceFieldRevisionName;

    private String sourceFieldEfectiveDateName;

    public ImportHistoryWithRevision_minimize_memory(

            boolean onlyLocal,

            VCSGisWorkspace workspace,

            FeatureStore sourceStore,

            VCSGisEntity targetEntity,

            String fieldCodeName,

            String fieldRevisionName,

            String fieldEfectiveDateName,

            String fieldOrderName,

            Expression deletedCondition,

            SimpleTaskStatus status

    ) {

        super(

                onlyLocal,

                workspace, 

                sourceStore, 

                targetEntity, 

                fieldCodeName, 

                fieldOrderName, 

                deletedCondition, 

                status

        );

        this.fieldEfectiveDateName = fieldEfectiveDateName;

        this.fieldRevisionName = fieldRevisionName;

    }

    @Override

    protected FeatureQuery getSourceQuery() throws DataException {

        FeatureQuery sourceQuery = sourceStore.createFeatureQuery();

        ExpressionBuilder expBuilder = ExpressionUtils.createExpressionBuilder();

        sourceQuery.getOrder().add(sourceFieldRevisionName, true);

        String sourceFieldOrderName = this.getSourceFieldOrderName();

        if (sourceFieldOrderName!=null ) {

            sourceQuery.getOrder().add(sourceFieldOrderName, true);

        }

        return sourceQuery;

    }

    @Override

    public int process() {

        if (StringUtils.isBlank(fieldRevisionName)) {

            LOGGER.warn("fieldRevisionName can't be blank,");

            return ERR_EXCEPTION;

        }

        FeatureSet sourceFeatures = null;

        try {

            status.message("Preparing data...");

            FeatureAttributeDescriptor sourceAttrRevisionName = getSourceFeatureType().getAttributeDescriptor(fieldRevisionName);

            if (sourceAttrRevisionName == null) {

                LOGGER.warn("Field '" + fieldRevisionName + "' (fieldRevisionName) has not been found in the source table.");

                return ERR_EXCEPTION;

            }

            sourceFieldRevisionName = sourceAttrRevisionName.getName();

            if( StringUtils.isNotBlank(fieldEfectiveDateName) ) {

                sourceFieldEfectiveDateName = getSourceFeatureType().getAttributeDescriptor(fieldEfectiveDateName).getName();

            }

            FeatureQuery theSourceQuery = this.getSourceQuery();

            sourceFeatures = sourceStore.getFeatureSet(theSourceQuery);

            sourceSize = sourceFeatures.size64();

            status.setRangeOfValues(0, sourceSize);

            status.setCurValue(0);

            int result = this.process_minimize_memory(

                    sourceFeatures, 

                    createGroupByRevisionNumber(sourceFieldRevisionName), 

                    sourceFieldEfectiveDateName

            );

            if (result != ERR_OK) {

                status.abort();

                return result;

            }            

            status.message("Finished.");

            status.terminate();

            if (this.hasDuplicatedDeletes()) {

                return ERR_EXCEPTION;

            }

            return ERR_OK;

        } catch (Exception ex) {

            LOGGER.warn("Can't import history", ex);

            status.abort();

            FeatureStore.cancelEditingQuietly(getTargetStore());

            return ERR_EXCEPTION;

        } finally {

            DisposeUtils.disposeQuietly(sourceFeatures);

            disposeTargetStore();

        }

    }

}