Package com.killcoding.datasource

Class DriverDataSource

java.lang.Object

com.killcoding.datasource.DriverDataSource

All Implemented Interfaces:

Wrapper, CommonDataSource, DataSource

public final class DriverDataSource
extends Object
implements DataSource

Implement and enhance the datasource class Support write connection status log Support abort tiemout connection Support failover

Field Summary

Fields

Modifier and Type	Field	Description
private String	abortMode
private File	appDataSource
private String	appDataSourcePath
private final List<Double>	aveUsageSum
private boolean	closed
private ConfigProperties	configProperties
private final Map<Integer,Long>	CONN_ERROR_TIME
private final ConnectionPool	connectionPool
private final ExecutorService	connectionPoolExecutor
private String	connProps
private long	countAbortedCloseFailed
private long	countAbortedIdle
private long	countAbortedInvalid
private long	countAbortedLifeCycle
private long	countAbortedTimeout
private long	countClosed
private long	countCreatedConnection
private String	DATASOURCE_ID
private final Map<String,Driver>	DRIVER_MARK
private String	driverClassName
private String	driverJar
private long	dynamicReloadConfigTime
private Boolean	enableFullStatusLog
private Boolean	enableStatusLog
private Long	idleTimeout
private boolean	inited
private Long	lifeCycle
private boolean	loadedConfig
private final Logger	log
private Integer	maxAveUsageSize
private Integer	maxPoolSize
private double	maxUsage
private Integer	minPoolSize
private String	name
private String	owner
private String	password
private final long	PID
private Long	refreshPoolTimer
private Integer	statusLogArchiveDays
private File	statusLogFile
private String	statusLogFolder
private List<DriverDataSource>	syncDriverDataSource
private Long	testValidTimeout
private Long	timeout
private Long	transactionTimeout
private String	url
private int	urlIndex
private final List<String>	urlList
private boolean	useExternalConfig
private String	username

Constructor Summary

Constructors

Constructor	Description
DriverDataSource()	New a DriverDataSource object Use env APP_DATASOURCE=/path/DataSource.properties
DriverDataSource(ConfigProperties configProperties)	New a DriverDataSource object
DriverDataSource(File appDataSource)	New a DriverDataSource object
DriverDataSource(String configPropertiesContent)	New a DriverDataSource object

Method Summary

