package foundation.persist.source;
|
|
import java.sql.Connection;
|
import java.sql.SQLException;
|
|
import com.alibaba.druid.pool.DruidDataSource;
|
|
import foundation.json.IJSONProvider;
|
import foundation.json.IJSONWriter;
|
import foundation.server.config.DBaseType;
|
import foundation.util.Util;
|
|
public class NamedDataSource extends DruidDataSource implements IJSONProvider {
|
|
private static final long serialVersionUID = 1L;
|
private String name;
|
private DBaseType dbaseType;
|
private String version;
|
|
public NamedDataSource(String name) throws SQLException {
|
this.name = name;
|
|
setInitialSize(1);
|
setFilters("stat,log4j2");
|
setMaxWait(60000);
|
setMinIdle(1);
|
setTimeBetweenEvictionRunsMillis(3000);
|
setMinEvictableIdleTimeMillis(3000 * 500);
|
setRemoveAbandonedTimeout(300);
|
setTestWhileIdle(true);
|
setTestOnBorrow(false);
|
setTestOnReturn(false);
|
|
dbaseType = DBaseType.Unknown;
|
}
|
|
public static NamedDataSource getInstance() {
|
return DataSourceManager.getMain();
|
}
|
|
public static NamedDataSource getInstance(String name) {
|
NamedDataSource result = DataSourceManager.getDataSource(name);
|
return result;
|
}
|
|
public void setConfigVesion(String version) {
|
this.version = version;
|
|
if (dbaseType != null) {
|
dbaseType.setVersion(version);
|
}
|
}
|
|
public String getConfigVersion() {
|
return version;
|
}
|
|
@Override
|
public void setUrl(String url) {
|
super.setUrl(url);
|
|
if (url == null) {
|
return;
|
}
|
|
url = url.toLowerCase();
|
if (url.indexOf("oracle") > 0) {
|
dbaseType = DBaseType.Oracle;
|
setValidationQuery("select 1 from dual");
|
}
|
else if (url.indexOf("mysql") > 0) {
|
dbaseType = DBaseType.MySQL;
|
setValidationQuery("select 1");
|
}
|
else if (url.indexOf("sqlserver") > 0) {
|
dbaseType = DBaseType.SQLServer;
|
setValidationQuery("select 1");
|
}
|
else {
|
dbaseType = DBaseType.Unknown;
|
setValidationQuery("select 1");
|
}
|
}
|
|
public ConnectionAgent getConnectionAgent() throws SQLException {
|
Connection conn = getConnection();
|
return new ConnectionAgent(name, conn, dbaseType, version);
|
}
|
|
public String getName() {
|
return name;
|
}
|
|
public DBaseType getDBaseType() {
|
return dbaseType;
|
}
|
|
public boolean keyEquals(String sourceKey) {
|
return this.name.equalsIgnoreCase(sourceKey);
|
}
|
|
public String getSchema() {
|
if (Util.isEmpty(jdbcUrl)) {
|
return null;
|
}
|
try {
|
int startIndex = jdbcUrl.lastIndexOf("/", jdbcUrl.lastIndexOf("/") + 1) + 1;
|
int endIndex = jdbcUrl.indexOf("?", startIndex);
|
String databaseName = jdbcUrl.substring(startIndex, endIndex);
|
// 提取database name（去掉前面的/）
|
// String databaseName = path.substring(1);
|
return databaseName;
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
return null;
|
}
|
|
@Override
|
public void writeJSON(IJSONWriter writer) {
|
writer.beginArray();
|
|
writer.write("name", getName());
|
writer.write("dbType", getDbType());
|
|
writer.write("connectCount", getConnectCount());
|
writer.write("activeCount", getActiveCount());
|
writer.write("closeCount", getCloseCount());
|
writer.write("destroyCount", getDestroyCount());
|
writer.write("scardCount", getDiscardCount());
|
writer.write("activePeak", getActivePeak());
|
writer.write("createTime", getCreatedTime());
|
|
writer.endArray();
|
}
|
}
|
