配置代碼MySQL數據庫讀寫分離

首先,設置一個攔截器,攔截數據庫操作,判斷類型,動態選擇數據源
DynamicDataSourceinterceptor.java

package com.mlr.dao.split;

import java.util.Locale;
import java.util.Properties;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.executor.keygen.SelectKeyGenerator;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlCommandType;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.plugin.Intercepts;
import org.apache.ibatis.plugin.Invocation;
import org.apache.ibatis.plugin.Plugin;
import org.apache.ibatis.plugin.Signature;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.transaction.support.TransactionSynchronizationManager;

@Intercepts({@Signature(type = Executor.class,method = "update",args={MappedStatement.class,Object.class}),
    @Signature(type = Executor.class,method = "query",args={MappedStatement.class,Object.class,
    RowBounds.class, ResultHandler.class})})
public class DynamicDataSourceinterceptor implements Interceptor {

  private static Logger logger = LoggerFactory.getLogger(DynamicDataSourceinterceptor.class);
  private static final String REGEX = ".*insert\\u0020.*|.*delete\\0020.*|.*update\\u0020.*";

  public java.lang.Object intercept(Invocation invocation) throws Throwable {
    boolean synchronizationActive = TransactionSynchronizationManager.isSynchronizationActive();
    //對于不是事務管理的操作


      Object[] objects = invocation.getArgs();
      String lookupKey =  DynamicDataSourceHolder.DB_MASTER;
      //獲取操作類型
      MappedStatement mappedStatement = (MappedStatement) objects[0];
    if (synchronizationActive != true) {
      if (mappedStatement.getSqlCommandType().equals(SqlCommandType.SELECT)) {
        if (mappedStatement.getId().contains(SelectKeyGenerator.SELECT_KEY_SUFFIX)) {
          lookupKey = DynamicDataSourceHolder.DB_MASTER;
        } else {
          BoundSql boundSql = mappedStatement.getSqlSource().getBoundSql(objects[1]);
          String sql = boundSql.getSql().toLowerCase(Locale.CHINA).replaceAll("[\\t\\n\\r]", "");

          //符合正則用主庫,不符合用從庫
          if (sql.matches(REGEX)) {
            lookupKey = DynamicDataSourceHolder.DB_MASTER;
          } else {
            lookupKey = DynamicDataSourceHolder.DB_SLAVE;
          }
        }
      }

    }
    //對于事務管理都用主庫
    else{
      lookupKey = DynamicDataSourceHolder.DB_MASTER;
    }
    logger.debug("設置方法[{}],use[{}]Strategy,SqlCommandType[{}]...",mappedStatement.getId(),lookupKey,mappedStatement.getSqlCommandType().name());
    DynamicDataSourceHolder.setDBType(lookupKey);
    return invocation.proceed();
  }

  /**
   * Executor支持增刪改查操作,當操作時候攔截下來判斷用哪個數據源
   */
  public java.lang.Object plugin(java.lang.Object target) {
    if (target instanceof Executor) {
      return Plugin.wrap(target, this);
    } else {
      return target;
    }
  }

  public void setProperties(Properties properties) {

  }
}


DynamicDataSourceHolder.java

package com.mlr.dao.split;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class DynamicDataSourceHolder {

  private static Logger logger = LoggerFactory.getLogger(DynamicDataSourceHolder.class);

  private static ThreadLocal<String> contextHolder = new ThreadLocal<String>();

  public static final String DB_MASTER = "master";
  public static final String DB_SLAVE = "slave";

  /**
   * 獲取類型
   * @return
   */
  public static String getDBType() {

    String db = contextHolder.get();

    if (db == null) {
      db = DB_MASTER;
    }
    return db;
  }

  /**
   * 設置連接類型
   * @param str
   */
  public static void setDBType(String str) {
    logger.debug("所使用的數據源為" + str);
    contextHolder.set(str);
  }

  /**
   * 清理連接類型
   * @param str
   */
  public static void clearDBType(String str) {
    contextHolder.remove();
  }


}

DynamicDataSource .java

package com.mlr.dao.split;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

public class DynamicDataSource extends AbstractRoutingDataSource{


  protected Object determineCurrentLookupKey() {
    return DynamicDataSourceHolder.getDBType();
  }
}


然后配置mybatis-config.xml中添加攔截器

<plugins>
    <plugin interceptor="com.mlr.dao.split.DynamicDataSourceinterceptor"></plugin>
  </plugins>

