mysql+spring+mybatis實現(xiàn)數(shù)據(jù)庫讀寫分離的代碼配置

更新時間：2019年03月11日 14:43:21 作者：小飛俠-2

今天小編就為大家分享一篇關于mysql+spring+mybatis實現(xiàn)數(shù)據(jù)庫讀寫分離的代碼配置，小編覺得內(nèi)容挺不錯的，現(xiàn)在分享給大家，具有很好的參考價值，需要的朋友一起跟隨小編來看看吧

場景：一個讀數(shù)據(jù)源一個讀寫數(shù)據(jù)源。

原理：借助spring的【org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource】這個抽象類實現(xiàn)，看名字可以了解到是一個路由數(shù)據(jù)源的東西，這個類中有一個方法

/** 
 * Determine the current lookup key. This will typically be 
 * implemented to check a thread-bound transaction context. 
 * <p>Allows for arbitrary keys. The returned key needs 
 * to match the stored lookup key type, as resolved by the 
 * {@link #resolveSpecifiedLookupKey} method. 
 */ 
protected abstract Object determineCurrentLookupKey();

每次去連數(shù)據(jù)庫的時候，spring會調(diào)用這個方法去找對應的數(shù)據(jù)源。返回值即對應的數(shù)據(jù)源的LookUpKey.那么這個LookUpKey在哪定義的呢？看下面的dataBase.xml的配置

<!--數(shù)據(jù)源 讀寫 --> 
<bean id="dataSourceRW" class="org.logicalcobwebs.proxool.ProxoolDataSource"> 
  <property name="alias" value="ihotelRW"></property> 
  <property name="delegateProperties"> 
    <value>user=${jdbc.username},password=${jdbc.password} 
    </value> 
  </property> 
  <property name="user" value="${jdbc.username}" /> 
  <property name="password" value="${jdbc.password}" /> 
  <property name="driver" value="${jdbc.driverClassName}" /> 
  <property name="driverUrl" value="${jdbc.url}" /> 
  <property name="maximumConnectionCount" value="${jdbc.maximumConnectionCount}"></property> 
  <property name="maximumActiveTime" value="${jdbc.maximumActiveTime}"></property> 
  <property name="maximumConnectionLifetime" value="${jdbc.maximumConnectionLifetime}"></property> 
  <property name="prototypeCount" value="${jdbc.prototypeCount}"></property> 
  <property name="houseKeepingSleepTime" value="${jdbc.houseKeepingSleepTime}"></property> 
  <property name="simultaneousBuildThrottle" value="${jdbc.simultaneousBuildThrottle}"></property> 
  <property name="houseKeepingTestSql" value="${jdbc.houseKeepingTestSql}"></property> 
  <property name="verbose" value="${jdbc.verbose}"></property> 
  <property name="statistics" value="${jdbc.statistics}"></property> 
  <property name="statisticsLogLevel" value="${jdbc.statisticsLogLevel}"></property> 
</bean> 
  <!--數(shù)據(jù)源 讀--> 
  <bean id="dataSourceR" class="org.logicalcobwebs.proxool.ProxoolDataSource"> 
    <property name="alias" value="ihotelR"></property> 
    <property name="delegateProperties"> 
      <value>user=${jdbc.r.username},password=${jdbc.r.password} 
      </value> 
    </property> 
    <property name="user" value="${jdbc.r.username}" /> 
    <property name="password" value="${jdbc.r.password}" /> 
    <property name="driver" value="${jdbc.r.driverClassName}" /> 
    <property name="driverUrl" value="${jdbc.r.url}" /> 
    <property name="maximumConnectionCount" value="${jdbc.maximumConnectionCount}"></property> 
    <property name="maximumActiveTime" value="${jdbc.maximumActiveTime}"></property> 
    <property name="maximumConnectionLifetime" value="${jdbc.maximumConnectionLifetime}"></property> 
    <property name="prototypeCount" value="${jdbc.prototypeCount}"></property> 
    <property name="houseKeepingSleepTime" value="${jdbc.houseKeepingSleepTime}"></property> 
    <property name="simultaneousBuildThrottle" value="${jdbc.simultaneousBuildThrottle}"></property> 
    <property name="houseKeepingTestSql" value="${jdbc.houseKeepingTestSql}"></property> 
    <property name="verbose" value="${jdbc.verbose}"></property> 
    <property name="statistics" value="${jdbc.statistics}"></property> 
    <property name="statisticsLogLevel" value="${jdbc.statisticsLogLevel}"></property> 
  </bean> 
  <!-- 動態(tài)數(shù)據(jù)源 --> 
  <bean id="dynamicDataSource" class="com.dao.datasource.DynamicDataSource"> 
    <!-- 通過key-value關聯(lián)數(shù)據(jù)源 --> 
    <property name="targetDataSources"> 
      <map> 
        <entry value-ref="dataSourceRW" key="dataSourceKeyRW"></entry> 
        <entry value-ref="dataSourceR" key="dataSourceKeyR"></entry> 
      </map> 
    </property> 
    <property name="defaultTargetDataSource" ref="dataSourceRW" />   
  </bean> 
