com.panet.imeta.core.database.ConnectionPoolUtil.java Source code

Introduction

Here is the source code for com.panet.imeta.core.database.ConnectionPoolUtil.java

Source

/*
 * Copyright (c) 2007 Pentaho Corporation.  All rights reserved. 
 * This software was developed by Pentaho Corporation and is provided under the terms 
 * of the GNU Lesser General Public License, Version 2.1. You may not use 
 * this file except in compliance with the license. If you need a copy of the license, 
 * please go to http://www.gnu.org/licenses/lgpl-2.1.txt. The Original Code is Pentaho 
 * Data Integration.  The Initial Developer is Pentaho Corporation.
 *
 * Software distributed under the GNU Lesser Public License is distributed on an "AS IS" 
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or  implied. Please refer to 
 * the license for the specific language governing your rights and limitations.
*/
package com.panet.imeta.core.database;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Iterator;
import java.util.Properties;

import org.apache.commons.dbcp.ConnectionFactory;
import org.apache.commons.dbcp.DriverManagerConnectionFactory;
import org.apache.commons.dbcp.PoolableConnectionFactory;
import org.apache.commons.dbcp.PoolingDriver;
import org.apache.commons.pool.impl.GenericObjectPool;

import com.panet.imeta.core.Const;
import com.panet.imeta.core.exception.KettleDatabaseException;
import com.panet.imeta.core.logging.LogWriter;

public class ConnectionPoolUtil {
    private static PoolingDriver pd = new PoolingDriver();

    public static final int defaultInitialNrOfConnections = 5;
    public static final int defaultMaximumNrOfConnections = 10;

    private static boolean isPoolRegistered(DatabaseMeta dbMeta, String partitionId)
            throws KettleDatabaseException {
        try {
            String name = dbMeta.getName() + Const.NVL(partitionId, "");
            ;
            return Const.indexOfString(name, pd.getPoolNames()) >= 0;
        } catch (SQLException e) {
            throw new KettleDatabaseException(
                    Messages.getString("Database.UnableToCheckIfConnectionPoolExists.Exception"), e);
        }
    }

    private static void createPool(DatabaseMeta databaseMeta, String partitionId, int initialSize, int maximumSize)
            throws KettleDatabaseException {
        LogWriter.getInstance().logBasic(databaseMeta.toString(),
                Messages.getString("Database.CreatingConnectionPool", databaseMeta.getName()));
        GenericObjectPool gpool = new GenericObjectPool();

        gpool.setMaxIdle(-1);
        gpool.setWhenExhaustedAction(GenericObjectPool.WHEN_EXHAUSTED_GROW);
        gpool.setMaxActive(maximumSize);

        String clazz = databaseMeta.getDriverClass();
        try {
            Class.forName(clazz).newInstance();
        } catch (Exception e) {
            throw new KettleDatabaseException(Messages.getString(
                    "Database.UnableToLoadConnectionPoolDriver.Exception", databaseMeta.getName(), clazz), e);
        }

        String url;
        String userName;
        String password;

        try {
            url = databaseMeta.environmentSubstitute(databaseMeta.getURL(partitionId));
            userName = databaseMeta.environmentSubstitute(databaseMeta.getUsername());
            password = databaseMeta.environmentSubstitute(databaseMeta.getPassword());
        } catch (RuntimeException e) {
            url = databaseMeta.getURL(partitionId);
            userName = databaseMeta.getUsername();
            password = databaseMeta.getPassword();
        }

        // Get the list of pool properties
        Properties originalProperties = databaseMeta.getConnectionPoolingProperties();
        //Add user/pass
        originalProperties.setProperty("user", Const.NVL(userName, ""));
        originalProperties.setProperty("password", Const.NVL(password, ""));

        // Now, replace the environment variables in there...
        Properties properties = new Properties();
        Iterator<Object> iterator = originalProperties.keySet().iterator();
        while (iterator.hasNext()) {
            String key = (String) iterator.next();
            String value = originalProperties.getProperty(key);
            properties.put(key, databaseMeta.environmentSubstitute(value));
        }

        // Create factory using these properties.
        //
        ConnectionFactory cf = new DriverManagerConnectionFactory(url, properties);

        new PoolableConnectionFactory(cf, gpool, null, null, false, false);

        for (int i = 0; i < initialSize; i++) {
            try {
                gpool.addObject();
            } catch (Exception e) {
                throw new KettleDatabaseException(
                        Messages.getString("Database.UnableToPreLoadConnectionToConnectionPool.Exception"), e);
            }
        }

        pd.registerPool(databaseMeta.getName(), gpool);

        LogWriter.getInstance().logBasic(databaseMeta.toString(),
                Messages.getString("Database.CreatedConnectionPool", databaseMeta.getName()));
    }

    public static Connection getConnection(DatabaseMeta dbMeta, String partitionId) throws Exception {
        return getConnection(dbMeta, partitionId, dbMeta.getInitialPoolSize(), dbMeta.getMaximumPoolSize());
    }

    public static Connection getConnection(DatabaseMeta dbMeta, String partitionId, int initialSize,
            int maximumSize) throws Exception {
        if (!isPoolRegistered(dbMeta, partitionId)) {
            createPool(dbMeta, partitionId, initialSize, maximumSize);
        }

        return DriverManager.getConnection("jdbc:apache:commons:dbcp:" + dbMeta.getName());

    }

}