Modifier and Type	Method	Description
protected void	abort(Integer hashCode)	Abort connection by hash code
private void	abortIdleTimeoutConnection()	Abort idle timeout connection Skip execute when 'idleTimeout' is 0
private void	abortInvalidConnection()	Abort invalid connection Skip execute when 'testValidTimeout' is 0
private void	abortLifeCycleEndedConnection()	Abort life cycle connection Skip execute when 'lifeCycle' is 0
private void	abortTimeoutConnection()	Abort timeout connection Skip execute when 'timeout' is 0 or 'transactionTimeout' is 0
private void	addCountAbortedCloseFailed(int value)	Add count abort close failed for write status log
private void	archiveLog()	Archive log
private void	callAbortConn(Connection conn)	Call abort connection Please see 'abortMode' options (rollback,close,abort) Default is 'abort'
private void	checkConnectionPoolSize()
void	close()	Close all connections
void	closeAll()	Loop close all connections
protected void	closeConnection(Integer hashCode)	Close connection
protected void	closeConnection(Connection conn)	Close connection
private void	countUsage(boolean calAve)
ConnectionBuilder	createConnectionBuilder()
private void	createConnectionToPool()	Create connection to pool When, (loadPoolSize + usedPoolSize) < maxPoolSize and loadPoolSize < minPoolSize
private void	deleteFilesOlderThan(File directory, long fileOutArchiveDaysMs)	Delete old archive log
static void	deregisterDriver()	Deregister JDBC Driver For system exit
String	getAbortMode()	Get abort mode
String	getAppDataSourcePath()	Get app datasource properties file path
double	getAveUsage()
ConfigProperties	getConfigProperties()	Get config properties
Connection	getConnection()	Get connection
Connection	getConnection(String arg0, String arg1)
private Connection	getConnectionFromDriver()	Get connection from driver
ConnectionPool	getConnectionPool()	Get connection pool size
private Properties	getConnectionProperties()	Get connection properties
String	getConnProps()	Get connection properties
String	getDriverClassName()	Get driver class name
String	getDriverJar()	Get jdbc driver jar path
String	getFingerprint()	For checksum master and sync data source connection url Pause sync where same connection url
Long	getIdleTimeout()	Get idle timeout
Long	getLifeCycle()	Get connection life cycle
int	getLoginTimeout()
PrintWriter	getLogWriter()
Integer	getMaxPoolSize()	Get max connection pool size
Integer	getMinPoolSize()	Get min connection pool size
String	getName()	Get thread name
private Integer	getOpenConnectionSize()
String	getOwner()	Get owner thread name For use sync DataSource
Logger	getParentLogger()
Long	getRefreshPoolTimer()	Get refresh pool timer
protected final List<DriverDataSource>	getSyncDataSourceList(String owner)	Get sync data source list
Long	getTestValidTimeout()	Get test valid timeout
Long	getTimeout()	Get execute timeout
Long	getTransactionTimeout()	Get execute transaction timeout
String	getUrl()	Get connection url
double	getUsage()	Get pool using
private void	initDs()
boolean	isLoadedConfig()	Check is loaded config
boolean	isWrapperFor(Class arg0)
private void	loadConnectionToPool()	Load connection to pool
private Driver	loadJdbcDriver(String driverJar, String driver)	Load jdbc driver jar
private void	manageConnectionPool()	Manage connection pool
private void	reloadConfig()	Reload config When 'appDataSource' is file object
void	setAbortMode(String abortMode)	Set abort mode
void	setAppDataSourcePath(String appDataSourcePath)	Set app datasource properties file path
void	setConfig(ConfigProperties _configProperties)	Set config
void	setConfig(String configPropertiesContent)	Set config
void	setConnProps(String connProps)	Set connection properties
void	setDriverClassName(String driverClassName)	Set driver class name
void	setDriverJar(String driverJar)	Set jdbc driver jar path
void	setEnableFullStatusLog(Boolean enableFullStatusLog)	Set enable full status log
void	setEnableStatusLog(Boolean enableStatusLog)	Set enable status log
void	setIdleTimeout(Long idleTimeout)	Set idle timeout
private void	setInternalConfig(ConfigProperties _configProperties)	Set internal config
private void	setInternalConfig(String configPropertiesContent)	Set internal config
void	setLifeCycle(Long lifeCycle)	Set connection life cycle
void	setLoginTimeout(int arg0)
void	setLogWriter(PrintWriter arg0)
void	setMaxAveUsageSize(Integer maxAveUsageSize)
void	setMaxPoolSize(Integer maxPoolSize)	Set max connection pool size
void	setMinPoolSize(Integer minPoolSize)	Set min connection pool size
void	setName(String name)	Set thread name
void	setPassword(String password)	Set database password
void	setRefreshPoolTimer(Long refreshPoolTimer)	Set refresh pool timer
void	setStatusLogArchiveDays(Integer statusLogArchiveDays)	Set status log archive days
void	setStatusLogFolder(String _statusLogFolder)	Set status log target folder
void	setTestValidTimeout(Long testValidTimeout)	Set test valid timeout
void	setTimeout(Long timeout)	Set execute timeout Default is 0s Disabled timeou if value is 0s
void	setTransactionTimeout(Long transactionTimeout)	Set execute transaction timeout Default is 0s Disabled timeou if value is 0s
void	setUrl(Object urlOrList)	Set connection url
void	setUsername(String username)	Set database username
private void	statusLogOutput()	Write connection status log
private void	throwConnIsNull()
private Connection	tryGetConnection(Driver driver, Properties props)	Try get connection Support multi url
Object	unwrap(Class arg0)	Abandon the following methods

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface javax.sql.CommonDataSource

createShardingKeyBuilder

Field Details

log

private final Logger log

DATASOURCE_ID

private String DATASOURCE_ID

PID

private final long PID

countCreatedConnection

private long countCreatedConnection

countAbortedIdle

private long countAbortedIdle

countAbortedTimeout

private long countAbortedTimeout

countAbortedLifeCycle

private long countAbortedLifeCycle

countAbortedInvalid

private long countAbortedInvalid

countAbortedCloseFailed

private long countAbortedCloseFailed

countClosed

private long countClosed

maxUsage

private double maxUsage

DRIVER_MARK

private final Map<String,Driver> DRIVER_MARK

CONN_ERROR_TIME

private final Map<Integer,Long> CONN_ERROR_TIME

inited

private boolean inited

enableStatusLog

private Boolean enableStatusLog

enableFullStatusLog

private Boolean enableFullStatusLog

statusLogArchiveDays

private Integer statusLogArchiveDays

statusLogFolder

private String statusLogFolder

statusLogFile

private File statusLogFile

maxAveUsageSize