<!--mybatis與Spring整合 開始 --> 
<bean id="sqlSessionFactory" name="sqlSessionFactory" 
  class="org.mybatis.spring.SqlSessionFactoryBean"> 
  <property name="configLocation" value="classpath:conf/core/sqlMapConfig.xml" /> 
  <property name="dataSource" ref="dynamicDataSource" /> 
</bean>

動態(tài)數(shù)據(jù)源dynamicDataSource中的dataSourceKeyRW、dataSourceKeyR就是

protected abstract Object determineCurrentLookupKey();

這個方法要返回的值。那么如何設置，讓這個方法的返回值是根據(jù)我們的需要返回dataSourceKeyRW、dataSourceKeyR呢？由于這個方法沒有入?yún)?，并且是spring自動調(diào)用的，因此考慮使用靜態(tài)變量存儲dataSource的key，在調(diào)用sql語句前設置靜態(tài)變量的值，然后在這個方法中得到靜態(tài)變量的值，返回。又考慮到多線程，同時可能會有很多請求，為避免線程之間相互干擾，考慮使用threadLocal。

先看存儲dataSourceKey的容器類。

public class DBContextHolder { 
  /** 
   * 線程threadlocal 
   */ 
  private static ThreadLocal<String> contextHolder = new ThreadLocal<>(); 
  private String DB_TYPE_RW = "dataSourceKeyRW"; 
  private String DB_TYPE_R = "dataSourceKeyR"; 
  public String getDbType() { 
    String db = contextHolder.get(); 
    if (db == null) { 
      db = DB_TYPE_RW;// 默認是讀寫庫 
    } 
    return db; 
  } 
  /** 
   * 設置本線程的dbtype 
   * @param str 
   * @see [相關類/方法](可選) 
   * @since [產(chǎn)品/模塊版本](可選) 
   */ 
  public void setDbType(String str) { 
    contextHolder.set(str); 
  } 
  /** 
   * clearDBType 
   * @Title: clearDBType 
   * @Description: 清理連接類型 
   */ 
  public static void clearDBType() { 
    contextHolder.remove(); 
  } 
}

動態(tài)數(shù)據(jù)源的實現(xiàn)類。

public class DynamicDataSource extends AbstractRoutingDataSource { 
  /* 
   * (non-Javadoc) 
   * @see javax.sql.CommonDataSource#getParentLogger() 
   */ 
  @Override 
  public Logger getParentLogger() throws SQLFeatureNotSupportedException { 
    // TODO Auto-generated method stub 
    return null; 
  } 
  /** 
   * override determineCurrentLookupKey 
   * <p> 
   * Title: determineCurrentLookupKey 
   * </p> 
   * <p> 
   * Description: 自動查找datasource 
   * </p> 
   * @return 
   */ 
  @Override 
  protected Object determineCurrentLookupKey() { 
    return DBContextHolder.getDbType(); 
  } 
}

在DAO層中設置數(shù)據(jù)庫類型。

/** 
   * 添加郵件 
   * @param sms 
   * @return 
   */ 
  public boolean insertEmail(Email email) { 
    //根據(jù)具體需要設置不同的數(shù)據(jù)庫 
    DBContextHolder.setDbType(DBContextHolder.DB_TYPE_RW); 
    //DBContextHolder.setDbType(DBContextHolder.DB_TYPE_R); 
    int result = this.getSqlSession().insert(STATEMENT + ".addEntity", 
        email); 
    return result == 1; 
  }

在本例中，我們是在DAO中指定數(shù)據(jù)庫，我們也可以根據(jù)需要在service或者controller中指定DB類型，需要記住的是setDbType是針對線程維度的。要考慮多線程的問題。

總結

以上就是這篇文章的全部內(nèi)容了，希望本文的內(nèi)容對大家的學習或者工作具有一定的參考學習價值，謝謝大家對腳本之家的支持。如果你想了解更多相關內(nèi)容請查看下面相關鏈接

您可能感興趣的文章: