使用Spring的AbstractRoutingDataSource實現(xiàn)多數(shù)據(jù)源切換示例
最近因為項目需要在做兩個項目間數(shù)據(jù)同步的需求，具體是項目1的數(shù)據(jù)通過消息隊列同步到項目2中，因為這個更新操作還涉及到更新多個庫的數(shù)據(jù)，所以就需要多數(shù)據(jù)源切換的操作。下面就講講在Spring中如何進行數(shù)據(jù)源切換。這里是使用AbstractRoutingDataSource類來完成具體的操作，AbstractRoutingDataSource是Spring2.0后增加的。
實現(xiàn)數(shù)據(jù)源切換的功能就是自定義一個類擴展AbstractRoutingDataSource抽象類，其實該相當于數(shù)據(jù)源DataSourcer的路由中介，可以實現(xiàn)在項目運行時根據(jù)相應(yīng)key值切換到對應(yīng)的數(shù)據(jù)源DataSource上。先看看AbstractRoutingDataSource的源碼：
public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean {
/* 只列出部分代碼 */
private Map<Object, Object> targetDataSources;
private Object defaultTargetDataSource;
private boolean lenientFallback = true;
private DataSourceLookup dataSourceLookup = new JndiDataSourceLookup();
private Map<Object, DataSource> 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并實現(xiàn)了InitializingBean，AbstractRoutingDataSource的getConnection()方法調(diào)用了determineTargetDataSource()的該方法，這里重點看determineTargetDataSource()方法代碼，方法里使用到了determineCurrentLookupKey()方法，它是AbstractRoutingDataSource類的抽象方法，也是實現(xiàn)數(shù)據(jù)源切換要擴展的方法，該方法的返回值就是項目中所要用的DataSource的key值，拿到該key后就可以在resolvedDataSource中取出對應(yīng)的DataSource，如果key找不到對應(yīng)的DataSource就使用默認的數(shù)據(jù)源。
自定義類擴展AbstractRoutingDataSource類時就是要重寫determineCurrentLookupKey()方法來實現(xiàn)數(shù)據(jù)源切換功能。下面是自定義的擴展AbstractRoutingDataSource類的實現(xiàn)：
/**
* 獲得數(shù)據(jù)源
*/
public class MultipleDataSource extends AbstractRoutingDataSource{
@Override
protected Object determineCurrentLookupKey() {
return DynamicDataSourceHolder.getRouteKey();
}
}
DynamicDataSourceHolder類如下，實現(xiàn)對數(shù)據(jù)源的操作功能：
/**
* 數(shù)據(jù)源操作類
*/
public class DynamicDataSourceHolder {
private static ThreadLocal<String> routeKey = new ThreadLocal<String>();
/**
* 獲取當前線程的數(shù)據(jù)源路由的key
*/
public static String getRouteKey()
{
String key = routeKey.get();
return key;
}
/**
* 綁定當前線程數(shù)據(jù)源路由的key
* 使用完成后必須調(diào)用removeRouteKey()方法刪除
*/
public static void setRouteKey(String key)
{
routeKey.set(key);
}
/**
* 刪除與當前線程綁定的數(shù)據(jù)源路由的key
*/
public static void removeRouteKey()
{
routeKey.remove();
}
}
下面在xml文件中配置多個數(shù)據(jù)源：
<!-- 數(shù)據(jù)源 -->
<bean id="dataSource1" class="org.apache.commons.dbcp.BasicDataSource">
<property name="driverClassName" value="net.sourceforge.jtds.jdbc.Driver">
</property>
<property name="url" value="jdbc:jtds:sqlserver://127.0.0.1;databaseName=test">
</property>
<property name="username" value="***"></property>
<property name="password" value="***"></property>
</bean>
<bean id="dataSource2" class="org.apache.commons.dbcp.BasicDataSource">
<property name="driverClassName" value="net.sourceforge.jtds.jdbc.Driver">
</property>
<property name="url" value="jdbc:jtds:sqlserver://127.0.0.2:1433;databaseName=test">
</property>
<property name="username" value="***"></property>
<property name="password" value="***"></property>
</bean>
<!-- 配置多數(shù)據(jù)源映射 -->
<bean id="multipleDataSource" class="MultipleDataSource" >
<property name="targetDataSources">
<map key-type="java.lang.String">
<entry value-ref="dataSource1" key="dataSource1"></entry>
<entry value-ref="dataSource2" key="dataSource2"></entry>
</map>
</property>
<!-- 默認數(shù)據(jù)源 -->
<property name="defaultTargetDataSource" ref="dataSource1" >
</property>
</bean>
到這里基本的配置就完成了，下面只要在需要切換數(shù)據(jù)源的地方調(diào)用方法就行了，一般是在dao層操作數(shù)據(jù)庫前進行切換的，只需在數(shù)據(jù)庫操作前加上如下代碼即可：
DynamicDataSourceHolder.setRouteKey("dataSource2");
上面介紹的是在dao層當需要切換數(shù)據(jù)源時手動加上切換數(shù)據(jù)源的代碼，也可以使用AOP的方式，把配置的數(shù)據(jù)源類型都設(shè)置成注解標簽，在dao層中需要切換數(shù)據(jù)源操作的方法或類上寫上注解標簽，這樣實現(xiàn)起來可操作性也更強。
@DataSourceKey("dataSource1")
public interface TestEntityMapper extends MSSQLMapper<TestEntity> {
public void insertTest(TestEntity testEntity);
}
DataSourceKey注解代碼如下：
@Target({ElementType.TYPE,ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DataSourceKey {
String value() default "";
}
注解配置完后就要寫一個實現(xiàn)數(shù)據(jù)源切換的類，如下：
public class MultipleDataSourceExchange {
/**
* 攔截目標方法，獲取由@DataSource指定的數(shù)據(jù)源標識，設(shè)置到線程存儲中以便切換數(shù)據(jù)源
*/
public void beforeDaoMethod(JoinPoint point) throws Exception {
Class<?> target = point.getTarget().getClass();
MethodSignature signature = (MethodSignature) point.getSignature();
// 默認使用目標類型的注解，如果沒有則使用其實現(xiàn)接口的注解類
for (Class<?> cls : target.getInterfaces()) {
resetDataSource(cls, signature.getMethod());
}
resetDataSource(target, signature.getMethod());
}
/**
* 提取目標對象方法注解和類注解中的數(shù)據(jù)源標識
*/
private void resetDataSource(Class<?> cls, Method method) {
try {
Class<?>[] types = method.getParameterTypes();
// 默認使用類注解
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配置文件上添加配置了（只列出部分配置）：
<bean id="multipleDataSourceExchange" class="MultipleDataSourceExchange "/>
<bean id="txManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
<property name="dataSource" ref="multipleDataSource" />
</bean>
<tx:advice id="txAdvice" transaction-manager="txManager">
<tx:attributes>
<tx:method name="insert*" propagation="NESTED" rollback-for="Exception"/>
<tx:method name="add*" propagation="NESTED" rollback-for="Exception"/>
...
</tx:attributes>
</tx:advice>
<aop:config>
<aop:pointcut id="service" expression="execution(* com.datasource..*.service.*.*(..))"/>
<!-- 注意切換數(shù)據(jù)源操作要比持久層代碼先執(zhí)行 -->
<aop:advisor advice-ref="multipleDataSourceExchange" pointcut-ref="service" order="1"/>
<aop:advisor advice-ref="txAdvice" pointcut-ref="service" order="2"/>
</aop:config>
到此就完成使用AOP的方式實現(xiàn)多數(shù)據(jù)源的動態(tài)切換了。
相關(guān)文章
中斷（Interrupt）一個線程意味著在該線程完成任務(wù)之前停止其正在進行的一切，有效地中止其當前的操作。這篇文章主要介紹了java 中斷線程的幾種方式 interrupt(),需要的朋友可以參考下2021-11-11
MyBatis 是支持定制化 SQL、存儲過程以及高級映射的優(yōu)秀的持久層框架。MyBatis 避免了幾乎所有的 JDBC 代碼和手動設(shè)置參數(shù)以及獲取結(jié)果集2017-05-05
這篇文章主要給大家介紹了關(guān)于JAVA中日期時間類用法的相關(guān)資料，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-09-09
Spring+SpringMVC+JDBC實現(xiàn)登錄的示例(附源碼)
這篇文章主要介紹了Spring+SpringMVC+JDBC實現(xiàn)登錄的示例代碼，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-05-05
這篇文章主要介紹了Idea 配置國內(nèi) Maven 源的教程,本文通過圖文并茂的形式給大家介紹的非常詳細,對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2023-11-11
Mybatis使用foreach批量更新數(shù)據(jù)報無效字符錯誤問題
這篇文章主要介紹了Mybatis使用foreach批量更新數(shù)據(jù)報無效字符錯誤問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-08-08
springboot1.X和2.X中如何解決Bean名字相同時覆蓋
這篇文章主要介紹了springboot1.X和2.X中如何解決Bean名字相同時覆蓋，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-03-03
Java內(nèi)存結(jié)構(gòu)和數(shù)據(jù)類型
本文重點給大家介紹java內(nèi)存結(jié)構(gòu)和數(shù)據(jù)類型知識，非常不錯，具有參考借鑒價值，需要的朋友參考下2016-12-12
ByteArrayOutputStream與InputStream互相轉(zhuǎn)換方式
這篇文章主要介紹了ByteArrayOutputStream與InputStream互相轉(zhuǎn)換方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2021-12-12
這篇文章主要介紹了spring MVC實踐需要注意的地方，幫助大家更好的理解和學習使用spring MVC，感興趣的朋友可以了解下2021-03-03