Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 gvSIG Association.

 *

 * 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 3

 * 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.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

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

import java.sql.Connection;

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

import org.gvsig.fmap.dal.resource.db.AbstractDBResourceNoBlocker;

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

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

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class JDBCResource extends AbstractDBResourceNoBlocker {

        final static private Logger logger = LoggerFactory

                        .getLogger(JDBCResource.class);

        public final static String NAME = "JDBCResource";

        public static final String DESCRIPTION = "JDBC Connection";

        protected DataSource dataSource = null;

        public JDBCResource(JDBCResourceParameters parameters)

                        throws InitializeException {

                super(parameters);

                registerJDBCDriver();

        }

        public String getName() throws AccessResourceException {

                JDBCResourceParameters params = (JDBCResourceParameters) this

                                .getParameters();

                return MessageFormat.format("JDBCResource({0},{1},{2},{3},{4})",

                                new Object[] {

                                params.getJDBCDriverClassName(),

                                                params.getHost(), params.getPort(),

                                params.getUser(), params.getDBName() });

        }

        @SuppressWarnings("unchecked")

        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);

                        }

                }

        }

        public Connection getJDBCConnection() throws AccessResourceException {

                return (Connection) get();

        }

        public void closeConnection(Connection connection) {

            try {

                logger.debug("before-close connection"+ this.getStatusInformation());

                connection.close();

                logger.debug("after close connection "+ this.getStatusInformation());

            } catch (Exception ex) {

                logger.warn("Problems closing connection.",ex);

            }            

        }

        private void debugPoolStatus(String src) {

                if (logger.isDebugEnabled() ) {

                        logger.debug(src + "  " + this.getStatusInformation());

                }

        }

        public String getStatusInformation() {

                if ( !(dataSource instanceof BasicDataSource) ) {

                    return "Connected: " + this.isConnected();

                }

                BasicDataSource ds = (BasicDataSource) dataSource;

                String s = "Connected: " + this.isConnected() + ", " +

                        "actives: "+ ds.getNumActive() + "/"+ ds.getMaxActive() + ", " + 

                        "idle: " + ds.getNumIdle() + "/"+ ds.getMaxIdle() ;

                return s;

        }

        protected synchronized Object getTheConnection() throws DataException {

                try {

                        Object conn = this.dataSource.getConnection();

                        debugPoolStatus("getTheConnection");

                        return conn;

                } catch (SQLException e) {

                        throw new JDBCSQLException(e);

                }

        }

        public boolean isThis(ResourceParameters parameters)

                        throws ResourceException {

                if (!super.isThis(parameters)) {

                        return false;

                }

                String dbName = ((JDBCResourceParameters) parameters).getDBName();

                String myDbName = ((JDBCResourceParameters) getParameters())

                                .getDBName();

                if (dbName != myDbName) {

                        if (!(dbName != null && dbName.equals(myDbName))) {

                                return false;

                        }

                }

                String driver = ((JDBCResourceParameters) parameters)

                                .getJDBCDriverClassName();

                String myDriver = ((JDBCResourceParameters) getParameters())

                                .getJDBCDriverClassName();

                if (driver != myDriver) {

                        if (!(driver != null && driver.equals(myDriver))) {

                                return false;

                        }

                }

                return true;

        }

        public boolean isConnected() {

                if (dataSource == null) {

                        return false;

                }

                if (dataSource instanceof BasicDataSource) {

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

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

                }

                return true;

        }

        protected DataSource createDataSource() {

                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);

                 */

                return dataSource;

        }

        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;

        }

        public void beginUse() {

            this.executeBegins();

        }

        public void endUse() {

           this.executeEnds();

        }

}