package com.walker.jdbc.ds;
|
|
import com.alibaba.druid.pool.DruidDataSource;
|
import com.walker.db.DatabaseType;
|
import com.walker.infrastructure.ApplicationRuntimeException;
|
import com.walker.infrastructure.utils.StringUtils;
|
import com.walker.jdbc.Constants;
|
import com.walker.jdbc.DataSourceMeta;
|
import com.walker.jdbc.util.TextUtils;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
|
/**
|
* 包装阿里 <code>DruidDataSource</code> 数据源，形成自己的类型，隔离依赖。
|
* @author 时克英
|
* @date 2023-03-15
|
*/
|
public class MyDruidDataSource extends DruidDataSource implements DataSourceMeta {
|
|
protected final transient Logger logger = LoggerFactory.getLogger(this.getClass());
|
|
@Override
|
public DatabaseType getDatabaseType() {
|
return this.databaseType;
|
}
|
|
@Override
|
public String getDatabaseName() {
|
return this.databaseName;
|
}
|
|
@Override
|
public String getIp() {
|
return this.ip;
|
}
|
|
@Override
|
public int getPort() {
|
return this.port;
|
}
|
|
@Override
|
public String getUsername() {
|
return username;
|
}
|
@Override
|
public String getPassword() {
|
return password;
|
}
|
@Override
|
public String getUrl() {
|
return this.jdbcUrl;
|
}
|
|
@Override
|
public void setPassword(String password) {
|
this.password = password;
|
}
|
@Override
|
public void setUsername(String username) {
|
this.username = username;
|
}
|
|
/**
|
* 重写druid设置url方法，初始化数据源。
|
* @param jdbcUrl
|
* @date 2023-03-15
|
*/
|
@Override
|
public void setUrl(String jdbcUrl) {
|
super.setUrl(jdbcUrl);
|
logger.info("jdbcUrl = " + jdbcUrl);
|
|
// 解析ip、端口、数据库名称
|
String[] ipAndPortAndService = TextUtils.parseIpAndPortFromUrl(jdbcUrl);
|
if(ipAndPortAndService != null){
|
this.ip = ipAndPortAndService[0];
|
this.port = Integer.parseInt(ipAndPortAndService[1]);
|
this.databaseName = ipAndPortAndService[2];
|
}
|
|
String[] values = jdbcUrl.split(StringUtils.SEPARATOR_COLON);
|
if(values == null || values.length <=1){
|
throw new ApplicationRuntimeException("DataSource set jdbcUrl error: " + jdbcUrl);
|
}
|
|
for(String v : values){
|
if(v.equalsIgnoreCase(DatabaseType.NAME_MYSQL)){
|
this.databaseType = DatabaseType.MYSQL;
|
return;
|
} else if(v.equalsIgnoreCase(DatabaseType.NAME_ORACLE)){
|
this.databaseType = DatabaseType.ORACLE;
|
return;
|
} else if(v.equalsIgnoreCase(DatabaseType.NAME_POSTGRES) || v.equalsIgnoreCase(DatabaseType.NAME_POSTGRESQL)){
|
this.databaseType = DatabaseType.POSTGRES;
|
return;
|
} else if(v.equalsIgnoreCase(DatabaseType.NAME_SQLITE)){
|
this.databaseType = DatabaseType.SQLITE;
|
return;
|
} else if(v.equalsIgnoreCase(DatabaseType.NAME_DAMENG)){
|
this.databaseType = DatabaseType.DAMENG;
|
return;
|
} else {
|
//
|
}
|
}
|
|
if(this.databaseType == null){
|
throw new UnsupportedOperationException("不支持的数据库URL参数：" + jdbcUrl);
|
}
|
logger.info("数据库类型：" + this.databaseType.name());
|
}
|
|
/**
|
* 重写druid设置url方法，初始化数据源。
|
* @param driverClassName
|
* @date 2023-03-15
|
*/
|
@Override
|
public void setDriverClassName(String driverClassName){
|
super.setDriverClassName(driverClassName);
|
logger.debug("driverClassName = " + driverClassName);
|
}
|
|
private DatabaseType databaseType = null;
|
private String username;
|
private String password;
|
private String databaseName;
|
|
//=============================================================================================
|
// 以下代码用于手动创建独立数据源使用，比如：采集模块动态创建: DatabaseConnector，2022-09-09，时克英
|
//=============================================================================================
|
|
protected String ip;
|
protected int port = 0;
|
|
public void setIp(String ip) {
|
assert (StringUtils.isNotEmpty(ip));
|
this.ip = ip;
|
}
|
|
public void setPort(int port) {
|
assert(port > 0 && port < Integer.MAX_VALUE);
|
this.port = port;
|
}
|
|
public void setDatabaseType(int databaseTypeValue) {
|
this.databaseType = DatabaseType.getType(databaseTypeValue);
|
String driverName = Constants.getDriverClassName(this.databaseType);
|
this.setDriverClassName(driverName);
|
}
|
|
public void setDatabaseName(String databaseName) {
|
if(StringUtils.isEmpty(databaseName)){
|
throw new IllegalArgumentException("databaseName 必须填写!");
|
}
|
if(databaseType == null){
|
throw new IllegalArgumentException("databaseType must be set firstly!");
|
}
|
if(databaseType != DatabaseType.DERBY && StringUtils.isEmpty(ip)){
|
throw new IllegalArgumentException("ip or port must be set firstly!");
|
}
|
String url = Constants.getJdbcUrl(this.databaseType, databaseName, this.ip, this.port);
|
this.setUrl(url);
|
}
|
}
|
