Application: gvSIG desktop » gvSIG data provider for PostgreSQL

/* 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.postgresql;

import java.sql.SQLException;

import java.text.MessageFormat;

import javax.sql.DataSource;

import org.apache.commons.dbcp.BasicDataSource;

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

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

import org.gvsig.fmap.dal.resource.exception.AccessResourceException;

import org.gvsig.fmap.dal.store.jdbc.JDBCResource;

import org.gvsig.fmap.dal.store.jdbc.JDBCResourceParameters;

import org.gvsig.fmap.dal.store.jdbc.exception.JDBCDriverClassNotFoundException;

import org.gvsig.fmap.dal.store.jdbc.exception.JDBCException;

import org.gvsig.fmap.dal.store.jdbc.exception.JDBCSQLException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class PostgreSQLResource extends JDBCResource {

        final static private Logger logger = LoggerFactory

                        .getLogger(PostgreSQLResource.class);

        public final static String NAME = "PostgreSQLResource";

        public static final String DESCRIPTION = "PostgreSQL Connection";

        public PostgreSQLResource(PostgreSQLResourceParameters parameters)

                        throws InitializeException {

                super(parameters);

        }

        public String getName() throws AccessResourceException {

                PostgreSQLResourceParameters params = (PostgreSQLResourceParameters) this.getParameters();

                return MessageFormat.format("PostgreSQLResource({0},{1})",

                                new Object[] { params.getUrl(),params.getUser() });

        }

        public String toString() {

            try {

                return this.getName();

            } catch(Exception ex) {

                return super.toString();

            }

        }

        protected void connectToDB() throws DataException {

                if (this.dataSource != null) {

                        return;

                }

                JDBCResourceParameters jdbcParams = (JDBCResourceParameters) this

                                .getParameters();

                BasicDataSource dataSource = new BasicDataSource();

                dataSource.setDriverClassName(jdbcParams.getJDBCDriverClassName());

                dataSource.setUsername(jdbcParams.getUser());

                dataSource.setPassword(jdbcParams.getPassword());

                dataSource.setUrl(jdbcParams.getUrl());

                dataSource.setMaxWait(60L * 1000); // FIXME

                // FIXME Set Pool parameters:

                /*

                dataSource.setMaxActive(maxActive);

                dataSource.setMaxIdle(maxActive);

                dataSource.setMaxOpenPreparedStatements(maxActive);

                dataSource.setMaxWait(maxActive);

                dataSource.setInitialSize(initialSize);

                dataSource.setDefaultReadOnly(defaultReadOnly);

                dataSource.setDefaultTransactionIsolation(defaultTransactionIsolation);

                dataSource.setMinEvictableIdleTimeMillis(minEvictableIdleTimeMillis);

                dataSource.setMinIdle(minIdle);

                dataSource.setTestOnBorrow(testOnBorrow);

                dataSource.setTestOnReturn(testOnReturn);

                dataSource.setTestWhileIdle(testOnReturn);

                dataSource

                                .setTimeBetweenEvictionRunsMillis(timeBetweenEvictionRunsMillis);

                dataSource.setAccessToUnderlyingConnectionAllowed(allow);

                dataSource.setLoginTimeout(seconds);

                dataSource.setLogWriter(out);

                */

                this.dataSource = dataSource;

        }        

        protected void registerJDBCDriver() throws InitializeException {

                String className = ((JDBCResourceParameters) getParameters())

                                .getJDBCDriverClassName();

                if (className == null) {

                        return;

                }

                Class theClass = null;

                try {

                        theClass = Class.forName(className);

                } catch (Exception e){

                        throw new InitializeException(e);

                }

                if (theClass == null) {

                        try {

                                throw new JDBCDriverClassNotFoundException(this.getName(),

                                                className);

                        } catch (AccessResourceException e) {

                                throw new InitializeException(e);

                        }

                }

        }

        protected DataSource createDataSource() {

                PostgreSQLResourceParameters jdbcParams = (PostgreSQLResourceParameters) this

                                .getParameters();

                BasicDataSource dataSource = new BasicDataSource();

                dataSource.setDriverClassName(jdbcParams.getJDBCDriverClassName());

                dataSource.setUsername(jdbcParams.getUser());

                dataSource.setPassword(jdbcParams.getPassword());

                dataSource.setUrl(jdbcParams.getUrl());

                dataSource.setMaxWait(60L * 1000); // FIXME

                // FIXME Set Pool parameters:

                /*

                dataSource.setMaxActive(maxActive);

                dataSource.setMaxIdle(maxActive);

                dataSource.setMaxOpenPreparedStatements(maxActive);

                dataSource.setMaxWait(maxActive);

                dataSource.setInitialSize(initialSize);

                dataSource.setDefaultReadOnly(defaultReadOnly);

                dataSource.setDefaultTransactionIsolation(defaultTransactionIsolation);

                dataSource.setMinEvictableIdleTimeMillis(minEvictableIdleTimeMillis);

                dataSource.setMinIdle(minIdle);

                dataSource.setTestOnBorrow(testOnBorrow);

                dataSource.setTestOnReturn(testOnReturn);

                dataSource.setTestWhileIdle(testOnReturn);

                dataSource

                        .setTimeBetweenEvictionRunsMillis(timeBetweenEvictionRunsMillis);

                dataSource.setAccessToUnderlyingConnectionAllowed(allow);

                dataSource.setLoginTimeout(seconds);

                dataSource.setLogWriter(out);

                 */

                return dataSource;

        }

        public boolean isConnected() {

                if (dataSource == null) {

                        return false;

                }

                if (dataSource instanceof BasicDataSource) {

                        return ((BasicDataSource) dataSource).getNumActive() > 0

                                        || ((BasicDataSource) dataSource).getNumIdle() > 0;

                }

                return true;

        }        

        private void logPoolStatus(String src) {

                if (logger.isDebugEnabled() && dataSource instanceof BasicDataSource) {

                        BasicDataSource ds = (BasicDataSource) dataSource;

                        logger.debug(src + "  actives:" + ds.getNumActive() + "("

                                        + ds.getMaxActive() + ") idle:" + ds.getNumIdle() + "("

                                        + ds.getMaxIdle() + ")");

                }

        }

        private static class CanGetConnectionException extends JDBCSQLException {

            public CanGetConnectionException(String datasource, SQLException cause) {

                super("Can't get a connection to the data source (%(datasource))", cause, "_CanGetConnectionException", 0);

                setValue("datasource", datasource);

            }

        }

        protected synchronized Object getTheConnection() throws DataException {

                try {

                        Object conn = this.dataSource.getConnection();

                        logPoolStatus("getTheConnection");

                        return conn;

                } catch (SQLException e) {

                        throw new CanGetConnectionException(this.toString(),e);

                }

        }        

}