| gvSIG desktop 1

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

*

* Copyright (C) 2004 IVER T.I. and Generalitat Valenciana.

*

* 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

*

* For more information, contact:

*

*  Generalitat Valenciana

*   Conselleria d'Infraestructures i Transport

*   Av. Blasco Ib??ez, 50

*   46010 VALENCIA

*   SPAIN

*

*      +34 963862235

*   gvsig@gva.es

*      www.gvsig.gva.es

*

*    or

*

*   IVER T.I. S.A

*   Salamanca 50

*   46005 Valencia

*   Spain

*

*   +34 963163400

*   dac@iver.es

*/

package es.gva.cit.gvsig.catalogClient.loaders;

import java.sql.Connection;

import java.sql.DriverManager;

import java.sql.SQLException;

import java.util.StringTokenizer;

import java.util.TreeMap;

import com.hardcode.driverManager.DriverLoadException;

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

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

import com.iver.cit.gvsig.fmap.drivers.jdbc.postgis.PostGisDriver;

import com.iver.cit.gvsig.fmap.layers.FLayer;

import com.iver.cit.gvsig.fmap.layers.LayerFactory;

/**

 * This class is used to load a POSTGIS resource in gvSIG

 * 

 * @author Jorge Piera Llodra (piera_jor@gva.es)

 */

public class PostgisLoader extends AbstractLoader{

    /**

     * This function adds a Postgis Layer to gvSIG 

     * @param jdbcUrl

            * JDBC url connection

            * @param table

            * Table to load

            */

           public boolean load(String jdbcUrl, String table) {

               FLayer flayer;

        try {

            flayer = createPostgisLayer(jdbcUrl, table);

            if (flayer == null){

                       return false;

                }else{

                        addLayerToView(flayer);

                        return true;

                }

        } catch (SQLException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        } catch (DriverLoadException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        } catch (ClassNotFoundException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

        return false;

           }

    /**

            * It returns a Postgis Layer

            * @param jdbcUrl

            * JDBC url connection

            * @param table

            * Table to load

            * @return

     * @throws SQLException

     * @throws DriverLoadException

     * @throws ClassNotFoundException

            */

    private FLayer createPostgisLayer(String jdbcUrl, String table) throws SQLException, DriverLoadException, ClassNotFoundException  {

        //jdbc:postgresql://sercartlin/carto_300k?user=gis&password=gis     

        //NAME=Comunicaciones&TABLENAME=comunic_lin_300k&ID=6&FIELDS=entity,layer,codigo,tipo,gid&GEOMFIELD=the_geom

        StringTokenizer sti = new StringTokenizer(jdbcUrl,"?");

        String dbURL = sti.nextToken();

        String p = sti.nextToken();

        TreeMap credentials = separateParams(p);

                String user = (String) credentials.get((String) "USER");

                String pwd = (String) credentials.get((String) "PASSWORD");

                TreeMap params = separateParams(table);

                String layerName = (String) params.get((String) "NAME");

                int fidField = Integer.valueOf((String) params.get((String) "ID")).intValue();

                String fields = getFieldsClause((String) params.get((String) "FIELDS"),

                        (String) params.get((String) "GEOMFIELD"));

                String tableName = (String) params.get((String) "TABLENAME");

                String whereClause = "";

                 Connection conn = DriverManager.getConnection(dbURL, user, pwd);

         conn.setAutoCommit(false);

         PostGisDriver pgd = new PostGisDriver();

         pgd.setData(conn, tableName, fields, whereClause, fidField);

         return LayerFactory.createDBLayer(pgd, layerName, null);

   }

    private String getFieldsClause(String fields,String geomField){

        return "ASBINARY(" + geomField + ", 'XDR')," + fields;

    }

    private TreeMap separateParams(String pairValues){

        TreeMap map = new TreeMap(); 

                String[] params = pairValues.split("&");

                for (int i = 0; i < params.length; i++) {

                        String[] nameValue = params[i].split("=");

                        map.put(nameValue[0].toUpperCase(), nameValue[1]);

                }

                return map;

    }

}