private Integer maxAveUsageSize

aveUsageSum

private final List<Double> aveUsageSum

urlList

private final List<String> urlList

urlIndex

private int urlIndex

maxPoolSize

private Integer maxPoolSize

minPoolSize

private Integer minPoolSize

lifeCycle

private Long lifeCycle

idleTimeout

private Long idleTimeout

timeout

private Long timeout

transactionTimeout

private Long transactionTimeout

refreshPoolTimer

private Long refreshPoolTimer

testValidTimeout

private Long testValidTimeout

configProperties

private ConfigProperties configProperties

driverJar

private String driverJar

connProps

private String connProps

driverClassName

private String driverClassName

url

private String url

username

private String username

password

private String password

name

private String name

abortMode

private String abortMode

appDataSourcePath

private String appDataSourcePath

useExternalConfig

private boolean useExternalConfig

loadedConfig

private boolean loadedConfig

closed

private boolean closed

appDataSource

private File appDataSource

owner

private String owner

syncDriverDataSource

private List<DriverDataSource> syncDriverDataSource

connectionPool

private final ConnectionPool connectionPool

connectionPoolExecutor

private final ExecutorService connectionPoolExecutor

dynamicReloadConfigTime

private long dynamicReloadConfigTime

Constructor Details

DriverDataSource

public DriverDataSource()

New a DriverDataSource object Use env APP_DATASOURCE=/path/DataSource.properties

DriverDataSource

public DriverDataSource(File appDataSource)

New a DriverDataSource object

Parameters:

appDataSource - - It is file object (e.g. /your_path/DataSource.properties)

DriverDataSource

public DriverDataSource(String configPropertiesContent)
                 throws IOException

New a DriverDataSource object

Parameters:

configPropertiesContent - - It is properties context

Throws:

IOException

DriverDataSource

public DriverDataSource(ConfigProperties configProperties)
                 throws IOException

New a DriverDataSource object

Parameters:

configProperties - - It is ConfigProperties object

Throws:

IOException

Method Details

initDs

private void initDs()

setAppDataSourcePath

public void setAppDataSourcePath(String appDataSourcePath)

Set app datasource properties file path

Parameters:

appDataSourcePath - - file path

getAppDataSourcePath

public String getAppDataSourcePath()

Get app datasource properties file path

isLoadedConfig

public boolean isLoadedConfig()

Check is loaded config

Returns:

boolean

setEnableStatusLog

public void setEnableStatusLog(Boolean enableStatusLog)

Set enable status log

Parameters:

enableStatusLog - - 'true' is enable write status log

setEnableFullStatusLog

public void setEnableFullStatusLog(Boolean enableFullStatusLog)

Set enable full status log

Parameters:

enableFullStatusLog - - 'true' is enable write status log when full connections

setStatusLogArchiveDays

public void setStatusLogArchiveDays(Integer statusLogArchiveDays)

Set status log archive days

Parameters:

statusLogArchiveDays - - Delete before x days log

setStatusLogFolder

public void setStatusLogFolder(String _statusLogFolder)

Set status log target folder

Parameters:

_statusLogFolder - - Target folder

setLifeCycle

public void setLifeCycle(Long lifeCycle)

Set connection life cycle

Parameters:

lifeCycle - - Unit is ms

getLifeCycle

public Long getLifeCycle()

Get connection life cycle

Returns:

Long - Unit is ms

setRefreshPoolTimer

public void setRefreshPoolTimer(Long refreshPoolTimer)

Set refresh pool timer

Parameters:

refreshPoolTimer - - Unit is ms

getRefreshPoolTimer

public Long getRefreshPoolTimer()

Get refresh pool timer

Returns:

Long - Unit is ms

setTestValidTimeout

public void setTestValidTimeout(Long testValidTimeout)

Set test valid timeout

Parameters:

testValidTimeout - Unit is seconds,default is 5s

getTestValidTimeout

public Long getTestValidTimeout()

Get test valid timeout

setIdleTimeout

public void setIdleTimeout(Long idleTimeout)

Set idle timeout

Parameters:

idleTimeout - - Unit is seconds, default is 60s

getIdleTimeout

public Long getIdleTimeout()

Get idle timeout

Returns:

Long - Unit is seconds

setTimeout

public void setTimeout(Long timeout)

Set execute timeout Default is 0s Disabled timeou if value is 0s

Parameters:

timeout - - Unit is seconds,

getTimeout

public Long getTimeout()

Get execute timeout

Returns:

Long - Unit is seconds,

setTransactionTimeout

