引用兩個(gè)或多個(gè)數(shù)據(jù)庫里的數(shù)據(jù)，項(xiàng)目數(shù)據(jù)庫的配置方法

hqman 發(fā)布于2019-08-16 15:52 / 610人閱讀

摘要：解決辦法可以在使用完切換數(shù)據(jù)庫的方法拿到需要的數(shù)據(jù)后，再次調(diào)用方法切換回接下來需要用到的數(shù)據(jù)庫。

1、首先配置兩個(gè)數(shù)據(jù)源（數(shù)據(jù)庫）以及一個(gè)動(dòng)態(tài)數(shù)據(jù)庫：

2、sqlSessionFactory引用的數(shù)據(jù)庫是動(dòng)態(tài)數(shù)據(jù)庫dynamicDataSource：

4、事務(wù)管理這里要管理的也是上面配置的dynamicDataSource動(dòng)態(tài)數(shù)據(jù)庫：

5、寫兩個(gè)數(shù)據(jù)源配置類DataSourceContextHolder.java和DynamicDataSource.java來配置數(shù)據(jù)源，利用ThreadLocal解決線程安全問題。

DataSourceContextHolder 類：

package com.datasource;

public class DataSourceContextHolder {
    private static final ThreadLocal contextHolder = new ThreadLocal();
    public static void setCustomerType(String customerType) {
        contextHolder.set(customerType);
    }
    public static String getCustomerType() {
        return contextHolder.get();
    }
    public static void clearCustomerType() {
        contextHolder.remove();
    }
}

DynamicDataSource 類繼承 AbstractRoutingDataSource，并實(shí)現(xiàn)determineCurrentLookupKey方法：

package com.datasource;

import java.sql.SQLFeatureNotSupportedException;
import java.util.logging.Logger;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

public class DynamicDataSource extends AbstractRoutingDataSource {

    @Override
    public Logger getParentLogger() throws SQLFeatureNotSupportedException {
        // TODO 自動(dòng)生成的方法存根
        return null;
    }

    @Override
    protected Object determineCurrentLookupKey() {
        // TODO 自動(dòng)生成的方法存根
        return DataSourceContextHolder.getCustomerType();
    }

}

6、最后就可以在需要切換數(shù)據(jù)庫的地方使用以下方法來切換數(shù)據(jù)庫了，要切換的數(shù)據(jù)庫名字即之前在配置動(dòng)態(tài)數(shù)據(jù)庫時(shí)給引用的數(shù)據(jù)庫賦的名字：

DataSourceContextHolder.setCustomerType("要切換的數(shù)據(jù)庫名字");

7、下面附上我的目錄結(jié)構(gòu)：

注：每次使用完切換數(shù)據(jù)庫的方法后，系統(tǒng)會(huì)自動(dòng)切換回默認(rèn)數(shù)據(jù)庫，不過這之間存在一點(diǎn)小延遲，會(huì)出現(xiàn)在調(diào)用完切換數(shù)據(jù)庫的方法后，立刻去跳轉(zhuǎn)到引用另外一個(gè)數(shù)據(jù)庫數(shù)據(jù)的頁面，系統(tǒng)還是使用著切換后的數(shù)據(jù)庫。

解決辦法：可以在使用完切換數(shù)據(jù)庫的方法拿到需要的數(shù)據(jù)后，再次調(diào)用 DataSourceContextHolder.setCustomerType() 方法切換回接下來需要用到的數(shù)據(jù)庫。