配置代码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阅读 212,686评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,668评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,160评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,736评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,847评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,043评论 1 291
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,129评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,872评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,318评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,645评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,777评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,470评论 4 333
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,126评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,861评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,095评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,589评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,687评论 2 351

推荐阅读更多精彩内容