com.landawn.abacus.dataSource.C3P0ConnectionManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of abacus-android Show documentation
A general and simple library for Android
There is a newer version: 1.10.1
/*
 * Copyright (c) 2015, Haiyang Li. All rights reserved.
 */

package com.landawn.abacus.dataSource;

import static com.landawn.abacus.dataSource.DataSourceConfiguration.DRIVER;
import static com.landawn.abacus.dataSource.DataSourceConfiguration.INITIAL_SIZE;
import static com.landawn.abacus.dataSource.DataSourceConfiguration.JNDI_NAME;
import static com.landawn.abacus.dataSource.DataSourceConfiguration.MAX_ACTIVE;
import static com.landawn.abacus.dataSource.DataSourceConfiguration.MAX_IDLE_TIME;
import static com.landawn.abacus.dataSource.DataSourceConfiguration.MAX_OPEN_PREPARED_STATEMENTS_PER_CONNECTION;
import static com.landawn.abacus.dataSource.DataSourceConfiguration.MAX_WAIT_TIME;
import static com.landawn.abacus.dataSource.DataSourceConfiguration.MIN_IDLE;
import static com.landawn.abacus.dataSource.DataSourceConfiguration.PASSWORD;
import static com.landawn.abacus.dataSource.DataSourceConfiguration.TEST_ON_BORROW;
import static com.landawn.abacus.dataSource.DataSourceConfiguration.TEST_ON_RETURN;
import static com.landawn.abacus.dataSource.DataSourceConfiguration.URL;
import static com.landawn.abacus.dataSource.DataSourceConfiguration.USER;
import static com.landawn.abacus.dataSource.DataSourceConfiguration.VALIDATION_QUERY;

import java.beans.PropertyVetoException;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Map;

import javax.sql.DataSource;

import com.landawn.abacus.exception.AbacusException;
import com.landawn.abacus.exception.UncheckedSQLException;
import com.landawn.abacus.logging.Logger;
import com.landawn.abacus.logging.LoggerFactory;
import com.landawn.abacus.util.N;

/**
 * 
 * @since 0.8
 * 
 * @author Haiyang Li
 */
class C3P0ConnectionManager extends AbstractConnectionManager {
    private static final Logger logger = LoggerFactory.getLogger(C3P0ConnectionManager.class);

    private final DataSource ds;
    private final com.mchange.v2.c3p0.ComboPooledDataSource cpds;

    public C3P0ConnectionManager(Map props) {
        super(props);

        if (properties.containsKey(JNDI_NAME)) {
            try {
                ds = com.mchange.v2.c3p0.DataSources.pooledDataSource(createJNDIDataSource(properties));
            } catch (SQLException e) {
                throw new UncheckedSQLException(e);
            }

            cpds = null;
        } else {
            cpds = new com.mchange.v2.c3p0.ComboPooledDataSource();
            cpds.setProperties(connectionProperties);

            try {
                cpds.setDriverClass(properties.get(DRIVER));
            } catch (PropertyVetoException e) {
                throw N.toRuntimeException(e);
            }

            cpds.setJdbcUrl(properties.get(URL));
            cpds.setUser(properties.get(USER));
            cpds.setPassword(properties.get(PASSWORD));

            cpds.setInitialPoolSize(Integer.valueOf(properties.get(INITIAL_SIZE)));
            cpds.setMinPoolSize(Integer.valueOf(properties.get(MIN_IDLE)));
            cpds.setMaxPoolSize(Integer.valueOf(properties.get(MAX_ACTIVE)));
            cpds.setMaxStatementsPerConnection(Integer.valueOf(properties.get(MAX_OPEN_PREPARED_STATEMENTS_PER_CONNECTION)));
            cpds.setMaxIdleTime(Integer.valueOf(properties.get(MAX_IDLE_TIME)));
            cpds.setCheckoutTimeout(Integer.valueOf(properties.get(MAX_WAIT_TIME)));
            cpds.setPreferredTestQuery(properties.get(VALIDATION_QUERY));
            cpds.setTestConnectionOnCheckout(Boolean.valueOf(properties.get(TEST_ON_BORROW)));
            cpds.setTestConnectionOnCheckin(Boolean.valueOf(properties.get(TEST_ON_RETURN)));

            ds = cpds;
        }
    }

    @Override
    public int getMaxActive() {
        if (cpds == null) {
            throw new UnsupportedOperationException();
        }

        return cpds.getMaxPoolSize();
    }

    @Override
    public int getNumActive() {
        if (cpds == null) {
            throw new UnsupportedOperationException();
        }

        try {
            return cpds.getNumConnections();
        } catch (SQLException e) {
            String msg = AbacusException.getErrorMsg(e);
            logger.warn(msg);
            throw new UncheckedSQLException(msg, e);
        }
    }

    @Override
    public Connection getConnection() {
        try {
            return ds.getConnection();
        } catch (SQLException e) {
            String msg = AbacusException.getErrorMsg(e);
            logger.warn(msg);
            throw new UncheckedSQLException(msg, e);
        }
    }

    @Override
    public void closeConnection(Connection conn) {
        if (conn != null) {
            try {
                conn.close();
            } catch (SQLException e) {
                String msg = AbacusException.getErrorMsg(e);
                logger.warn(msg);
                throw new UncheckedSQLException(msg, e);
            }
        }
    }

    @Override
    public void detroyConnection(Connection conn) {
        closeConnection(conn);
    }

    @Override
    public void close() {
        if (cpds != null) {
            cpds.close();
        }
    }
}