Spring中使用AbstractRoutingDataSource如何切換多數(shù)據(jù)源？針對(duì)這個(gè)問(wèn)題，這篇文章詳細(xì)介紹了相對(duì)應(yīng)的分析和解答，希望可以幫助更多想解決這個(gè)問(wèn)題的小伙伴找到更簡(jiǎn)單易行的方法。
創(chuàng)新互聯(lián)是一家專業(yè)提供岳陽(yáng)縣企業(yè)網(wǎng)站建設(shè),專注與網(wǎng)站制作、網(wǎng)站設(shè)計(jì)、H5建站、小程序制作等業(yè)務(wù)。10年已為岳陽(yáng)縣眾多企業(yè)、政府機(jī)構(gòu)等服務(wù)。創(chuàng)新互聯(lián)專業(yè)網(wǎng)站設(shè)計(jì)公司優(yōu)惠進(jìn)行中。
最近因?yàn)轫?xiàng)目需要在做兩個(gè)項(xiàng)目間數(shù)據(jù)同步的需求，具體是項(xiàng)目1的數(shù)據(jù)通過(guò)消息隊(duì)列同步到項(xiàng)目2中，因?yàn)檫@個(gè)更新操作還涉及到更新多個(gè)庫(kù)的數(shù)據(jù)，所以就需要多數(shù)據(jù)源切換的操作。下面就講講在Spring中如何進(jìn)行數(shù)據(jù)源切換。這里是使用AbstractRoutingDataSource類來(lái)完成具體的操作，AbstractRoutingDataSource是Spring2.0后增加的。
實(shí)現(xiàn)數(shù)據(jù)源切換的功能就是自定義一個(gè)類擴(kuò)展AbstractRoutingDataSource抽象類，其實(shí)該相當(dāng)于數(shù)據(jù)源DataSourcer的路由中介，可以實(shí)現(xiàn)在項(xiàng)目運(yùn)行時(shí)根據(jù)相應(yīng)key值切換到對(duì)應(yīng)的數(shù)據(jù)源DataSource上。先看看AbstractRoutingDataSource的源碼：
重慶分公司
重慶分公司
public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean {
/* 只列出部分代碼 */
private Map targetDataSources;
private Object defaultTargetDataSource;
private boolean lenientFallback = true;
private DataSourceLookup dataSourceLookup = new JndiDataSourceLookup();
private Map resolvedDataSources;
private DataSource resolvedDefaultDataSource;
@Override
public Connection getConnection() throws SQLException {
return determineTargetDataSource().getConnection();
}
@Override
public Connection getConnection(String username, String password) throws SQLException {
return determineTargetDataSource().getConnection(username, password);
}
protected DataSource determineTargetDataSource() {
Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");
Object lookupKey = determineCurrentLookupKey();
DataSource dataSource = this.resolvedDataSources.get(lookupKey);
if (dataSource == null && (this.lenientFallback || lookupKey == null)) {
dataSource = this.resolvedDefaultDataSource;
}
if (dataSource == null) {
throw new IllegalStateException("Cannot determine target DataSource for lookup key [" + lookupKey + "]");
}
return dataSource;
}
protected abstract Object determineCurrentLookupKey();
} 從源碼可以看出AbstractRoutingDataSource繼承了AbstractDataSource并實(shí)現(xiàn)了InitializingBean，AbstractRoutingDataSource的getConnection()方法調(diào)用了determineTargetDataSource()的該方法，這里重點(diǎn)看determineTargetDataSource()方法代碼，方法里使用到了determineCurrentLookupKey()方法，它是AbstractRoutingDataSource類的抽象方法，也是實(shí)現(xiàn)數(shù)據(jù)源切換要擴(kuò)展的方法，該方法的返回值就是項(xiàng)目中所要用的DataSource的key值，拿到該key后就可以在resolvedDataSource中取出對(duì)應(yīng)的DataSource，如果key找不到對(duì)應(yīng)的DataSource就使用默認(rèn)的數(shù)據(jù)源。
自定義類擴(kuò)展AbstractRoutingDataSource類時(shí)就是要重寫determineCurrentLookupKey()方法來(lái)實(shí)現(xiàn)數(shù)據(jù)源切換功能。下面是自定義的擴(kuò)展AbstractRoutingDataSource類的實(shí)現(xiàn)：
/**
* 獲得數(shù)據(jù)源
*/
public class MultipleDataSource extends AbstractRoutingDataSource{
@Override
protected Object determineCurrentLookupKey() {
return DynamicDataSourceHolder.getRouteKey();
}
}DynamicDataSourceHolder類如下，實(shí)現(xiàn)對(duì)數(shù)據(jù)源的操作功能：
/**
* 數(shù)據(jù)源操作類
*/
public class DynamicDataSourceHolder {
private static ThreadLocal routeKey = new ThreadLocal();
/**
* 獲取當(dāng)前線程的數(shù)據(jù)源路由的key
*/
public static String getRouteKey()
{
String key = routeKey.get();
return key;
}
/**
* 綁定當(dāng)前線程數(shù)據(jù)源路由的key
* 使用完成后必須調(diào)用removeRouteKey()方法刪除
*/
public static void setRouteKey(String key)
{
routeKey.set(key);
}
/**
* 刪除與當(dāng)前線程綁定的數(shù)據(jù)源路由的key
*/
public static void removeRouteKey()
{
routeKey.remove();
}
} 下面在xml文件中配置多個(gè)數(shù)據(jù)源：
到這里基本的配置就完成了，下面只要在需要切換數(shù)據(jù)源的地方調(diào)用方法就行了，一般是在dao層操作數(shù)據(jù)庫(kù)前進(jìn)行切換的，只需在數(shù)據(jù)庫(kù)操作前加上如下代碼即可：
DynamicDataSourceHolder.setRouteKey("dataSource2");上面介紹的是在dao層當(dāng)需要切換數(shù)據(jù)源時(shí)手動(dòng)加上切換數(shù)據(jù)源的代碼，也可以使用AOP的方式，把配置的數(shù)據(jù)源類型都設(shè)置成注解標(biāo)簽，在dao層中需要切換數(shù)據(jù)源操作的方法或類上寫上注解標(biāo)簽，這樣實(shí)現(xiàn)起來(lái)可操作性也更強(qiáng)。
@DataSourceKey("dataSource1")
public interface TestEntityMapper extends MSSQLMapper {
public void insertTest(TestEntity testEntity);
} DataSourceKey注解代碼如下：
@Target({ElementType.TYPE,ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DataSourceKey {
String value() default "";
}注解配置完后就要寫一個(gè)實(shí)現(xiàn)數(shù)據(jù)源切換的類，如下：
public class MultipleDataSourceExchange {
/**
* 攔截目標(biāo)方法，獲取由@DataSource指定的數(shù)據(jù)源標(biāo)識(shí)，設(shè)置到線程存儲(chǔ)中以便切換數(shù)據(jù)源
*/
public void beforeDaoMethod(JoinPoint point) throws Exception {
Class<?> target = point.getTarget().getClass();
MethodSignature signature = (MethodSignature) point.getSignature();
// 默認(rèn)使用目標(biāo)類型的注解，如果沒(méi)有則使用其實(shí)現(xiàn)接口的注解類
for (Class<?> cls : target.getInterfaces()) {
resetDataSource(cls, signature.getMethod());
}
resetDataSource(target, signature.getMethod());
}
/**
* 提取目標(biāo)對(duì)象方法注解和類注解中的數(shù)據(jù)源標(biāo)識(shí)
*/
private void resetDataSource(Class<?> cls, Method method) {
try {
Class<?>[] types = method.getParameterTypes();
// 默認(rèn)使用類注解
if (cls.isAnnotationPresent(DataSourceKey.class)) {
DataSourceKey source = cls.getAnnotation(DataSourceKey.class);
DynamicDataSourceHolder.setRouteKey(source.value());
}
// 方法注解可以覆蓋類注解
Method m = cls.getMethod(method.getName(), types);
if (m != null && m.isAnnotationPresent(DataSourceKey.class)) {
DataSourceKey source = m.getAnnotation(DataSourceKey.class);
DynamicDataSourceHolder.setRouteKey(source.value());
}
} catch (Exception e) {
System.out.println(cls + ":" + e.getMessage());
}
}
}代碼寫完后就要在xml配置文件上添加配置了（只列出部分配置）：