public void setTransactionTimeout(Long transactionTimeout)

Set execute transaction timeout Default is 0s Disabled timeou if value is 0s

Parameters:

idleTimeout - - Unit is seconds,

getTransactionTimeout

public Long getTransactionTimeout()

Get execute transaction timeout

Returns:

Long - Unit is seconds,

setMaxPoolSize

public void setMaxPoolSize(Integer maxPoolSize)

Set max connection pool size

Parameters:

maxPoolSize -

getMaxPoolSize

public Integer getMaxPoolSize()

Get max connection pool size

Returns:

Integer

setDriverJar

public void setDriverJar(String driverJar)

Set jdbc driver jar path

Parameters:

driverJar - - Driver jar file or folder path

getDriverJar

public String getDriverJar()

Get jdbc driver jar path

Returns:

String

setConnProps

public void setConnProps(String connProps)

Set connection properties

Parameters:

connProps - - Sample: autoReconnection=true;key=value

getConnProps

public String getConnProps()

Get connection properties

Returns:

connProps - Sample: autoReconnection=true;key=value

setDriverClassName

public void setDriverClassName(String driverClassName)

Set driver class name

Parameters:

driverClassName -

getDriverClassName

public String getDriverClassName()

Get driver class name

Returns:

String

setUrl

public void setUrl(Object urlOrList)

Set connection url

Parameters:

urlOrList - - String or List

getUrl

public String getUrl()

Get connection url

setMinPoolSize

public void setMinPoolSize(Integer minPoolSize)

Set min connection pool size

Parameters:

minPoolSize -

getMinPoolSize

public Integer getMinPoolSize()

Get min connection pool size

setAbortMode

public void setAbortMode(String abortMode)

Set abort mode

Parameters:

abortMode - - 'close' or 'abort', default is 'abort'

getAbortMode

public String getAbortMode()

Get abort mode

Returns:

abortMode - 'close' or 'abort', default is 'abort'

setName

public void setName(String name)

Set thread name

Parameters:

name - - It is thread name, default is 'DriverDataSource'

getName

public String getName()

Get thread name

Returns:

String

setUsername

public void setUsername(String username)

Set database username

Parameters:

username - - Database username

setPassword

public void setPassword(String password)

Set database password

Parameters:

password - - Database password

getOpenConnectionSize

private Integer getOpenConnectionSize()

getConnectionPool

public ConnectionPool getConnectionPool()

Get connection pool size

Returns:

Integer

abortTimeoutConnection

private void abortTimeoutConnection()

Abort timeout connection Skip execute when 'timeout' is 0 or 'transactionTimeout' is 0

abortLifeCycleEndedConnection

private void abortLifeCycleEndedConnection()

Abort life cycle connection Skip execute when 'lifeCycle' is 0

abortIdleTimeoutConnection

private void abortIdleTimeoutConnection()

Abort idle timeout connection Skip execute when 'idleTimeout' is 0

abortInvalidConnection

private void abortInvalidConnection()

Abort invalid connection Skip execute when 'testValidTimeout' is 0

createConnectionToPool

private void createConnectionToPool()

Create connection to pool When, (loadPoolSize + usedPoolSize) < maxPoolSize and loadPoolSize < minPoolSize

setInternalConfig

private void setInternalConfig(ConfigProperties _configProperties)

Set internal config

Parameters:

_configProperties -

setInternalConfig

private void setInternalConfig(String configPropertiesContent)
                        throws IOException

Set internal config

Parameters:

configPropertiesContent -

Throws:

IOException

setConfig

public void setConfig(String configPropertiesContent)
               throws IOException

Set config

Parameters:

configPropertiesContent -

Throws:

IOException

setConfig

public void setConfig(ConfigProperties _configProperties)
               throws IOException

Set config

Parameters:

_configProperties -

Throws:

IOException

reloadConfig

private void reloadConfig()

Reload config When 'appDataSource' is file object

archiveLog

private void archiveLog()

Archive log

deleteFilesOlderThan

private void deleteFilesOlderThan(File directory,
 long fileOutArchiveDaysMs)
                           throws IOException

Delete old archive log

Throws:

IOException

statusLogOutput

private void statusLogOutput()

Write connection status log

manageConnectionPool

private void manageConnectionPool()

Manage connection pool

getConnection

public Connection getConnection()
                         throws SQLException

Get connection

Specified by:

getConnection in interface DataSource

Returns:

Connection

Throws:

SQLException

throwConnIsNull

private void throwConnIsNull()
                      throws SQLException

Throws:

