perf: 修改数据源配置方式，由列表改为 map

Author: cadecode

simple-application/src/main/resources/application.yml

@@ -19,9 +19,8 @@ mybatis:
 # 自定义配置
 simple:
   datasource:
-    # 数据源列表
     dbs:
-      - name: db1
+      db1:
         # 数据源类型
         type: com.zaxxer.hikari.HikariDataSource
         # 是否设置为默认数据源
@@ -32,7 +31,7 @@ simple:
           driver-class-name: com.mysql.cj.jdbc.Driver
           username: root
           password: xxxx
-      - name: db2
+      db2:
         type: com.zaxxer.hikari.HikariDataSource
         config:
           jdbc-url: jdbc:mysql://localhost:3306/demo_cluster?useSSL=true&useUnicode=true&characterEncoding=UTF-8&serverTimezone=Asia/Shanghai

simple-framework/src/main/java/top/cadecode/framework/config/DataSourceConfig.java

@@ -15,11 +15,8 @@
 
 import javax.sql.DataSource;
 import java.util.HashMap;
-import java.util.List;
 import java.util.Map;
 import java.util.Set;
-import java.util.concurrent.atomic.AtomicInteger;
-import java.util.stream.Collectors;
 
 /**
  * @author Cade Li
@@ -33,36 +30,33 @@
 @ConfigurationProperties("simple.datasource")
 public class DataSourceConfig {
 
-    private static final String DBS_PREFIX = "simple.datasource.dbs[";
-    private static final String DBS_SUFFIX = "].config";
+    private static final String DBS_PREFIX = "simple.datasource.dbs.";
+    private static final String DBS_SUFFIX = ".config";
 
     private final Environment environment;
 
     // 注入数据库配置
-    private List<DbConfigObject> dbs;
+    private Map<String, DbConfigObject> dbs;
 
     @Bean
     public DynamicDataSource dynamicDataSource() {
         log.info("配置动态数据源");
         DynamicDataSource dynamicDS = new DynamicDataSource();
         // 获取数据库 key 集合
-        Set<String> keys = dbs.stream()
-                .map(DbConfigObject::getName)
-                .collect(Collectors.toSet());
+        Set<String> keys = dbs.keySet();
         // 存储 keys
         DynamicDataSourceHolder.addDataSourceKeys(keys);
         // 定义数据源容器
         Map<Object, Object> dataSourceMap = new HashMap<>();
         // 遍历 dbs
-        AtomicInteger index = new AtomicInteger();
-        dbs.forEach(db -> {
-            log.info("解析数据源配置 " + db.getName());
-            DataSource dataSource = getDataSource(db.getType(), db.getName(), index.getAndIncrement());
-            dataSourceMap.put(db.getName(), dataSource);
+        dbs.forEach((name, db) -> {
+            log.info("解析数据源配置 " + name);
+            DataSource dataSource = getDataSource(db.getType(), name);
+            dataSourceMap.put(name, dataSource);
             // 设置默认数据源
             if (!dynamicDS.hasDefaultDataSource() && db.isDefaultFlag()) {
                 dynamicDS.setDefaultTargetDataSource(dataSource);
-                log.info("设置默认数据源为 " + db.getName());
+                log.info("设置默认数据源为 " + name);
             }
         });
         // 判断是否设置默认数据源
@@ -83,7 +77,7 @@ public PlatformTransactionManager transactionManager() {
      * 获取数据源实例工具方法
      */
     @SuppressWarnings("unchecked")
-    private DataSource getDataSource(String type, String name, int index) {
+    private DataSource getDataSource(String type, String name) {
         Class<DataSource> dataSourceClass;
         try {
             dataSourceClass = (Class<DataSource>) Class.forName(type);
@@ -93,7 +87,7 @@ private DataSource getDataSource(String type, String name, int index) {
             }
             // 生成数据源实例
             return Binder.get(environment)
-                    .bind(DBS_PREFIX + index + DBS_SUFFIX, dataSourceClass)
+                    .bind(DBS_PREFIX + name + DBS_SUFFIX, dataSourceClass)
                     .get();
         } catch (ClassNotFoundException e) {
             throw new IllegalArgumentException("数据源 " + name + " 的类型 " + type + " 不存在");
@@ -107,7 +101,6 @@ private DataSource getDataSource(String type, String name, int index) {
      */
     @Data
     public static class DbConfigObject {
-        private String name;
         private String type;
         private boolean defaultFlag;
     }