Application: gvSIG desktop

package org.gvsig.data.datastores.vectorial.db.jdbc.postgresqlbin;

import java.math.BigDecimal;

import java.nio.ByteBuffer;

import java.sql.Connection;

import java.sql.Date;

import java.sql.DriverManager;

import java.sql.ResultSet;

import java.sql.ResultSetMetaData;

import java.sql.Statement;

import java.sql.Timestamp;

import java.sql.Types;

import java.util.Iterator;

import org.gvsig.data.datastores.vectorial.db.DBFeatureType;

import org.gvsig.data.datastores.vectorial.db.jdbc.JDBCAttributeDescriptor;

import org.gvsig.data.datastores.vectorial.db.jdbc.JDBCFeature;

import org.gvsig.data.datastores.vectorial.db.jdbc.JDBCStore;

import org.gvsig.data.datastores.vectorial.db.jdbc.exception.JDBCDriverNotFoundException;

import org.gvsig.data.datastores.vectorial.db.jdbc.exception.SQLException;

import org.gvsig.data.datastores.vectorial.db.jdbc.postgresql.PostgresqlFeature;

import org.gvsig.data.datastores.vectorial.db.jdbc.postgresql.PostgresqlStore;

import org.gvsig.data.datastores.vectorial.db.jdbc.postgresql.PostgresqlStoreParameters;

import org.gvsig.data.datastores.vectorial.db.jdbc.postgresql.PostgresqlStoreUtils;

import org.gvsig.data.exception.InitializeException;

import org.gvsig.data.exception.ReadException;

import org.gvsig.data.vectorial.DefaultAttributeDescriptor;

import org.gvsig.data.vectorial.IFeature;

import org.gvsig.data.vectorial.IFeatureAttributeDescriptor;

import com.iver.cit.gvsig.fmap.core.IGeometry;

import com.iver.cit.gvsig.fmap.drivers.WKBParser2;

import com.iver.cit.gvsig.fmap.drivers.XTypes;

import com.vividsolutions.jts.io.WKBWriter;

public class PostgresqlBinStoreUtils {

        private static WKBParser2 wkbParser = new WKBParser2();

        private static WKBWriter wkbWriter = new WKBWriter();

        private static int cursorCount=0;

        private static String baseCursorName=null;

        static Connection getConnection(String dbUrl, String dbUser, String dbPass) throws InitializeException {

                //TODO: Aqu? habria que implementar la llamada

                //      al Resource Manager para comprobar si ya hay

                //                una connexion a la BD

                String connID = getConnectionResourceID(dbUrl, dbUser);

                Connection conn = null;

//                IResource res = ResourceManager.getResourceManager().getResource(connID);

                try {

                        Class.forName("org.postgresql.Driver");

                } catch (ClassNotFoundException e) {

                        throw new JDBCDriverNotFoundException("org.postgresql.Driver",e);

                }

                try {

                        conn = DriverManager.getConnection(dbUrl, dbUser, dbPass);

                        conn.setAutoCommit(false);

                } catch (java.sql.SQLException e1) {

                        throw new InitializeException(PostgresqlStore.DATASTORE_NAME,e1);

                }

                //TODO: Registrar en el Resource manager

                // ResourceManager.getResourceManager().addResource(res);

                return conn;

        }

        static String getConnectionResourceID(String dbUrl,String dbUser){

                return PostgresqlStore.CONNECTION_STRING+";"+dbUrl+";"+dbUser;

        }

        static String addLimitsToSQL(String aSql,int fetchSize,int page){

                return aSql+ " limit " + fetchSize + " offset " + (fetchSize*page);

        }

        static String createCursorName() {

                if (baseCursorName == null){

                        baseCursorName = "gv_"+ System.currentTimeMillis();

                }

                cursorCount++;

                return baseCursorName+"_"+cursorCount;

        }