SQLException

checkConnectionPoolSize

private void checkConnectionPoolSize()
                              throws SQLException

Throws:

SQLException

getConnectionProperties

private Properties getConnectionProperties()

Get connection properties

Returns:

Properties

getConnectionFromDriver

private Connection getConnectionFromDriver()
                                    throws SQLException

Get connection from driver

Returns:

Connection

Throws:

SQLException

loadConnectionToPool

private void loadConnectionToPool()
                           throws SQLException

Load connection to pool

Throws:

SQLException

closeConnection

protected void closeConnection(Connection conn)
                        throws SQLException

Close connection

Parameters:

conn - - Connection object

Throws:

SQLException

closeConnection

protected void closeConnection(Integer hashCode)
                        throws SQLException

Close connection

Parameters:

hashCode - - Connection hash code

Throws:

SQLException

addCountAbortedCloseFailed

private void addCountAbortedCloseFailed(int value)

Add count abort close failed for write status log

Parameters:

value - - Change value

close

public void close()

Close all connections

closeAll

public void closeAll()

Loop close all connections

deregisterDriver

public static void deregisterDriver()

Deregister JDBC Driver For system exit

tryGetConnection

private Connection tryGetConnection(Driver driver,
 Properties props)

Try get connection Support multi url

Parameters:

driver - - Driver

props - - Connection properties ,var name is 'connProps'

abort

protected void abort(Integer hashCode)

Abort connection by hash code

Parameters:

hashCode - - Connection hash code

callAbortConn

private void callAbortConn(Connection conn)

Call abort connection Please see 'abortMode' options (rollback,close,abort) Default is 'abort'

Parameters:

conn - - Connection

loadJdbcDriver

private Driver loadJdbcDriver(String driverJar,
 String driver)
                       throws SQLException

Load jdbc driver jar

Parameters:

driverJar - - It is jar file path or jar dir path

driver -

Returns:

Driver

Throws:

SQLException

getOwner

public String getOwner()

Get owner thread name For use sync DataSource

Returns:

String

getConfigProperties

public ConfigProperties getConfigProperties()

Get config properties

Returns:

ConfigProperties

getSyncDataSourceList

protected final List<DriverDataSource> getSyncDataSourceList(String owner)
                                                      throws SQLException

Get sync data source list

Parameters:

owner - - From master data source thread name

Returns:

List

Throws:

SQLException

getFingerprint

public String getFingerprint()

For checksum master and sync data source connection url Pause sync where same connection url

Returns:

String

getAveUsage

public double getAveUsage()

setMaxAveUsageSize

public void setMaxAveUsageSize(Integer maxAveUsageSize)

countUsage

private void countUsage(boolean calAve)

getUsage

public double getUsage()

Get pool using

Returns:

double

unwrap

public Object unwrap(Class arg0)
              throws SQLException

Abandon the following methods

Specified by:

unwrap in interface Wrapper

Throws:

SQLException

isWrapperFor

public boolean isWrapperFor(Class arg0)
                     throws SQLException

Specified by:

isWrapperFor in interface Wrapper

Throws:

SQLException

getParentLogger

public Logger getParentLogger()
                       throws SQLFeatureNotSupportedException

Specified by:

getParentLogger in interface CommonDataSource

Throws:

SQLFeatureNotSupportedException

setLogWriter

public void setLogWriter(PrintWriter arg0)
                  throws SQLException

Specified by:

setLogWriter in interface CommonDataSource

Specified by:

setLogWriter in interface DataSource

Throws:

SQLException

getLogWriter

public PrintWriter getLogWriter()
                         throws SQLException

Specified by:

getLogWriter in interface CommonDataSource

Specified by:

getLogWriter in interface DataSource

Throws:

SQLException

getLoginTimeout

public int getLoginTimeout()
                    throws SQLException

Specified by:

getLoginTimeout in interface CommonDataSource

Specified by:

getLoginTimeout in interface DataSource

Throws:

SQLException

setLoginTimeout

public void setLoginTimeout(int arg0)
                     throws SQLException

Specified by:

setLoginTimeout in interface CommonDataSource

Specified by:

setLoginTimeout in interface DataSource

Throws:

SQLException

getConnection

public Connection getConnection(String arg0,
 String arg1)
                         throws SQLException

Specified by:

getConnection in interface DataSource

Throws:

SQLException

createConnectionBuilder

public ConnectionBuilder createConnectionBuilder()
                                          throws SQLException

Specified by:

createConnectionBuilder in interface DataSource

Throws:

SQLException