Java注解實現動態(tài)數據源切換的實例代碼
當一個項目中有多個數據源（也可以是主從庫）的時候，我們可以利用注解在mapper接口上標注數據源，從而來實現多個數據源在運行時的動態(tài)切換。
實現原理
在Spring 2.0.1中引入了AbstractRoutingDataSource, 該類充當了DataSource的路由中介, 能有在運行時, 根據某種key值來動態(tài)切換到真正的DataSource上。
看下AbstractRoutingDataSource：
public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean
AbstractRoutingDataSource繼承了AbstractDataSource，獲取數據源部分：
/**
* Retrieve the current target DataSource. Determines the
* {@link #determineCurrentLookupKey() current lookup key}, performs
* a lookup in the {@link #setTargetDataSources targetDataSources} map,
* falls back to the specified
* {@link #setDefaultTargetDataSource default target DataSource} if necessary.
* @see #determineCurrentLookupKey()
*/
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;
}
抽象方法 determineCurrentLookupKey() 返回DataSource的key值，然后根據這個key從resolvedDataSources這個map里取出對應的DataSource，如果找不到，則用默認的resolvedDefaultDataSource。
我們要做的就是實現抽象方法 determineCurrentLookupKey() 返回數據源的key值。
使用方法
定義注解：
/**
* Created by huangyangquan on 2016/11/30.
*/
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface DataSource {
DataSourceType value();
}
注解為數據源的名稱，可定義一個枚舉類表示：
/**
* Created by huangyangquan on 2016/11/30.
*/
public enum DataSourceType {
MASTER,
SLAVE
}
注解定義好了，我們利用Spring的AOP根據注解內容對數據源進行選擇，這里需要利用上面提到的 AbstractRoutingDataSource 類，該類是能夠實現數據源切換的關鍵所在。
定義類DynamicDataSource繼承AbstractRoutingDataSource，并實現 determineCurrentLookupKey() ，返回數據源的key值。
/**
* Created by huangyangquan on 2016/11/30.
*/
public class DynamicDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
return DynamicDataSourceHolder.getDataSourceType();
}
}
DynamicDataSourceHolder 是我們管理DataSource的類，將一次數據庫操作的數據源名稱保存在DynamicDataSourceHolder中，以供后面的操作在此context中取數據源key，其中DataSourceType使用了線程本地變量來保證線程安全。
/**
* Created by huangyangquan on 2016/11/30.
*/
public class DynamicDataSourceHolder {
// 線程本地環(huán)境
private static final ThreadLocal<DataSourceType> contextHolder = new ThreadLocal<DataSourceType>();
// 設置數據源類型
public static void setDataSourceType(DataSourceType dataSourceType) {
Assert.notNull(dataSourceType, "DataSourceType cannot be null");
contextHolder.set(dataSourceType);
}
// 獲取數據源類型
public static DataSourceType getDataSourceType() {
return (DataSourceType) contextHolder.get();
}
// 清除數據源類型
public static void clearDataSourceType() {
contextHolder.remove();
}
}
我們在Spring的配置文件中配置數據源key值得對應關系：
<bean id="spyGhotelDataSource" class="com.aheizi.config.DynamicDataSource">
<property name="targetDataSources">
<map key-type="java.lang.String">
<entry key="MASTER" value-ref="TEST-MASTER-DB"></entry>
<entry key="SLAVE" value-ref="TEST-SLAVE-DB"></entry>
</map>
</property>
<property name="defaultTargetDataSource" ref="TEST-MASTER-DB">
</property>
</bean>
設置targetDataSources和defaultTargetDataSource。 TEST-MASTER-DB 和 TEST-SLAVE-DB 表示主庫的從庫，是我們的兩個數據源。
接下來配置AOP切面：
<aop:aspectj-autoproxy proxy-target-class="false" />
<bean id="manyDataSourceAspect" class="com.aheizi.config.DataSourceAspect" />
<aop:config>
<aop:aspect id="dataSourceCut" ref="manyDataSourceAspect">
<aop:pointcut expression="execution(* com.aheizi.dao.*.*(..))"
id="dataSourceCutPoint" /><!-- 配置切點 -->
<aop:before pointcut-ref="dataSourceCutPoint" method="before" />
</aop:aspect>
</aop:config>
以下是切面中before執(zhí)行的DataSourceAspect的實現，主要實現的功能是獲取方法上的注解，根據注解名稱將值設置到DynamicDataSourceHolder中，這樣在執(zhí)行查詢的時候， determineCurrentLookupKey() 返回數據源的key值就是我們希望的那個數據源了。
/**
* Created by huangyangquan on 2016/11/30.
*/
public class DataSourceAspect {
private static final Logger LOG = LoggerFactory.getLogger(DataSourceAspect.class);
public void before(JoinPoint point){
Object target = point.getTarget();
String method = point.getSignature().getName();
Class<?>[] classz = target.getClass().getInterfaces();
Class<?>[] parameterTypes = ((MethodSignature) point.getSignature()).getMethod().getParameterTypes();
try {
Method m = classz[0].getMethod(method, parameterTypes);
if (m != null && m.isAnnotationPresent(DataSource.class)) {
// 訪問mapper中的注解
DataSource data = m.getAnnotation(DataSource.class);
switch (data.value()) {
case MASTER:
DynamicDataSourceHolder.setDataSourceType(DataSourceType.MASTER);
LOG.info("using dataSource:{}", DataSourceType.MASTER);
break;
case SLAVE:
DynamicDataSourceHolder.setDataSourceType(DataSourceType.SLAVE);
LOG.info("using dataSource:{}", DataSourceType.SLAVE);
break;
}
}
} catch (Exception e) {
LOG.error("dataSource annotation error:{}", e.getMessage());
// 若出現異常，手動設為主庫
DynamicDataSourceHolder.setDataSourceType(DataSourceType.MASTER);
}
}
}
這樣我們就實現了一個動態(tài)數據源切換的功能。
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。
相關文章
這篇文章主要介紹了如何使用Mockito調用靜態(tài)方法和void方法的操作，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2021-07-07
這篇文章主要介紹了Java中的接口及其應用場景解讀，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-05-05
這篇文章主要為大家詳細介紹了Java控制臺實現猜拳游戲小游戲，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-11-11
這篇文章主要幫助大家輕松掌握Java適配器模式,具有一定的參考價值，感興趣的小伙伴們可以參考一下2016-09-09
這篇文章主要介紹了Java中初始化塊詳解及實例代碼的相關資料,在Java中,有兩種初始化塊:靜態(tài)初始化塊和非靜態(tài)初始化塊，需要的朋友可以參考下2017-03-03
這篇文章主要介紹了Java如何通過反射獲取對象的屬性和值問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-02-02
關于springBoot yml文件的list讀取問題總結(親測)
這篇文章主要介紹了關于springBoot yml文件的list讀取問題總結，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2021-12-12
JWT作為一個開放的標準（ RFC 7519 ），定義了一種簡潔的，自包含的方法用于通信雙方之間以Json對象的形式安全的傳遞信息。接下來通過本文給大家介紹springSecurity+jwt實現互踢功能,需要的朋友可以參考下2021-11-11
在本篇文章里小編給各位分享了一篇關于java中abstract修改類的實例方法，有需要的朋友們可以學習下。2020-12-12
java substring(a)與substring(a,b)的使用說明
這篇文章主要介紹了java substring(a)與substring(a,b)的使用說明，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-10-10