摘要: 案例基於Spring+mybatis。 設計的目的：解決不確定（個數和種類）資料來源的切換問題

2017-06-06 11:31:57補充：近日，在本文的基礎之上，擴充套件了下，使用atomikos來管理事務，保證多資料來源操作時，事務一致性。（https://my.oschina.net/simpleton/blog/916108）

另外，感謝朋友對本文的關注和對博主的支援，最近有很多朋友聯絡我希望深入探討下本文涉及內容，不過由於近日太忙，沒有及時回覆大家，請見諒。

近日，博主有個業務需求，就是根據資料庫儲存的不同資料來源資訊，動態建立資料來源並實現業務不同而轉到不同的資料來源上處理。

資料庫儲存起來的資料來源資訊是不確定的，可以刪除和新增，這些是業務前提。

在網上找了下相關資料，對於使用Spring配置，直接配置多套資料來源，使用AOP動態切換的方式居多，這種方式博主以前也使用過，很強大。不過有個前提就是多個數據源的資訊是預先就確定的。那麼對於不確定資料來源資訊的業務需求，就只有使用程式碼動態實現資料來源初始化和銷燬操作了。

好了，有了這些思路，可以開始準備寫程式碼了。

1、建立一個執行緒上下文物件（使用ThreadLocal，保證執行緒安全）。上下文物件中主要維護了資料來源的KEY和資料來源的地址等資訊，當KEY對應的資料來源找不到時，根據資料來源地址、驅動和使用者名稱等建立 一個數據源，這裡也是業務中需要解決的一個核心問題（JAVA動態建立資料來源）。
 

/**
* Copyright (c) 2015 - 2016 eay Inc.
* All rights reserved.
*/
package com.eya.pubservice.datasource;

import java.util.HashMap;
import java.util.Map;

