|
|
@@ -8,8 +8,10 @@ import org.springframework.context.EnvironmentAware;
|
|
|
import org.springframework.context.annotation.ImportBeanDefinitionRegistrar;
|
|
|
import org.springframework.core.env.Environment;
|
|
|
import org.springframework.core.type.AnnotationMetadata;
|
|
|
+import org.springframework.stereotype.Component;
|
|
|
|
|
|
import javax.sql.DataSource;
|
|
|
+import java.sql.*;
|
|
|
import java.util.HashMap;
|
|
|
import java.util.Map;
|
|
|
|
|
|
@@ -20,7 +22,7 @@ import java.util.Map;
|
|
|
* 初始化数据源和提供了执行动态切换数据源的工具类
|
|
|
* EnvironmentAware（获取配置文件配置的属性值）
|
|
|
*/
|
|
|
-
|
|
|
+@Component
|
|
|
public class DynamicDataSourceRegister implements ImportBeanDefinitionRegistrar, EnvironmentAware {
|
|
|
|
|
|
|
|
|
@@ -31,43 +33,115 @@ public class DynamicDataSourceRegister implements ImportBeanDefinitionRegistrar,
|
|
|
//用户自定义数据源
|
|
|
private Map<String, DataSource> slaveDataSources = new HashMap<>();
|
|
|
|
|
|
+ Map<Object, Object> targetDataSources = new HashMap<Object, Object>();
|
|
|
+
|
|
|
+ public DataSource getDefaultDataSource() {
|
|
|
+ return defaultDataSource;
|
|
|
+ }
|
|
|
+
|
|
|
+ public void setDefaultDataSource(DataSource defaultDataSource) {
|
|
|
+ this.defaultDataSource = defaultDataSource;
|
|
|
+ }
|
|
|
+
|
|
|
+ public Map<String, DataSource> getSlaveDataSources() {
|
|
|
+ return slaveDataSources;
|
|
|
+ }
|
|
|
+
|
|
|
+ public void setSlaveDataSources(Map<String, DataSource> slaveDataSources) {
|
|
|
+ this.slaveDataSources = slaveDataSources;
|
|
|
+ }
|
|
|
+
|
|
|
+ public Map<Object, Object> getTargetDataSources() {
|
|
|
+ return targetDataSources;
|
|
|
+ }
|
|
|
+
|
|
|
+ public void setTargetDataSources(Map<Object, Object> targetDataSources) {
|
|
|
+ this.targetDataSources = targetDataSources;
|
|
|
+ }
|
|
|
+
|
|
|
+ // 读取主数据源
|
|
|
+ Map<String, Object> dsDeMap = new HashMap<>();
|
|
|
+
|
|
|
@Override
|
|
|
public void setEnvironment(Environment environment) {
|
|
|
initDefaultDataSource(environment);
|
|
|
- initslaveDataSources(environment);
|
|
|
+ initslaveDataSources();
|
|
|
}
|
|
|
|
|
|
private void initDefaultDataSource(Environment env) {
|
|
|
// 读取主数据源
|
|
|
- Map<String, Object> dsMap = new HashMap<>();
|
|
|
- dsMap.put("driver", env.getProperty("spring.datasource.driver-class-name"));
|
|
|
- dsMap.put("url", env.getProperty("spring.datasource.url"));
|
|
|
- dsMap.put("username", env.getProperty("spring.datasource.username"));
|
|
|
- dsMap.put("password", env.getProperty("spring.datasource.password"));
|
|
|
- System.out.println("主数据源:" + dsMap);
|
|
|
- defaultDataSource = buildDataSource(dsMap);
|
|
|
+ dsDeMap = new HashMap<>();
|
|
|
+ dsDeMap.put("driver", env.getProperty("spring.datasource.driver-class-name"));
|
|
|
+ dsDeMap.put("url", env.getProperty("spring.datasource.url"));
|
|
|
+ dsDeMap.put("username", env.getProperty("spring.datasource.username"));
|
|
|
+ dsDeMap.put("password", env.getProperty("spring.datasource.password"));
|
|
|
+ System.out.println("主数据源:" + dsDeMap);
|
|
|
+ defaultDataSource = buildDataSource(dsDeMap);
|
|
|
}
|
|
|
|
|
|
|
|
|
- private void initslaveDataSources(Environment env) {
|
|
|
+ private void initslaveDataSources(){
|
|
|
// 读取配置文件获取更多数据源
|
|
|
- String dsPrefixs = env.getProperty("slave.datasource.names");
|
|
|
- for (String dsPrefix : dsPrefixs.split(",")) {
|
|
|
- // 多个数据源
|
|
|
- Map<String, Object> dsMap = new HashMap<>();
|
|
|
- dsMap.put("driver", env.getProperty("slave.datasource." + dsPrefix + ".driver-class-name"));
|
|
|
- dsMap.put("url", env.getProperty("slave.datasource." + dsPrefix + ".url"));
|
|
|
- dsMap.put("username", env.getProperty("slave.datasource." + dsPrefix + ".username"));
|
|
|
- dsMap.put("password", env.getProperty("slave.datasource." + dsPrefix + ".password"));
|
|
|
- DataSource ds = buildDataSource(dsMap);
|
|
|
- System.out.println("MAP" + dsMap);
|
|
|
- slaveDataSources.put(dsPrefix, ds);
|
|
|
+// String dsPrefixs = env.getProperty("slave.datasource.names");
|
|
|
+// for (String dsPrefix : dsPrefixs.split(",")) {
|
|
|
+// // 多个数据源
|
|
|
+// Map<String, Object> dsMap = new HashMap<>();
|
|
|
+// dsMap.put("driver", env.getProperty("slave.datasource." + dsPrefix + ".driver-class-name"));
|
|
|
+// dsMap.put("url", env.getProperty("slave.datasource." + dsPrefix + ".url"));
|
|
|
+// dsMap.put("username", env.getProperty("slave.datasource." + dsPrefix + ".username"));
|
|
|
+// dsMap.put("password", env.getProperty("slave.datasource." + dsPrefix + ".password"));
|
|
|
+// DataSource ds = buildDataSource(dsMap);
|
|
|
+// System.out.println("MAP" + dsMap);
|
|
|
+// slaveDataSources.put(dsPrefix, ds);
|
|
|
+// }
|
|
|
+
|
|
|
+ String driver = String.valueOf(dsDeMap.get("driver"));
|
|
|
+ String url = String.valueOf(dsDeMap.get("url"));
|
|
|
+ String userName = String.valueOf(dsDeMap.get("username"));
|
|
|
+ String passWord = String.valueOf(dsDeMap.get("password"));
|
|
|
+ String sql = "select id, bases_name, addrass, port, database_type, data_name, user_name, pass_word from bi_databases";
|
|
|
+
|
|
|
+ Connection conn = null;
|
|
|
+ PreparedStatement pstmt = null;
|
|
|
+ ResultSet rs = null;//定义存放查询结果的结果集
|
|
|
+ try {
|
|
|
+ Class.forName(driver);
|
|
|
+ conn = DriverManager.getConnection(url, userName, passWord);
|
|
|
+ pstmt = conn.prepareStatement(sql);
|
|
|
+ rs = pstmt.executeQuery();//运行查询操作
|
|
|
+ if(rs.next()){
|
|
|
+ System.out.println("rs:" + rs);
|
|
|
+ System.out.println("userName:" +rs.getString("user_name"));
|
|
|
+ // 多个数据源
|
|
|
+ Map<String, Object> dsMap = new HashMap<>();
|
|
|
+ String dei = rs.getString("database_type");
|
|
|
+ dsMap.put("driver", "oracle.jdbc.driver.OracleDriver");
|
|
|
+ dsMap.put("url", "jdbc:" + "oracle" + ":thin:@"+ rs.getString("addrass") + ":"
|
|
|
+ + rs.getString("port") + ":" + rs.getString("data_name"));
|
|
|
+ dsMap.put("username", rs.getString("user_name"));
|
|
|
+ dsMap.put("password", rs.getString("pass_word"));
|
|
|
+ DataSource ds = buildDataSource(dsMap);
|
|
|
+ slaveDataSources.put(rs.getString("user_name"), ds);
|
|
|
+ }
|
|
|
+ } catch (ClassNotFoundException e) {
|
|
|
+ // TODO Auto-generated catch block
|
|
|
+ e.printStackTrace();
|
|
|
+ } catch (SQLException e) {
|
|
|
+ // TODO Auto-generated catch block
|
|
|
+ e.printStackTrace();
|
|
|
+ }finally {
|
|
|
+ try {
|
|
|
+ conn.close();
|
|
|
+ } catch (SQLException e) {
|
|
|
+ e.printStackTrace();
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
|
|
|
+
|
|
|
@Override
|
|
|
public void registerBeanDefinitions(AnnotationMetadata annotationMetadata, BeanDefinitionRegistry beanDefinitionRegistry) {
|
|
|
- Map<Object, Object> targetDataSources = new HashMap<Object, Object>();
|
|
|
+
|
|
|
//添加默认数据源
|
|
|
targetDataSources.put("dataSource", this.defaultDataSource);
|
|
|
DynamicDataSourceContextHolder.dataSourceIds.add("dataSource");
|
heqinwei