package com.common.component;
|
|
import com.alibaba.druid.filter.Filter;
|
import com.alibaba.druid.pool.DruidDataSource;
|
import com.jfinal.kit.StrKit;
|
import com.jfinal.plugin.IPlugin;
|
import com.jfinal.plugin.activerecord.IDataSourceProvider;
|
import java.sql.SQLException;
|
import java.util.ArrayList;
|
import java.util.Iterator;
|
import java.util.List;
|
import javax.sql.DataSource;
|
|
public class DruidPlugin implements IPlugin, IDataSourceProvider {
|
protected String name = null;
|
protected String url;
|
protected String username;
|
protected String password;
|
protected String publicKey;
|
protected String driverClass = null;
|
protected int initialSize = 1;
|
protected int minIdle = 10;
|
protected int maxActive = 32;
|
protected long maxWait = -1L;
|
protected long timeBetweenEvictionRunsMillis = 60000L;
|
protected long minEvictableIdleTimeMillis = 1800000L;
|
protected long timeBetweenConnectErrorMillis = 30000L;
|
protected String validationQuery = "select 1";
|
protected String connectionInitSql = null;
|
protected String connectionProperties = null;
|
protected boolean testWhileIdle = true;
|
protected boolean testOnBorrow = false;
|
protected boolean testOnReturn = false;
|
protected boolean removeAbandoned = false;
|
protected long removeAbandonedTimeoutMillis = 300000L;
|
protected boolean logAbandoned = false;
|
protected int maxPoolPreparedStatementPerConnectionSize = -1;
|
protected Integer defaultTransactionIsolation = null;
|
protected Integer validationQueryTimeout = null;
|
protected Integer timeBetweenLogStatsMillis = null;
|
protected Boolean keepAlive = null;
|
protected String filters;
|
protected List<Filter> filterList;
|
protected DruidDataSource ds;
|
protected volatile boolean isStarted = false;
|
|
public DruidPlugin(String url, String username, String password) {
|
this.url = url;
|
this.username = username;
|
this.password = password;
|
this.validationQuery = autoCheckValidationQuery(url);
|
}
|
|
public DruidPlugin(String url, String username, String password, String driverClass) {
|
this.url = url;
|
this.username = username;
|
this.password = password;
|
this.driverClass = driverClass;
|
this.validationQuery = autoCheckValidationQuery(url);
|
}
|
|
public DruidPlugin(String url, String username, String password, String driverClass, String filters) {
|
this.url = url;
|
this.username = username;
|
this.password = password;
|
this.driverClass = driverClass;
|
this.filters = filters;
|
this.validationQuery = autoCheckValidationQuery(url);
|
}
|
|
private static String autoCheckValidationQuery(String url) {
|
if (url.startsWith("jdbc:oracle")) {
|
return "select 1 from dual";
|
} else if (url.startsWith("jdbc:db2")) {
|
return "select 1 from sysibm.sysdummy1";
|
} else if (url.startsWith("jdbc:hsqldb")) {
|
return "select 1 from INFORMATION_SCHEMA.SYSTEM_USERS";
|
} else {
|
return url.startsWith("jdbc:derby") ? "select 1 from INFORMATION_SCHEMA.SYSTEM_USERS" : "select 1";
|
}
|
}
|
|
public void setConnectionInitSql(String sql) {
|
this.connectionInitSql = sql;
|
}
|
|
public final String getName() {
|
return this.name;
|
}
|
|
public final void setName(String name) {
|
this.name = name;
|
}
|
|
public DruidPlugin setFilters(String filters) {
|
this.filters = filters;
|
return this;
|
}
|
|
public synchronized DruidPlugin addFilter(Filter filter) {
|
if (this.filterList == null) {
|
this.filterList = new ArrayList();
|
}
|
|
this.filterList.add(filter);
|
return this;
|
}
|
|
public boolean start() {
|
if (this.isStarted) {
|
return true;
|
} else {
|
this.ds = new DruidDataSource();
|
if (this.name != null) {
|
this.ds.setName(this.name);
|
}
|
|
this.ds.setUrl(this.url);
|
this.ds.setUsername(this.username);
|
this.ds.setPassword(this.password);
|
if (this.driverClass != null) {
|
this.ds.setDriverClassName(this.driverClass);
|
}
|
|
this.ds.setInitialSize(this.initialSize);
|
this.ds.setMinIdle(this.minIdle);
|
this.ds.setMaxActive(this.maxActive);
|
this.ds.setMaxWait(this.maxWait);
|
this.ds.setTimeBetweenConnectErrorMillis(this.timeBetweenConnectErrorMillis);
|
this.ds.setTimeBetweenEvictionRunsMillis(this.timeBetweenEvictionRunsMillis);
|
this.ds.setMinEvictableIdleTimeMillis(this.minEvictableIdleTimeMillis);
|
this.ds.setValidationQuery(this.validationQuery);
|
if (StrKit.notBlank(this.connectionInitSql)) {
|
List<String> connectionInitSqls = new ArrayList();
|
connectionInitSqls.add(this.connectionInitSql);
|
this.ds.setConnectionInitSqls(connectionInitSqls);
|
}
|
this.ds.setConnectionErrorRetryAttempts(5);
|
this.ds.setTestWhileIdle(this.testWhileIdle);
|
this.ds.setTestOnBorrow(this.testOnBorrow);
|
this.ds.setTestOnReturn(this.testOnReturn);
|
this.ds.setRemoveAbandoned(this.removeAbandoned);
|
this.ds.setRemoveAbandonedTimeoutMillis(this.removeAbandonedTimeoutMillis);
|
this.ds.setLogAbandoned(this.logAbandoned);
|
this.ds.setMaxPoolPreparedStatementPerConnectionSize(this.maxPoolPreparedStatementPerConnectionSize);
|
if (this.defaultTransactionIsolation != null) {
|
this.ds.setDefaultTransactionIsolation(this.defaultTransactionIsolation);
|
}
|
|
if (this.validationQueryTimeout != null) {
|
this.ds.setValidationQueryTimeout(this.validationQueryTimeout);
|
}
|
|
if (this.timeBetweenLogStatsMillis != null) {
|
this.ds.setTimeBetweenLogStatsMillis((long)this.timeBetweenLogStatsMillis);
|
}
|
|
if (this.keepAlive != null) {
|
this.ds.setKeepAlive(this.keepAlive);
|
}
|
|
boolean hasSetConnectionProperties = false;
|
if (StrKit.notBlank(this.filters)) {
|
try {
|
this.ds.setFilters(this.filters);
|
if (this.filters.contains("config")) {
|
if (StrKit.isBlank(this.publicKey)) {
|
throw new RuntimeException("Druid连接池的filter设定了config时，必须设定publicKey");
|
}
|
|
String decryptStr = "config.decrypt=true;config.decrypt.key=" + this.publicKey;
|
String cp = this.connectionProperties;
|
if (StrKit.isBlank(cp)) {
|
cp = decryptStr;
|
} else {
|
cp = cp + ";" + decryptStr;
|
}
|
|
this.ds.setConnectionProperties(cp);
|
hasSetConnectionProperties = true;
|
}
|
} catch (SQLException var4) {
|
throw new RuntimeException(var4);
|
}
|
}
|
|
if (!hasSetConnectionProperties && StrKit.notBlank(this.connectionProperties)) {
|
this.ds.setConnectionProperties(this.connectionProperties);
|
}
|
|
this.addFilterList(this.ds);
|
this.isStarted = true;
|
return true;
|
}
|
}
|
|
private void addFilterList(DruidDataSource ds) {
|
if (this.filterList != null) {
|
List<Filter> targetList = ds.getProxyFilters();
|
Iterator var3 = this.filterList.iterator();
|
|
while(var3.hasNext()) {
|
Filter add = (Filter)var3.next();
|
boolean found = false;
|
Iterator var6 = targetList.iterator();
|
|
while(var6.hasNext()) {
|
Filter target = (Filter)var6.next();
|
if (add.getClass().equals(target.getClass())) {
|
found = true;
|
break;
|
}
|
}
|
|
if (!found) {
|
targetList.add(add);
|
}
|
}
|
}
|
|
}
|
|
public boolean stop() {
|
if (this.ds != null) {
|
this.ds.close();
|
}
|
|
this.ds = null;
|
this.isStarted = false;
|
return true;
|
}
|
|
public DataSource getDataSource() {
|
return this.ds;
|
}
|
|
public DruidPlugin set(int initialSize, int minIdle, int maxActive) {
|
this.initialSize = initialSize;
|
this.minIdle = minIdle;
|
this.maxActive = maxActive;
|
return this;
|
}
|
|
public DruidPlugin setDriverClass(String driverClass) {
|
this.driverClass = driverClass;
|
return this;
|
}
|
|
public DruidPlugin setInitialSize(int initialSize) {
|
this.initialSize = initialSize;
|
return this;
|
}
|
|
public DruidPlugin setMinIdle(int minIdle) {
|
this.minIdle = minIdle;
|
return this;
|
}
|
|
public DruidPlugin setMaxActive(int maxActive) {
|
this.maxActive = maxActive;
|
return this;
|
}
|
|
public DruidPlugin setMaxWait(long maxWait) {
|
this.maxWait = maxWait;
|
return this;
|
}
|
|
public DruidPlugin setDefaultTransactionIsolation(int defaultTransactionIsolation) {
|
this.defaultTransactionIsolation = defaultTransactionIsolation;
|
return this;
|
}
|
|
public DruidPlugin setValidationQueryTimeout(int validationQueryTimeout) {
|
this.validationQueryTimeout = validationQueryTimeout;
|
return this;
|
}
|
|
public DruidPlugin setTimeBetweenLogStatsMillis(int timeBetweenLogStatsMillis) {
|
this.timeBetweenLogStatsMillis = timeBetweenLogStatsMillis;
|
return this;
|
}
|
|
public DruidPlugin setKeepAlive(boolean keepAlive) {
|
this.keepAlive = keepAlive;
|
return this;
|
}
|
|
public DruidPlugin setTimeBetweenEvictionRunsMillis(long timeBetweenEvictionRunsMillis) {
|
this.timeBetweenEvictionRunsMillis = timeBetweenEvictionRunsMillis;
|
return this;
|
}
|
|
public DruidPlugin setMinEvictableIdleTimeMillis(long minEvictableIdleTimeMillis) {
|
this.minEvictableIdleTimeMillis = minEvictableIdleTimeMillis;
|
return this;
|
}
|
|
public DruidPlugin setValidationQuery(String validationQuery) {
|
this.validationQuery = validationQuery;
|
return this;
|
}
|
|
public DruidPlugin setTestWhileIdle(boolean testWhileIdle) {
|
this.testWhileIdle = testWhileIdle;
|
return this;
|
}
|
|
public DruidPlugin setTestOnBorrow(boolean testOnBorrow) {
|
this.testOnBorrow = testOnBorrow;
|
return this;
|
}
|
|
public DruidPlugin setTestOnReturn(boolean testOnReturn) {
|
this.testOnReturn = testOnReturn;
|
return this;
|
}
|
|
public DruidPlugin setMaxPoolPreparedStatementPerConnectionSize(int maxPoolPreparedStatementPerConnectionSize) {
|
this.maxPoolPreparedStatementPerConnectionSize = maxPoolPreparedStatementPerConnectionSize;
|
return this;
|
}
|
|
public final DruidPlugin setTimeBetweenConnectErrorMillis(long timeBetweenConnectErrorMillis) {
|
this.timeBetweenConnectErrorMillis = timeBetweenConnectErrorMillis;
|
return this;
|
}
|
|
public final DruidPlugin setRemoveAbandoned(boolean removeAbandoned) {
|
this.removeAbandoned = removeAbandoned;
|
return this;
|
}
|
|
public final DruidPlugin setRemoveAbandonedTimeoutMillis(long removeAbandonedTimeoutMillis) {
|
this.removeAbandonedTimeoutMillis = removeAbandonedTimeoutMillis;
|
return this;
|
}
|
|
public final DruidPlugin setLogAbandoned(boolean logAbandoned) {
|
this.logAbandoned = logAbandoned;
|
return this;
|
}
|
|
public final DruidPlugin setConnectionProperties(String connectionProperties) {
|
this.connectionProperties = connectionProperties;
|
return this;
|
}
|
|
public final DruidPlugin setPublicKey(String publicKey) {
|
this.publicKey = publicKey;
|
return this;
|
}
|
}
|