        static DBFeatureType getFeatureType(Connection connection, PostgresqlStoreParameters params) throws ReadException{

                DBFeatureType featureType = new DBFeatureType();

                loadFieldsToFeatureType(connection, params, featureType);

                featureType.setFieldsId(params.getFieldsId());

                if (params.getDefaultGeometryField() != null && params.getDefaultGeometryField() != ""){

                        if (featureType.getFieldIndex(params.getDefaultGeometryField())< 0){

                                // FIXME: crear una nueva excepcion??

                                throw new InitializeException(

                                                PostgresqlStore.DATASTORE_NAME,

                                                new Exception("Geometry Field '"+ params.getDefaultGeometryField() +"' not Found"));

                        }

                        featureType.setDefaultGeometry(params.getDefaultGeometryField());

                }

                return featureType;

        }

        private static void loadFieldsToFeatureType(Connection conn,PostgresqlStoreParameters params,DBFeatureType featureType) throws ReadException{

                String sql="";

                String columns=params.getFieldsString();

                if (params.getSqlSoure() != null){

                        sql = params.getSqlSoure();

                } else {

                        sql = "Select "+columns+" from " + params.tableID() + " limit 1;";

                }

                try {

                        Statement stAux = conn.createStatement();

                        ResultSet rs = stAux.executeQuery(sql);

                        ResultSetMetaData rsMetadata = rs.getMetaData();

                        int i;

                        featureType.setTableID(params.tableID());

                        DefaultAttributeDescriptor attr;

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

                                attr = getAttributeFromJDBC(conn,rsMetadata,i);

                                featureType.add(attr);

                                attr.setOrdinal(i-1);

                        }

                        rs.close();

                } catch (java.sql.SQLException e) {

                        // TODO Auto-generated catch block

                        throw new SQLException(sql,"getFeatureType",e);

                }

        }

        private static DefaultAttributeDescriptor getAttributeFromJDBC(Connection conn,ResultSetMetaData rsMetadata,int colIndex) throws SQLException{

                JDBCAttributeDescriptor column= new JDBCAttributeDescriptor();

                try {

                        column.setName(rsMetadata.getColumnName(colIndex));

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

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

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

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

                        switch (rsMetadata.getColumnType(colIndex)) {

                        case java.sql.Types.INTEGER:

                                column.setType(IFeatureAttributeDescriptor.TYPE_INT);

                                break;

                        case java.sql.Types.BIGINT:

                                column.setType(IFeatureAttributeDescriptor.TYPE_LONG);

                                break;

                        case java.sql.Types.REAL:

                                column.setType(IFeatureAttributeDescriptor.TYPE_DOUBLE);

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

                                break;

                        case java.sql.Types.DOUBLE:

                                column.setType(IFeatureAttributeDescriptor.TYPE_DOUBLE);

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

                                break;

                        case java.sql.Types.CHAR:

                                column.setType(IFeatureAttributeDescriptor.TYPE_STRING);

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

                                break;

                        case java.sql.Types.VARCHAR:

                                column.setType(IFeatureAttributeDescriptor.TYPE_STRING);

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

                                break;

                        case java.sql.Types.FLOAT:

                                column.setType(IFeatureAttributeDescriptor.TYPE_FLOAT);

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

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

                                break;

                        case java.sql.Types.DECIMAL:

                                column.setType(IFeatureAttributeDescriptor.TYPE_FLOAT);

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

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

                                break;

                        case java.sql.Types.DATE:

                                column.setType(IFeatureAttributeDescriptor.TYPE_DATE);

                                break;

                        case java.sql.Types.BOOLEAN:

                                column.setType(IFeatureAttributeDescriptor.TYPE_BOOLEAN);

                                break;

                        default:

                                column.setType(IFeatureAttributeDescriptor.TYPE_OBJECT);

                                break;

                        }

                } catch (java.sql.SQLException e){

                        throw new SQLException("","load attribute definition",e);

                }

                return column;

        }

        static String getJDBCUrl(String host, String db, String port) {

                String url;

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

                return url;

        }

}