/**
* 當前正在使用的資料來源資訊的執行緒上線文
* @create ll
* @createDate 2017年3月27日 下午2:37:07
* @update
* @updateDate
*/
public class DBContextHolder {
/* 資料來源的KEY

/
public static final String DATASOURCE_KEY = “DATASOURCE_KEY”;
/* 資料來源的URL /
public static final String DATASOURCE_URL = “DATASOURCE_URL”;
/* 資料來源的驅動 /
public static final String DATASOURCE_DRIVER = “DATASOURCE_DRIVER”;
/* 資料來源的使用者名稱 /
public static final String DATASOURCE_USERNAME = “DATASOURCE_USERNAME”;
/* 資料來源的密碼 /
public static final String DATASOURCE_PASSWORD = “DATASOURCE_PASSWORD”;

private static final ThreadLocal<Map<String, Object>> contextHolder = new ThreadLocal<Map<String, Object>>();

public static void setDBType(Map<String, Object> dataSourceConfigMap) {
    contextHolder.set(dataSourceConfigMap);
}

public static Map<String, Object> getDBType() {
    Map<String, Object> dataSourceConfigMap = contextHolder.get();
    if (dataSourceConfigMap == null) {
        dataSourceConfigMap = new HashMap<String, Object>();
    }
    return dataSourceConfigMap;
}

public static void clearDBType() {
    contextHolder.remove();
}

}

2、建立一個AbstractRoutingDataSource的子類，實現其determineCurrentLookupKey方法，用於決定使用哪一個資料來源。說明一下，這裡實現了ApplicationContextAware介面，用於在Spring載入完成後，注入Spring上下文物件，用於獲取Bean。

/**
* Copyright (c) 2015 - 2016 eya Inc.
* All rights reserved.
*/
package com.eya.pubservice.datasource;

import java.util.Map;

import javax.sql.DataSource;

import org.apache.commons.collections.MapUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

/**
* 動態資料來源父類
* @create ll
* @createDate 2017年3月27日 下午2:38:05
* @update
* @updateDate
*/
public abstract class AbstractDynamicDataSource extends AbstractRoutingDataSource
implements
ApplicationContextAware {

/** 日誌 */
protected Logger logger = LoggerFactory.getLogger(getClass());
/** 預設的資料來源KEY */
protected static final String DEFAULT_DATASOURCE_KEY = "defaultDataSource";

/** 資料來源KEY-VALUE鍵值對 */
public Map<Object, Object> targetDataSources;

/** spring容器上下文 */
private static ApplicationContext ctx;

public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
    ctx = applicationContext;
}

public static ApplicationContext getApplicationContext() {
    return ctx;
}

public static Object getBean(String name) {
    return ctx.getBean(name);
}

/**
 * @param targetDataSources the targetDataSources to set
 */
public void setTargetDataSources(Map<Object, Object> targetDataSources) {
    this.targetDataSources = targetDataSources;
    super.setTargetDataSources(targetDataSources);
    // afterPropertiesSet()方法呼叫時用來將targetDataSources的屬性寫入resolvedDataSources中的
    super.afterPropertiesSet();
}

/**
 * 建立資料來源
 * @param driverClassName 資料庫驅動名稱
 * @param url 連線地址
 * @param username 使用者名稱
 * @param password 密碼
 * @return 資料來源{@link T}
 * @Author : ll. create at 2017年3月27日 下午2:44:34
 */
public abstract T createDataSource(String driverClassName, String url, String username,
                                   String password);

/**
 * 設定系統當前使用的資料來源
 * <p>資料來源為空或者為0時，自動切換至預設資料來源，即在配置檔案中定義的預設資料來源
 * @see org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource#determineCurrentLookupKey()
 */
@Override
protected Object determineCurrentLookupKey() {
    logger.info("【設定系統當前使用的資料來源】");
    Map<String, Object> configMap = DBContextHolder.getDBType();
    logger.info("【當前資料來源配置為：{}】", configMap);
    if (MapUtils.isEmpty(configMap)) {
        // 使用預設資料來源
        return DEFAULT_DATASOURCE_KEY;
    }
    // 判斷資料來源是否需要初始化
    this.verifyAndInitDataSource();
    logger.info("【切換至資料來源：{}】", configMap);
    return configMap.get(DBContextHolder.DATASOURCE_KEY);
}

/**
 * 判斷資料來源是否需要初始化
 * @Author : ll. create at 2017年3月27日 下午3:57:43
 */
private void verifyAndInitDataSource() {
    Map<String, Object> configMap = DBContextHolder.getDBType();
    Object obj = this.targetDataSources.get(configMap.get(DBContextHolder.DATASOURCE_KEY));
    if (obj != null) {
        return;
    }
    logger.info("【初始化資料來源】");
    T datasource = this.createDataSource(configMap.get(DBContextHolder.DATASOURCE_DRIVER)
        .toString(), configMap.get(DBContextHolder.DATASOURCE_URL).toString(),
        configMap.get(DBContextHolder.DATASOURCE_USERNAME).toString(),
        configMap.get(DBContextHolder.DATASOURCE_PASSWORD).toString());
    this.addTargetDataSource(configMap.get(DBContextHolder.DATASOURCE_KEY).toString(),
        datasource);
}

/**
 * 往資料來源key-value鍵值對集合新增新的資料來源
 * @param key 新的資料來源鍵
 * @param dataSource 新的資料來源
 * @Author : ll. create at 2017年3月27日 下午2:56:49
 */
private void addTargetDataSource(String key, T dataSource) {
    this.targetDataSources.put(key, dataSource);
    super.setTargetDataSources(this.targetDataSources);
    // afterPropertiesSet()方法呼叫時用來將targetDataSources的屬性寫入resolvedDataSources中的
    super.afterPropertiesSet();
}

}

3、編寫AbstractDynamicDataSource的實現類，使用com.alibaba.druid.pool.DruidDataSource資料來源。主要實現建立資料來源的方法（createDataSource）

/**
* Copyright (c) 2015 - 2016 eya Inc.
* All rights reserved.
*/
package com.eya.pubservice.datasource;

import java.sql.SQLException;
import java.util.List;

import org.apache.commons.lang3.StringUtils;

