這裡是接上篇文章
Spring Boot 動態切換資料來源（負載均衡）
留下的問題，配置檔案中資料庫的資訊更改後代碼中能夠動態獲取。所以這裡重點說下優化後的內容。

先看配置檔案

slave:
  hosts: slave1,slave2
hikari:
  - master:
    jdbc-url: jdbc:mysql://master-host:3306/mydb?useUnicode=true&characterEncoding=utf8&useSSL=true&allowMultiQueries=true&verifyServerCertificate=false
 

    username: root
    password: root
    maximum-pool-size: 10
    pool-name: master
    connection-timeout: 30000
    idle-timeout: 600000
    max-lifetime: 1765000
    data-source-properties:
      cachePrepStmts: true
      prepStmtCacheSize: 250
      prepStmtCacheSqlLimit: 2048
      useServerPrepStmts: true
 

      useLocalSessionState: true
      useLocalTransactionState: true
      rewriteBatchedStatements: true
      cacheResultSetMetadata: true
      cacheServerConfiguration: true
      elideSetAutoCommits: true
      maintainTimeStats: false
  - slave1:
    jdbc-url: jdbc:mysql://slave1-host:3306/mydb?useUnicode=true&characterEncoding=utf8&useSSL=true&allowMultiQueries=true&verifyServerCertificate=false
 

    username: root
    password: root
    maximum-pool-size: 10
    pool-name: slave1
    connection-timeout: 30000
    idle-timeout: 600000
    max-lifetime: 1765000
    read-only: true
  - slave2:
    jdbc-url: jdbc:mysql://slave2-host:3306/mydb?useUnicode=true&characterEncoding=utf8&useSSL=true&allowMultiQueries=true&verifyServerCertificate=false
    username: root
    password: root
    maximum-pool-size: 10
    pool-name: slave2
    connection-timeout: 30000
    idle-timeout: 600000
    max-lifetime: 1765000
    read-only: true

這裡的改動是master和slave前面加了“-”用來支援集合獲取這塊資訊。
另外pool-name改為了master和slave1、slave2，這個很重要，後邊有用到。因為獲取不到前邊的- master和-slave等資訊，所以這裡使用pool-name來確定是哪個資料來源。

DBProperties

import com.zaxxer.hikari.HikariDataSource;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;

import java.util.Map;
/**
 * @author Created by pangkunkun on 2017/12/18.
 */
@Component
@ConfigurationProperties(prefix = "")
public class DBProperties {
    //一次性從配置檔案中讀取所有資料來源的配置
    private Map<String, HikariDataSource> hikari;

    public Map<String, HikariDataSource> getHikari() {
        return hikari;
    }

    public void setHikari(Map<String, HikariDataSource> hikari) {
        this.hikari = hikari;
    }
}

這裡做了重點改動，原先是針對配置檔案中的欄位名來定義欄位的，這裡我直接用一個map來獲取配置檔案中配置的所有資料庫相關資訊。通過ConfigurationProperties來獲取配置檔案中集合這塊不屬於本文的講解，所以有需要的自己瞭解下。

DataSourceConfig

import com.zaxxer.hikari.HikariDataSource;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.transaction.PlatformTransactionManager;

import javax.annotation.Resource;
import javax.sql.DataSource;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

/**
 * @author Created by pangkunkun on 2018/8/28.
 */
@Configuration
public class DataSourceConfig {

    private static final Logger logger = LoggerFactory.getLogger(DataSourceConfig.class);

    @Autowired
    private DBProperties properties;

    private static final String KEY_MASTER = "master";

    @Bean(name = "dataSource")
    public DataSource dataSource() {
        //按照目標資料來源名稱和目標資料來源物件的對映存放在Map中
        Map<Object, Object> targetDataSources = new HashMap<>();
        //獲取配置檔案中的資料來源
        Map<String, HikariDataSource> hikaris = properties.getHikari();
        Set<String> keys = hikaris.keySet();
        HikariDataSource hikariDataSource = null;
        HikariDataSource masterDB = null;
        String poolName = "";
        for (String key : keys){
            hikariDataSource = hikaris.get(key);
            poolName = hikariDataSource.getPoolName();
            targetDataSources.put(hikariDataSource.getPoolName(),hikariDataSource);
            if (poolName.equals(KEY_MASTER)){
                masterDB = hikariDataSource;
            }
        }

        //採用是想AbstractRoutingDataSource的物件包裝多資料來源
        DynamicDataSource dataSource = new DynamicDataSource();
        dataSource.setTargetDataSources(targetDataSources);
        //設定預設的資料來源，當拿不到資料來源時，使用此配置
        if (null != masterDB){
            dataSource.setDefaultTargetDataSource(masterDB);
        }else {
            logger.error("Can't find master db, project will be exit");
            System.exit(0);
        }
        return dataSource;
    }

    @Bean
    public PlatformTransactionManager txManager() {
        return new DataSourceTransactionManager(dataSource());
    }
}

在上篇文章中這塊程式碼是將DBProperties中的幾個資料庫資訊一一放到map中，這裡通過DBProperties中的map獲取所有資料庫資訊放到資料來源的map中，通過poolname來區分資料庫的資訊。並且增加了master資料庫的判斷，如果識別不到master資料庫就會退出專案（這個功能看情況自己選擇是否需要）。

最後提一點，我打算最近抽時間把這塊做成一個公共元件，所有動態資訊通過配置檔案設定，以後就不用再動這塊程式碼了（希望自己能搞定吧^_^）。