免费观看又色又爽又黄的小说免费_美女福利视频国产片_亚洲欧美精品_美国一级大黄大色毛片

Java注解實現動態數據源切換的實例代碼

當一個項目中有多個數據源（也可以是主從庫）的時候，我們可以利用注解在mapper接口上標注數據源，從而來實現多個數據源在運行時的動態切換。

網站建設哪家好，找創新互聯建站！專注于網頁設計、網站建設、微信開發、微信平臺小程序開發、集團企業網站建設等服務項目。為回饋新老客戶創新互聯還提供了鐵山免費建站歡迎大家使用！

實現原理

在Spring 2.0.1中引入了AbstractRoutingDataSource, 該類充當了DataSource的路由中介, 能有在運行時, 根據某種key值來動態切換到真正的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 {

  // 線程本地環境
  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執行的DataSourceAspect的實現，主要實現的功能是獲取方法上的注解，根據注解名稱將值設置到DynamicDataSourceHolder中，這樣在執行查詢的時候， 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);
    }
  }

}

這樣我們就實現了一個動態數據源切換的功能。

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持創新互聯。

分享標題：Java注解實現動態數據源切換的實例代碼
當前鏈接：http://m.newbst.com/article36/ijsipg.html

成都網站建設公司_創新互聯，為您提供品牌網站設計、動態網站、搜索引擎優化、網站營銷、服務器托管、外貿網站建設

廣告

聲明：本網站發布的內容（圖片、視頻和文字）以用戶投稿、用戶轉載內容為主，如果涉及侵權請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網站立場，如需處理請聯系客服。電話：028-86922220；郵箱：631063699@qq.com。內容未經允許不得轉載，或轉載時需注明來源： 創新互聯

營銷型網站建設