org.killbill.commons.jdbi.guice.DataSourceProvider Maven / Gradle / Ivy
/*
* Copyright 2010-2014 Ning, Inc.
* Copyright 2014 Groupon, Inc
* Copyright 2014 The Billing Project, LLC
*
* Ning licenses this file to you under the Apache License, version 2.0
* (the "License"); you may not use this file except in compliance with the
* License. You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*/
package org.killbill.commons.jdbi.guice;
import java.net.URI;
import java.util.concurrent.TimeUnit;
import javax.inject.Inject;
import javax.inject.Provider;
import javax.sql.DataSource;
import org.skife.config.TimeSpan;
import com.google.common.annotations.VisibleForTesting;
import com.jolbox.bonecp.BoneCPConfig;
import com.jolbox.bonecp.BoneCPDataSource;
import com.mchange.v2.c3p0.ComboPooledDataSource;
import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;
public class DataSourceProvider implements Provider {
private final DaoConfig config;
private final String poolName;
private final boolean useMariaDB;
private DatabaseType databaseType;
private String dataSourceClassName;
private String driverClassName;
@VisibleForTesting
static enum DatabaseType {
GENERIC,
MYSQL,
H2
}
@Inject
public DataSourceProvider(final DaoConfig config) {
this(config, null);
}
public DataSourceProvider(final DaoConfig config, final String poolName) {
this(config, poolName, true);
}
public DataSourceProvider(final DaoConfig config, final String poolName, final boolean useMariaDB) {
this.config = config;
this.poolName = poolName;
this.useMariaDB = useMariaDB;
parseJDBCUrl();
}
@Override
public DataSource get() {
return getDataSource();
}
private DataSource getDataSource() {
final DataSource ds;
if (DataSourceConnectionPoolingType.C3P0.equals(config.getConnectionPoolingType())) {
loadDriver();
ds = getC3P0DataSource();
} else if (DataSourceConnectionPoolingType.BONECP.equals(config.getConnectionPoolingType())) {
loadDriver();
ds = getBoneCPDatSource();
} else if (DataSourceConnectionPoolingType.HIKARICP.equals(config.getConnectionPoolingType())) {
if (dataSourceClassName != null) {
loadDriver();
}
ds = getHikariCPDataSource();
} else {
throw new IllegalArgumentException("DataSource " + config.getConnectionPoolingType() + " unsupported");
}
return ds;
}
private DataSource getHikariCPDataSource() {
final HikariConfig hikariConfig = new HikariConfig();
hikariConfig.setUsername(config.getUsername());
hikariConfig.setPassword(config.getPassword());
hikariConfig.setMinimumIdle(config.getMinIdle());
hikariConfig.setMaximumPoolSize(config.getMaxActive());
hikariConfig.setConnectionTimeout(toMilliSeconds(config.getConnectionTimeout()));
hikariConfig.setIdleTimeout(toMilliSeconds(config.getIdleMaxAge()));
hikariConfig.setMaxLifetime(toMilliSeconds(config.getMaxConnectionAge()));
// TODO config.getIdleConnectionTestPeriod() ?
hikariConfig.setRegisterMbeans(true);
// TODO Not yet supported
// hikariConfig.setRecordMetrics(true);
if (poolName != null) {
hikariConfig.setPoolName(poolName);
}
hikariConfig.addDataSourceProperty("url", config.getJdbcUrl());
hikariConfig.addDataSourceProperty("user", config.getUsername());
hikariConfig.addDataSourceProperty("password", config.getPassword());
if (DatabaseType.MYSQL.equals(databaseType)) {
// TODO How to configure these on MariaDB?
if (!useMariaDB) {
hikariConfig.addDataSourceProperty("cachePrepStmts", config.isPreparedStatementsCacheEnabled());
hikariConfig.addDataSourceProperty("prepStmtCacheSize", config.getPreparedStatementsCacheSize());
hikariConfig.addDataSourceProperty("prepStmtCacheSqlLimit", config.getPreparedStatementsCacheSqlLimit());
if (Float.valueOf(config.getMySQLServerVersion()) >= 5.1) {
hikariConfig.addDataSourceProperty("useServerPrepStmts", config.isServerSidePreparedStatementsEnabled());
}
}
}
if (dataSourceClassName != null) {
hikariConfig.setDataSourceClassName(dataSourceClassName);
} else {
// Old-school DriverManager-based JDBC
hikariConfig.setJdbcUrl(config.getJdbcUrl());
if (driverClassName != null) {
hikariConfig.setDriverClassName(driverClassName);
}
}
return new HikariDataSource(hikariConfig);
}
private DataSource getBoneCPDatSource() {
final BoneCPConfig dbConfig = new BoneCPConfig();
dbConfig.setJdbcUrl(config.getJdbcUrl());
dbConfig.setUsername(config.getUsername());
dbConfig.setPassword(config.getPassword());
dbConfig.setMinConnectionsPerPartition(config.getMinIdle());
dbConfig.setMaxConnectionsPerPartition(config.getMaxActive());
dbConfig.setConnectionTimeout(config.getConnectionTimeout().getPeriod(), config.getConnectionTimeout().getUnit());
dbConfig.setIdleMaxAge(config.getIdleMaxAge().getPeriod(), config.getIdleMaxAge().getUnit());
dbConfig.setMaxConnectionAge(config.getMaxConnectionAge().getPeriod(), config.getMaxConnectionAge().getUnit());
dbConfig.setIdleConnectionTestPeriod(config.getIdleConnectionTestPeriod().getPeriod(), config.getIdleConnectionTestPeriod().getUnit());
dbConfig.setPartitionCount(1);
dbConfig.setDisableJMX(false);
dbConfig.setStatementsCacheSize(config.getPreparedStatementsCacheSize());
dbConfig.setPoolName(poolName);
return new BoneCPDataSource(dbConfig);
}
private DataSource getC3P0DataSource() {
final ComboPooledDataSource cpds = new ComboPooledDataSource();
cpds.setJdbcUrl(config.getJdbcUrl());
cpds.setUser(config.getUsername());
cpds.setPassword(config.getPassword());
// http://www.mchange.com/projects/c3p0/#minPoolSize
// Minimum number of Connections a pool will maintain at any given time.
cpds.setMinPoolSize(config.getMinIdle());
// http://www.mchange.com/projects/c3p0/#maxPoolSize
// Maximum number of Connections a pool will maintain at any given time.
cpds.setMaxPoolSize(config.getMaxActive());
// http://www.mchange.com/projects/c3p0/#checkoutTimeout
// The number of milliseconds a client calling getConnection() will wait for a Connection to be checked-in or
// acquired when the pool is exhausted. Zero means wait indefinitely. Setting any positive value will cause the getConnection()
// call to time-out and break with an SQLException after the specified number of milliseconds.
cpds.setCheckoutTimeout(toMilliSeconds(config.getConnectionTimeout()));
// http://www.mchange.com/projects/c3p0/#maxIdleTime
// Seconds a Connection can remain pooled but unused before being discarded. Zero means idle connections never expire.
cpds.setMaxIdleTime(toSeconds(config.getIdleMaxAge()));
// http://www.mchange.com/projects/c3p0/#maxConnectionAge
// Seconds, effectively a time to live. A Connection older than maxConnectionAge will be destroyed and purged from the pool.
// This differs from maxIdleTime in that it refers to absolute age. Even a Connection which has not been much idle will be purged
// from the pool if it exceeds maxConnectionAge. Zero means no maximum absolute age is enforced.
cpds.setMaxConnectionAge(toSeconds(config.getMaxConnectionAge()));
// http://www.mchange.com/projects/c3p0/#idleConnectionTestPeriod
// If this is a number greater than 0, c3p0 will test all idle, pooled but unchecked-out connections, every this number of seconds.
cpds.setIdleConnectionTestPeriod(toSeconds(config.getIdleConnectionTestPeriod()));
// The number of PreparedStatements c3p0 will cache for a single pooled Connection.
// If both maxStatements and maxStatementsPerConnection are zero, statement caching will not be enabled.
// If maxStatementsPerConnection is zero but maxStatements is a non-zero value, statement caching will be enabled,
// and a global limit enforced, but otherwise no limit will be set on the number of cached statements for a single Connection.
// If set, maxStatementsPerConnection should be set to about the number distinct PreparedStatements that are used
// frequently in your application, plus two or three extra so infrequently statements don't force the more common
// cached statements to be culled. Though maxStatements is the JDBC standard parameter for controlling statement caching,
// users may find maxStatementsPerConnection more intuitive to use.
cpds.setMaxStatementsPerConnection(config.getPreparedStatementsCacheSize());
cpds.setDataSourceName(poolName);
return cpds;
}
private int toSeconds(final TimeSpan timeSpan) {
return toSeconds(timeSpan.getPeriod(), timeSpan.getUnit());
}
private int toSeconds(final long period, final TimeUnit timeUnit) {
return (int) TimeUnit.SECONDS.convert(period, timeUnit);
}
private int toMilliSeconds(final TimeSpan timeSpan) {
return toMilliSeconds(timeSpan.getPeriod(), timeSpan.getUnit());
}
private int toMilliSeconds(final long period, final TimeUnit timeUnit) {
return (int) TimeUnit.MILLISECONDS.convert(period, timeUnit);
}
private void parseJDBCUrl() {
final URI uri = URI.create(config.getJdbcUrl().substring(5));
final String schemeLocation;
if (uri.getPath() != null) {
schemeLocation = null;
} else if (uri.getSchemeSpecificPart() != null) {
final String[] schemeParts = uri.getSchemeSpecificPart().split(":");
schemeLocation = schemeParts[0];
} else {
schemeLocation = null;
}
dataSourceClassName = config.getDataSourceClassName();
driverClassName = config.getDriverClassName();
if ("mysql".equals(uri.getScheme())) {
databaseType = DatabaseType.MYSQL;
if (dataSourceClassName == null) {
if (useMariaDB) {
dataSourceClassName = "org.mariadb.jdbc.MySQLDataSource";
} else {
dataSourceClassName = "com.mysql.jdbc.jdbc2.optional.MysqlDataSource";
}
}
if (driverClassName == null) {
if (useMariaDB) {
driverClassName = "org.mariadb.jdbc.Driver";
} else {
driverClassName = "com.mysql.jdbc.Driver";
}
}
} else if ("h2".equals(uri.getScheme()) && ("mem".equals(schemeLocation) || "file".equals(schemeLocation))) {
databaseType = DatabaseType.H2;
if (dataSourceClassName == null) {
dataSourceClassName = "org.h2.jdbcx.JdbcDataSource";
}
if (driverClassName == null) {
driverClassName = "org.h2.Driver";
}
} else {
databaseType = DatabaseType.GENERIC;
}
}
private void loadDriver() {
if (driverClassName != null) {
try {
Class.forName(driverClassName).newInstance();
} catch (final Exception e) {
throw new IllegalStateException(e);
}
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy