Application: gvSIG desktop » gvSIG data provider for MS SQLServer

package org.gvsig.mssqlserver.dal.operations;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import org.gvsig.fmap.dal.DataTypes;

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

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

import org.gvsig.fmap.dal.feature.spi.FeatureProvider;

import org.gvsig.fmap.dal.feature.spi.FeatureReferenceProviderServices;

import org.gvsig.fmap.dal.feature.spi.FeatureStoreProvider;

import org.gvsig.fmap.dal.store.jdbc2.JDBCHelper;

import org.gvsig.fmap.dal.store.jdbc2.OperationsFactory.TableReference;

import org.gvsig.fmap.dal.store.jdbc2.spi.JDBCSQLBuilderBase;

import org.gvsig.fmap.dal.store.jdbc2.spi.operations.PerformChangesOperation;

import org.gvsig.mssqlserver.dal.MSSQLServerHelper;

import org.gvsig.mssqlserver.dal.MSSQLServerSQLBuilder.MSSQLServerCreateIndexBuilder;

@SuppressWarnings("UseSpecificCatch")

public class MSSQLServerPerformChangesOperation extends PerformChangesOperation {

    public MSSQLServerPerformChangesOperation(JDBCHelper helper) {

        this(helper, null, null, null, null, null, null);

    }

    public MSSQLServerPerformChangesOperation(JDBCHelper helper,

            TableReference table,

            FeatureType featureType,

            Iterator<FeatureReferenceProviderServices> deleteds,

            Iterator<FeatureProvider> inserteds,

            Iterator<FeatureProvider> updateds,

            Iterator<FeatureStoreProvider.FeatureTypeChanged> featureTypesChanged) {

        super(helper, table, featureType, deleteds, inserteds, updateds, featureTypesChanged);

    }

    protected MSSQLServerHelper getHelper() {

        return (MSSQLServerHelper) this.helper;

    }

    @Override

    protected List<String> buildCreateIndexSQL(

            TableReference table,

            FeatureType original,

            FeatureType target

        ) {

        for (FeatureAttributeDescriptor attrTarget : target) {

            boolean createIndex = false;

            if( attrTarget.isIndexed() ) {

                FeatureAttributeDescriptor attrOriginal = original.getAttributeDescriptor(attrTarget.getName());

                if ( attrOriginal == null) {

                    createIndex = true;

                } else {

                    if( attrOriginal.isIndexed() ) {

                        createIndex = false;

                    } else {

                        createIndex = true;

                    }

                }

            }

            if( createIndex ) {

                if( attrTarget.getType()==DataTypes.GEOMETRY ) {

                    try {

                        return this.getHelper().createOrUpdateSpatialIndexSql(

                                this.getConnection(),

                                table.getDatabase(),

                                table.getSchema(),

                                table.getTable(),

                                attrTarget.getName()

                        );

                    } catch (Exception ex) {

                        throw new RuntimeException("Can't create or update spatial index.", ex);

                    }

                } else {

                    JDBCSQLBuilderBase sqlbuilder = createSQLBuilder();

                    MSSQLServerCreateIndexBuilder create_index = (MSSQLServerCreateIndexBuilder) sqlbuilder.create_index();

                    create_index.if_not_exist();

                    create_index.name("idx_" + table + "_" + attrTarget.getName());

                    create_index.column(attrTarget.getName());

                    create_index.table()

                        .database(table.getDatabase())

                        .schema(table.getSchema())

                        .name(table.getTable());

                    return create_index.toStrings();

                }

            }

        }

        return new ArrayList<>();

    }    

}