Application: gvSIG desktop » gvSIG data provider for MS SQLServer

package org.gvsig.mssqlserver.dal.operations;

import java.sql.Connection;

import java.sql.ResultSet;

import java.sql.ResultSetMetaData;

import java.sql.Statement;

import java.util.List;

import org.cresques.cts.IProjection;

import org.gvsig.expressionevaluator.ExpressionBuilder;

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

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

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

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

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

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.FetchFeatureTypeOperation;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.type.GeometryType;

import org.gvsig.mssqlserver.dal.MSSQLServerHelper;

public class MSSQLServerFetchFeatureTypeOperation extends FetchFeatureTypeOperation {

    public MSSQLServerFetchFeatureTypeOperation(

            JDBCHelper helper

    ) {

        super(helper);

    }

    public MSSQLServerFetchFeatureTypeOperation(

            JDBCHelper helper,

            EditableFeatureType featureType,

            TableReference table,

            List<String> primaryKeys,

            String defaultGeometryColumn,

            IProjection crs

    ) {

        super(helper, featureType, table, primaryKeys, defaultGeometryColumn, crs);

    }

    @Override

    public void fetch(EditableFeatureType featureType, Connection conn, TableReference table, List<String> pks, String defaultGeometryColumn, IProjection crs) throws DataException {

        super.fetch(featureType, conn, table, pks, defaultGeometryColumn, crs);

        ((MSSQLServerHelper) this.helper).setLastUsedFeatureType(featureType);

    }

    @Override

    protected void fetchGeometryTypeAndSRS(

            EditableFeatureAttributeDescriptor attr,

            ResultSetMetaData rsMetadata,

            int colIndex

    ) {

        try {

            GeometryType geomType = GeometryLocator.getGeometryManager().getGeometryType(

                    Geometry.TYPES.GEOMETRY,

                    Geometry.SUBTYPES.GEOM2D

            );

            attr.setGeometryType(geomType);

            String typeName = rsMetadata.getColumnTypeName(colIndex);

            attr.setAdditionalInfo("SQLServer_type_name", typeName);

            attr.setSRS(getProjectionOfColumn(attr.getName()));

        } catch (Exception ex) {

            LOGGER.warn("Can't get default geometry type.", ex);

        }

    }

    private IProjection getProjectionOfColumn(String columnName) {

        JDBCSQLBuilderBase sqlbuilder = this.createSQLBuilder();

        ExpressionBuilder expbuilder = sqlbuilder.expression();

        sqlbuilder.select().limit(1);

        sqlbuilder.select().column().value(

                expbuilder.ST_SRID(expbuilder.column(columnName))

        );

        sqlbuilder.select().where().set(

                expbuilder.and(

                        expbuilder.not_is_null(expbuilder.column(columnName)),

                        expbuilder.not_is_null(expbuilder.ST_SRID(expbuilder.column(columnName)))

                )

        );

        sqlbuilder.select().from().table()

                .database(this.getTable().getDatabase())

                .schema(this.getTable().getSchema())

                .name(this.getTable().getTable());

        Statement st = null;

        ResultSet rs = null;

        try {

            st = this.getConnection().createStatement();

            rs = JDBCUtils.executeQuery(st, sqlbuilder.toString());

            if (rs.next()) {

                int code = rs.getInt(1);

                try {

                    IProjection proj = this.helper.getSRSSolver().getProjection(conn, code);

                    return proj;

                } catch (Exception ex) {

                    LOGGER.warn("Can't create projection from code '" + code + "'.", ex);

                }

            }

        } catch (Exception ex) {

            // Do nothing, return null

        } finally {

            JDBCUtils.closeQuietly(rs);

            JDBCUtils.closeQuietly(st);

        }

        return null;

    }

}