io.logspace.jvm.agent.shaded.quartz.utils.PoolingConnectionProvider Maven / Gradle / Ivy
The newest version!
/*
* Copyright 2001-2009 Terracotta, Inc.
*
* Licensed 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 io.logspace.jvm.agent.shaded.quartz.utils;
import java.beans.PropertyVetoException;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Properties;
import io.logspace.jvm.agent.shaded.quartz.SchedulerException;
import io.logspace.jvm.agent.shaded.mchange.v2.c3p0.ComboPooledDataSource;
/**
*
* A ConnectionProvider implementation that creates its own
* pool of connections.
*
*
*
* This class uses C3PO (http://www.mchange.com/projects/c3p0/index.html) as
* the underlying pool implementation.
*
* @see DBConnectionManager
* @see ConnectionProvider
*
* @author Sharada Jambula
* @author James House
* @author Mohammad Rezaei
*/
public class PoolingConnectionProvider implements ConnectionProvider {
/*
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*
* Constants.
*
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*/
/** The JDBC database driver. */
public static final String DB_DRIVER = "driver";
/** The JDBC database URL. */
public static final String DB_URL = "URL";
/** The database user name. */
public static final String DB_USER = "user";
/** The database user password. */
public static final String DB_PASSWORD = "password";
/** The maximum number of database connections to have in the pool. Default is 10. */
public static final String DB_MAX_CONNECTIONS = "maxConnections";
/**
* The maximum number of prepared statements that will be cached per connection in the pool.
* Depending upon your JDBC Driver this may significantly help performance, or may slightly
* hinder performance.
* Default is 120, as Quartz uses over 100 unique statements. 0 disables the feature.
*/
public static final String DB_MAX_CACHED_STATEMENTS_PER_CONNECTION = "maxCachedStatementsPerConnection";
/**
* The database sql query to execute every time a connection is returned
* to the pool to ensure that it is still valid.
*/
public static final String DB_VALIDATION_QUERY = "validationQuery";
/**
* The number of seconds between tests of idle connections - only enabled
* if the validation query property is set. Default is 50 seconds.
*/
public static final String DB_IDLE_VALIDATION_SECONDS = "idleConnectionValidationSeconds";
/**
* Whether the database sql query to validate connections should be executed every time
* a connection is retrieved from the pool to ensure that it is still valid. If false,
* then validation will occur on check-in. Default is false.
*/
public static final String DB_VALIDATE_ON_CHECKOUT = "validateOnCheckout";
/** Discard connections after they have been idle this many seconds. 0 disables the feature. Default is 0.*/
private static final String DB_DISCARD_IDLE_CONNECTIONS_SECONDS = "discardIdleConnectionsSeconds";
/** Default maximum number of database connections in the pool. */
public static final int DEFAULT_DB_MAX_CONNECTIONS = 10;
/** Default maximum number of database connections in the pool. */
public static final int DEFAULT_DB_MAX_CACHED_STATEMENTS_PER_CONNECTION = 120;
/*
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*
* Data members.
*
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*/
private ComboPooledDataSource datasource;
/*
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*
* Constructors.
*
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*/
public PoolingConnectionProvider(String dbDriver, String dbURL,
String dbUser, String dbPassword, int maxConnections,
String dbValidationQuery) throws SQLException, SchedulerException {
initialize(
dbDriver, dbURL, dbUser, dbPassword,
maxConnections, DEFAULT_DB_MAX_CACHED_STATEMENTS_PER_CONNECTION, dbValidationQuery, false, 50, 0);
}
/**
* Create a connection pool using the given properties.
*
*
* The properties passed should contain:
*
* - {@link #DB_DRIVER}- The database driver class name
*
- {@link #DB_URL}- The database URL
*
- {@link #DB_USER}- The database user
*
- {@link #DB_PASSWORD}- The database password
*
- {@link #DB_MAX_CONNECTIONS}- The maximum # connections in the pool,
* optional
*
- {@link #DB_VALIDATION_QUERY}- The sql validation query, optional
*
*
*
* @param config
* configuration properties
*/
public PoolingConnectionProvider(Properties config) throws SchedulerException, SQLException {
PropertiesParser cfg = new PropertiesParser(config);
initialize(
cfg.getStringProperty(DB_DRIVER),
cfg.getStringProperty(DB_URL),
cfg.getStringProperty(DB_USER, ""),
cfg.getStringProperty(DB_PASSWORD, ""),
cfg.getIntProperty(DB_MAX_CONNECTIONS, DEFAULT_DB_MAX_CONNECTIONS),
cfg.getIntProperty(DB_MAX_CACHED_STATEMENTS_PER_CONNECTION, DEFAULT_DB_MAX_CACHED_STATEMENTS_PER_CONNECTION),
cfg.getStringProperty(DB_VALIDATION_QUERY),
cfg.getBooleanProperty(DB_VALIDATE_ON_CHECKOUT, false),
cfg.getIntProperty(DB_IDLE_VALIDATION_SECONDS, 50),
cfg.getIntProperty(DB_DISCARD_IDLE_CONNECTIONS_SECONDS, 0));
}
/*
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*
* Interface.
*
* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*/
/**
* Create the underlying C3PO ComboPooledDataSource with the
* default supported properties.
* @throws SchedulerException
*/
private void initialize(
String dbDriver,
String dbURL,
String dbUser,
String dbPassword,
int maxConnections,
int maxStatementsPerConnection,
String dbValidationQuery,
boolean validateOnCheckout,
int idleValidationSeconds,
int maxIdleSeconds) throws SQLException, SchedulerException {
if (dbURL == null) {
throw new SQLException(
"DBPool could not be created: DB URL cannot be null");
}
if (dbDriver == null) {
throw new SQLException(
"DBPool '" + dbURL + "' could not be created: " +
"DB driver class name cannot be null!");
}
if (maxConnections < 0) {
throw new SQLException(
"DBPool '" + dbURL + "' could not be created: " +
"Max connections must be greater than zero!");
}
datasource = new ComboPooledDataSource();
try {
datasource.setDriverClass(dbDriver);
} catch (PropertyVetoException e) {
throw new SchedulerException("Problem setting driver class name on datasource: " + e.getMessage(), e);
}
datasource.setJdbcUrl(dbURL);
datasource.setUser(dbUser);
datasource.setPassword(dbPassword);
datasource.setMaxPoolSize(maxConnections);
datasource.setMinPoolSize(1);
datasource.setMaxIdleTime(maxIdleSeconds);
datasource.setMaxStatementsPerConnection(maxStatementsPerConnection);
if (dbValidationQuery != null) {
datasource.setPreferredTestQuery(dbValidationQuery);
if(!validateOnCheckout)
datasource.setTestConnectionOnCheckin(true);
else
datasource.setTestConnectionOnCheckout(true);
datasource.setIdleConnectionTestPeriod(idleValidationSeconds);
}
}
/**
* Get the C3PO ComboPooledDataSource created during initialization.
*
*
* This can be used to set additional data source properties in a
* subclass's constructor.
*
*/
public ComboPooledDataSource getDataSource() {
return datasource;
}
public Connection getConnection() throws SQLException {
return datasource.getConnection();
}
public void shutdown() throws SQLException {
datasource.close();
}
public void initialize() throws SQLException {
// do nothing, already initialized during constructor call
}
}