import com.alibaba.druid.filter.Filter;
import com.alibaba.druid.pool.DruidDataSource;

/**
* Druid資料來源
*

摘抄自http://www.68idc.cn/help/buildlang/java/20160606618505.html
* @create ll
* @createDate 2017年3月27日 下午2:40:17
* @update
* @updateDate
*/
public class DruidDynamicDataSource extends AbstractDynamicDataSource {

private boolean testWhileIdle = true;
private boolean testOnBorrow = false;
private boolean testOnReturn = false;

// 是否開啟連線洩露自動檢測
private boolean removeAbandoned = false;
// 連線長時間沒有使用，被認為發生洩露時長
private long removeAbandonedTimeoutMillis = 300 * 1000;
// 發生洩露時是否需要輸出 log，建議在開啟連線洩露檢測時開啟，方便排錯
private boolean logAbandoned = false;

// 只要maxPoolPreparedStatementPerConnectionSize>0,poolPreparedStatements就會被自動設定為true，使用oracle時可以設定此值。
//    private int maxPoolPreparedStatementPerConnectionSize = -1;

// 配置監控統計攔截的filters
private String filters; // 監控統計："stat" 防SQL注入："wall" 組合使用： "stat,wall"
private List<Filter> filterList;

/*
 * 建立資料來源
 * @see com.cdelabcare.pubservice.datasource.IDynamicDataSource#createDataSource(java.lang.String, java.lang.String, java.lang.String, java.lang.String)
 */
@Override
public DruidDataSource createDataSource(String driverClassName, String url, String username,
                                        String password) {
    DruidDataSource parent = (DruidDataSource) super.getApplicationContext().getBean(
        DEFAULT_DATASOURCE_KEY);
    DruidDataSource ds = new DruidDataSource();
    ds.setUrl(url);
    ds.setUsername(username);
    ds.setPassword(password);
    ds.setDriverClassName(driverClassName);
    ds.setInitialSize(parent.getInitialSize());
    ds.setMinIdle(parent.getMinIdle());
    ds.setMaxActive(parent.getMaxActive());
    ds.setMaxWait(parent.getMaxWait());
    ds.setTimeBetweenConnectErrorMillis(parent.getTimeBetweenConnectErrorMillis());
    ds.setTimeBetweenEvictionRunsMillis(parent.getTimeBetweenEvictionRunsMillis());
    ds.setMinEvictableIdleTimeMillis(parent.getMinEvictableIdleTimeMillis());

    ds.setValidationQuery(parent.getValidationQuery());
    ds.setTestWhileIdle(testWhileIdle);
    ds.setTestOnBorrow(testOnBorrow);
    ds.setTestOnReturn(testOnReturn);

    ds.setRemoveAbandoned(removeAbandoned);
    ds.setRemoveAbandonedTimeoutMillis(removeAbandonedTimeoutMillis);
    ds.setLogAbandoned(logAbandoned);

    // 只要maxPoolPreparedStatementPerConnectionSize>0,poolPreparedStatements就會被自動設定為true，參照druid的原始碼
    ds.setMaxPoolPreparedStatementPerConnectionSize(parent
        .getMaxPoolPreparedStatementPerConnectionSize());

    if (StringUtils.isNotBlank(filters))
        try {
            ds.setFilters(filters);
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }

    addFilterList(ds);
    return ds;
}

private void addFilterList(DruidDataSource ds) {
    if (filterList != null) {
        List<Filter> targetList = ds.getProxyFilters();
        for (Filter add : filterList) {
            boolean found = false;
            for (Filter target : targetList) {
                if (add.getClass().equals(target.getClass())) {
                    found = true;
                    break;
                }
            }
            if (!found)
                targetList.add(add);
        }
    }
}

}

4、使用Spring配置預設資料來源。系統執行肯定有一套預設的資料來源（否則動態建立的資料來源資訊從哪裡來呢？上面提到的，動態建立的資料來源資訊是存放在資料庫中的）。這裡我貼出完整的Spring配置。