在spring-dao.xml中配置數據源

<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
  xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
  xmlns:context="http://www.springframework.org/schema/context"
  xsi:schemaLocation="http://www.springframework.org/schema/beans
    http://www.springframework.org/schema/beans/spring-beans.xsd
    http://www.springframework.org/schema/context
    http://www.springframework.org/schema/context/spring-context.xsd">
  <!-- 配置整合mybatis過程 -->
  <!-- 1.配置數據庫相關參數properties的屬性:${url} -->
  <context:property-placeholder location="classpath:jdbc.properties"/>

  <!-- 2.數據庫連接池 -->
  <bean id="abstractDataSource" abstract="true" destroy-method="close"
    class="com.mchange.v2.c3p0.ComboPooledDataSource">
    <!-- 配置連接池屬性 -->
    <property name="driverClass" value="${jdbc.driver}"/>
    <property name="jdbcUrl" value="${jdbc.url}"/>
    <property name="user" value="${jdbc.username}"/>
    <property name="password" value="${jdbc.password}"/>


    <!-- c3p0連接池的私有屬性 -->
    <property name="maxPoolSize" value="30"/>
    <property name="minPoolSize" value="10"/>
    <!-- 關閉連接后不自動commit -->
    <property name="autoCommitOnClose" value="false"/>
    <!-- 獲取連接超時時間 -->
    <property name="checkoutTimeout" value="10000"/>
    <!-- 當獲取連接失敗重試次數 -->
    <property name="acquireRetryAttempts" value="2"/>
  </bean>
  <!--主庫數據源-->
  <bean id="master" parent="abstractDataSource">
    <property name="driverClass" value="${jdbc.driver}"/>
    <property name="jdbcUrl" value="${jdbc.master.url}"/>
    <property name="user" value="${jdbc.username}"/>
    <property name="password" value="${jdbc.password}"/>
  </bean>
  <!--從庫數據源-->
  <bean id="slave" parent="abstractDataSource">
    <property name="driverClass" value="${jdbc.driver}"/>
    <property name="jdbcUrl" value="${jdbc.slave.url}"/>
    <property name="user" value="${jdbc.username}"/>
    <property name="password" value="${jdbc.password}"/>
  </bean>
  <!--配置動態數據源,這里targetDataSource就是路由數據源所對應名稱-->
  <bean id="dynamicDataSource" class="com.mlr.dao.split.DynamicDataSource">
    <property name="targetDataSources">
      <map>
        <entry  value-ref="master" key="master"></entry>
        <entry key="master" value-ref="master"></entry>
    </map>
  </property>
</bean>
  <!--懶加載,程序運行后判斷sql-->
<bean id="dataSource" class="org.springframework.jdbc.datasource.LazyConnectionDataSourceProxy">
  <property name="targetDataSource">
    <ref bean="dynamicDataSource"></ref>
  </property>
</bean>
  <!-- 3.配置SqlSessionFactory對象 -->
<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
<!-- 注入數據庫連接池 -->
<property name="dataSource" ref="dataSource"/>
<!-- 配置MyBaties全局配置文件:mybatis-config.xml -->
<property name="configLocation" value="classpath:mybatis-config.xml"/>
<!-- 掃描entity包 使用別名 -->
<property name="typeAliasesPackage" value="com.mlr.entity"/>
<!-- 掃描sql配置文件:mapper需要的xml文件 -->
<property name="mapperLocations" value="classpath:mapper/*.xml"/>
</bean>

  <!-- 4.配置掃描Dao接口包,動態實現Dao接口,注入到spring容器中 -->
<bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
<!-- 注入sqlSessionFactory -->
<property name="sqlSessionFactoryBeanName" value="sqlSessionFactory"/>
<!-- 給出需要掃描Dao接口包 -->
<property name="basePackage" value="com.mlr.dao"/>
</bean>
  </beans>
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 230,002評論 6 542
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 99,400評論 3 429
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 178,136評論 0 383
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,714評論 1 317
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 72,452評論 6 412
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,818評論 1 328
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,812評論 3 446
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,997評論 0 290
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 49,552評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 41,292評論 3 358
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,510評論 1 374
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 39,035評論 5 363
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,721評論 3 348
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 35,121評論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,429評論 1 294
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 52,235評論 3 398
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 48,480評論 2 379

推薦閱讀更多精彩內容