Application: gvSIG desktop

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

*

* This program is free software; you can redistribute it and/or

* modify it under the terms of the GNU General Public License

* as published by the Free Software Foundation; either version 2

* of the License, or (at your option) any later version.

*

* This program is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

* GNU General Public License for more details.

*

* You should have received a copy of the GNU General Public License

* along with this program; if not, write to the Free Software

* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

* MA  02110-1301, USA.

*

*/

/*

* AUTHORS (In addition to CIT):

* 2009 IVER T.I   {{Task}}

*/

package org.gvsig.fmap.dal.store.postgesql;

import java.sql.Connection;

import java.sql.ResultSet;

import java.sql.ResultSetMetaData;

import java.sql.Statement;

import java.util.Arrays;

import java.util.List;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

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

public class PostgreSQLUtils {

        private static PostgreSQLUtils instance = null;

        static PostgreSQLUtils getInstance() {

                if (instance == null) {

                        instance = new PostgreSQLUtils();

                }

                return instance;

        }

        PostgreSQLResourceParameters getResourceParameters(

                        PostgreSQLStoreParameters stParams) {

                PostgreSQLResourceParameters result = new PostgreSQLResourceParameters(

                                stParams.getHost(), 

                                stParams.getPort(),

                                stParams.getDBName(), 

                                stParams.getUser(),

                                stParams.getPassword(), 

                                stParams.getJDBCDriverClassName(), 

                                stParams.getUseSSL());

                return result;

        }

        void loadFeatureType(Connection conn,

                        EditableFeatureType featureType,

                        PostgreSQLStoreParameters params, String sql)

                        throws DataException {

                loadFeatureType(conn, featureType, sql, params.getPkFields(), params

                                .getDefaultGeometry());

        }

        private void loadFeatureType(Connection conn,

                        EditableFeatureType featureType, String sql, String[] pks,

                        String defGeomName)

                        throws DataException {

                try {

                        Statement stAux = conn.createStatement();

                        stAux.setFetchSize(1);

                        ResultSet rs = stAux.executeQuery(sql);

                        ResultSetMetaData rsMetadata = rs.getMetaData();

                        List pksList = Arrays.asList(pks);

                        int i;

                        EditableFeatureAttributeDescriptor attr;

                        for (i = 1; i <= rsMetadata.getColumnCount(); i++) {

                                attr = getAttributeFromJDBC(featureType, conn, rsMetadata, i);

                                if (pksList.contains(attr.getName())) {

                                        attr.setIsPrimaryKey(true);

                                }

                                if (attr.getName().equals(defGeomName)

                                                && attr.getDataType() == DataTypes.GEOMETRY) {

                                        featureType.setDefaultGeometryAttributeName(defGeomName);

                                }

                        }

                        rs.close();

                        stAux.close();

                } catch (java.sql.SQLException e) {

                        throw new ReadException("", e); // FIXME exception

                } finally {

                }

        }

        private EditableFeatureAttributeDescriptor getAttributeFromJDBC(

                        EditableFeatureType fType, Connection conn,

                        ResultSetMetaData rsMetadata, int colIndex) throws ReadException {

                EditableFeatureAttributeDescriptor column;

                try {

                        switch (rsMetadata.getColumnType(colIndex)) {

                        case java.sql.Types.INTEGER:

                                column = fType.add(rsMetadata.getColumnName(colIndex),

                                                DataTypes.INT);

                                break;

                        case java.sql.Types.BIGINT:

                                column = fType.add(rsMetadata.getColumnName(colIndex),

                                                DataTypes.LONG);

                                break;

                        case java.sql.Types.REAL:

                                column = fType.add(rsMetadata.getColumnName(colIndex),

                                                DataTypes.DOUBLE);

                                break;

                        case java.sql.Types.DOUBLE:

                                column = fType.add(rsMetadata.getColumnName(colIndex),

                                                DataTypes.DOUBLE);

                                break;

                        case java.sql.Types.CHAR:

                                column = fType.add(rsMetadata.getColumnName(colIndex),

                                                DataTypes.STRING);

                                break;

                        case java.sql.Types.VARCHAR:

                                column = fType.add(rsMetadata.getColumnName(colIndex),

                                                DataTypes.STRING);

                                break;

                        case java.sql.Types.FLOAT:

                                column = fType.add(rsMetadata.getColumnName(colIndex),

                                                DataTypes.FLOAT);

                                break;

                        case java.sql.Types.DECIMAL:

                                column = fType.add(rsMetadata.getColumnName(colIndex),

                                                DataTypes.FLOAT);

                                break;

                        case java.sql.Types.DATE:

                                column = fType.add(rsMetadata.getColumnName(colIndex),

                                                DataTypes.DATE);

                                break;

                        case java.sql.Types.TIME:

                                column = fType.add(rsMetadata.getColumnName(colIndex),

                                                DataTypes.TIME);

                                break;

                        case java.sql.Types.TIMESTAMP:

                                column = fType.add(rsMetadata.getColumnName(colIndex),

                                                DataTypes.TIMESTAMP);

                                break;

                        case java.sql.Types.BOOLEAN:

                                column = fType.add(rsMetadata.getColumnName(colIndex),

                                                DataTypes.BOOLEAN);

                                break;

                        case java.sql.Types.OTHER:

                                if (rsMetadata.getColumnTypeName(colIndex).equalsIgnoreCase(

                                                "geometry")) {

                                        column = fType.add(rsMetadata.getColumnName(colIndex),

                                                        DataTypes.GEOMETRY);

                                        break;

                                }

                                // No hacemos break para que se quede en default

                        default:

                                column = fType.add(rsMetadata.getColumnName(colIndex),

                                                DataTypes.OBJECT);

                                break;

                        }

                        // column.setCaseSensitive(rsMetadata.isCaseSensitive(colIndex));

                        // column.setSqlType(rsMetadata.getColumnType(colIndex));

                        column

                                        .setAllowNull(rsMetadata.isNullable(colIndex) == ResultSetMetaData.columnNullable);

                        column.setIsAutomatic(rsMetadata.isAutoIncrement(colIndex));

                        column.setIsReadOnly(rsMetadata.isReadOnly(colIndex));

                        // column.setWritable(rsMetadata.isWritable(colIndex));

                        // column.setClassName(rsMetadata.getColumnClassName(colIndex));

                        // column.setCatalogName(rsMetadata.getCatalogName(colIndex));

                        // column.setDefinitelyWritable(rsMetadata

                        //                 .isDefinitelyWritable(colIndex));

                        // column.setLabel(rsMetadata.getColumnLabel(colIndex));

                        // column.setSchemaName(rsMetadata.getSchemaName(colIndex));

                        // column.setTableName(rsMetadata.getTableName(colIndex));

                        // column.setCatalogName(rsMetadata.getCatalogName(colIndex));

                        // column.setSqlTypeName();

                        // column.setSearchable(rsMetadata.isSearchable(colIndex));

                        // column.setSigned(rsMetadata.isSigned(colIndex));

                        // column.setCurrency(rsMetadata.isCurrency(colIndex));

                        column.setPrecision(rsMetadata.getPrecision(colIndex));

                        column.setSize(rsMetadata.getColumnDisplaySize(colIndex));

                } catch (java.sql.SQLException e) {

                        throw new ReadException("", e); // FIXME

                }

                return column;

        }

        public String getJdbcUrl(String host, Integer port, String db) {

                String url;

                String sport = "";

                if (port != null) {

                        sport = ":" + port;

                }

                url = "jdbc:postgresql://" + host + sport + "/" + db;

                return url;

        }

}