文档章节

读写分离(spring事务代理+mybaits拦截器实现)

bodomg
 bodomg
发布于 2017/02/13 14:31
字数 2241
阅读 376
收藏 3

1、读写分离

         采用继承DataSourceTransactionManager控制事务读写分离,以及mybatis拦截器控制方法读写分离,通过继承AbstractRoutingDataSource获取动态数据源。

1.1、代码说明

1.1.1、本地数据源管理

创建本地数据源类型管理类,使用ThreadLocal保存本地数据源类型

bodsite-common - com.bodsite.common.datasource. DataSourceHandler

package com.bodsite.common.datasource;

/**

 * @Description:本地线程数据源

 * @author bod

 * @date

 *

 */

public class DataSourceHandler {

   public enum DYNAMIC_DATA_SOURCE{

      MASTER,//主库(写)

      SLAVE;//从库(读)

   }

  

   private static final ThreadLocal<DYNAMIC_DATA_SOURCE> dataSourceThreadLocal = new ThreadLocal<DYNAMIC_DATA_SOURCE>();

  

   protected static void set(DYNAMIC_DATA_SOURCE dynamic_data_source){

      dataSourceThreadLocal.set(dynamic_data_source);

   }

  

   /**

    * 设置为主库

    * @author bod

    */

   protected static void setMaster(){

      dataSourceThreadLocal.set(DYNAMIC_DATA_SOURCE.MASTER);

   }

  

   /**

    * 设置为读库

    * @author bod

    */

   protected static void setSlave(){

      dataSourceThreadLocal.set(DYNAMIC_DATA_SOURCE.SLAVE);

   }

 

   /**

    * 判断是否为主库

    * @author bod

    */

   protected static boolean isMaster(){

      return isThis(DYNAMIC_DATA_SOURCE.MASTER);

   }

  

   /**

    * 判断是否为从

    * @author bod

    */

   protected static boolean isSlave(){

      return isThis(DYNAMIC_DATA_SOURCE.SLAVE);

   }

 

   protected static boolean isThis(DYNAMIC_DATA_SOURCE dynamic_data_source){

      if(dataSourceThreadLocal.get()==null){

         return false;

      }

      return dynamic_data_source == dataSourceThreadLocal.get();

   }

  

   protected static void DataSoruceClean(){

      dataSourceThreadLocal.remove();

   }

}

1.1.2、mybatis拦截,设置数据源

创建mybatis拦截器,拦截update、query方法,设置数据源,如果有事务,不做拦截,有事务的情况,在事务管理器中进行设置。

bodsite-common - com.bodsite.common.datasource. DynamicDataSourceInterceptor

package com.bodsite.common.datasource;

 

import java.util.Properties;

 

import org.apache.ibatis.executor.Executor;

import org.apache.ibatis.executor.keygen.SelectKeyGenerator;

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.springframework.transaction.support.TransactionSynchronizationManager;

 

/**

 * @Description: mybatis plugin 拦截器-设置数据源

 * @author bod

 * @date

 *

 */

@Intercepts({

      @Signature(method = "query", type = Executor.class, args = { MappedStatement.class, Object.class,

            RowBounds.class, ResultHandler.class }), // method:方法名,type:类,args:方法参数

      @Signature(method = "update", type = Executor.class, args = { MappedStatement.class, Object.class }) })

public class DynamicDataSourceInterceptor implements Interceptor {

 

   @Override

   public Object intercept(Invocation invocation) throws Throwable {

      // 是否有事务

      boolean synchronizationActive = TransactionSynchronizationManager.isSynchronizationActive();

      if (!synchronizationActive) {

         Object[] args = invocation.getArgs();

         MappedStatement ms = (MappedStatement) args[0];

         if(ms.getSqlCommandType().equals(SqlCommandType.SELECT)){//查询

              //!selectKey 为自增id查询主键(SELECT LAST_INSERT_ID() )方法,使用主库

                if(ms.getId().contains(SelectKeyGenerator.SELECT_KEY_SUFFIX)) {

                DataSourceHandler.setMaster();

                }else{

                DataSourceHandler.setSlave();

                }

         }else{//其他

            DataSourceHandler.setMaster();

         }

      }

      Object result = invocation.proceed();

      DataSourceHandler.DataSoruceClean();

      return result;

   }

 

   @Override

   public Object plugin(Object target) {

      if (target instanceof Executor) {

         return Plugin.wrap(target, this);

      } else {

         return target;

      }

   }

 

   @Override

   public void setProperties(Properties properties) {

 

   }

 

}

 

1.1.3、spring事务管理,设置数据源

创建动态事务管理类,继承DataSourceTransactionManager,根据读写判断,设置数据源。

bodsite-common - com.bodsite.common.datasource. DataSourceTransactionManager

   package com.bodsite.common.datasource;

 

import org.springframework.jdbc.datasource.DataSourceTransactionManager;

import org.springframework.transaction.TransactionDefinition;

 

/**

 *

 * @Description:根据事务这是数据源(必须有事务才能进入)

 * @author bod

 * @date

 *

 */

public class  DynamicDataSourceTransactionManager extends DataSourceTransactionManager{

 

   /**

    *

    */

   private static final long serialVersionUID = 1L;

 

   @Override

   protected void doBegin(Object transaction, TransactionDefinition definition) {

      boolean readOnly = definition.isReadOnly();

      if(readOnly){//只读

         DataSourceHandler.setSlave();

      }else{//读写

         DataSourceHandler.setMaster();

      }

      super.doBegin(transaction, definition);

   }

 

   @Override

   protected void doCleanupAfterCompletion(Object transaction) {

      super.doCleanupAfterCompletion(transaction);

      DataSourceHandler.DataSoruceClean();

   }

  

}

 

1.1.4、动态获取数据源

创建动态获取数据源类,继承AbstractRoutingDataSource,根据读写判断,设置数据源。

提供两种获取数据源方式:

1、通过重写determineCurrentLookupKey方法,返回数据源名称,走AbstractRoutingDataSource的determineTargetDataSource(根据determineCurrentLookupKey返回的名称)方法获取数据源。

2、通过重写determineTargetDataSource。直接返回数据源。

bodsite-common - com.bodsite.common.datasource. DynamicDataSource

package com.bodsite.common.datasource;

 

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.concurrent.ThreadLocalRandom;

import java.util.concurrent.atomic.AtomicInteger;

 

import javax.sql.DataSource;

 

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

 

import com.bodsite.common.logger.Logger;

import com.bodsite.common.logger.LoggerFactory;

 

/**

 * @Description:动态数据源

 * @author bod

 * @date

 *

 */

public class DynamicDataSource extends AbstractRoutingDataSource {

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

   private DataSource master;

   private List<DataSource> slaves;

    private int slaveSize; //读数据源个数

   private Integer strategy;// 默认,0:轮询,1,随机

   private AtomicInteger counter = new AtomicInteger();

   private Map<Object, Object> targetDataSources = new HashMap<>();

 

   /******************* 1、加载数据源关系, 设置数据源名,会根据数据源名返回数据源 ************/

   /**

    * 设置数据源名

    */

   @Override

   protected Object determineCurrentLookupKey() {

      return getDataSourceName();

   }

 

   /**

    * 设置数据源映射关系

    */

   @Override

   public void afterPropertiesSet() {

      if (this.master == null) {

         throw new IllegalArgumentException("Property 'targetDataSources' is required");

      }

      setDefaultTargetDataSource(master);

      targetDataSources.put(DataSourceHandler.DYNAMIC_DATA_SOURCE.MASTER.name(), master);

      if (slaves != null && !slaves.isEmpty()) {

         this.slaveSize = slaves.size();

         for (int i = 0; i < slaveSize; i++) {

            targetDataSources.put(DataSourceHandler.DYNAMIC_DATA_SOURCE.SLAVE.name() + i, slaves.get(i));

         }

      }

      setTargetDataSources(targetDataSources);

      super.afterPropertiesSet();

   }

 

   /******************* 2、直接返回数据源 ***********************/

   /**

    * 原方法:根据数据源名称返回数据源,自定义直接返回数据源

    */

   /*@Override

   protected DataSource determineTargetDataSource() {

      return (DataSource) targetDataSources.get(getDataSourceName());

   }*/

 

   /**

    * 获取数据源名称

    * @author bod

    */

   public String getDataSourceName() {

      String dataSourceName = null;

      if (DataSourceHandler.isMaster()) {

         dataSourceName = DataSourceHandler.DYNAMIC_DATA_SOURCE.MASTER.name();

      } else if (DataSourceHandler.isSlave() && slaves != null && !slaves.isEmpty()) {

         int index = 0;

         if (strategy == null || strategy == 0) {

            int count = counter.incrementAndGet();

            index = count%slaveSize;

         }else if(strategy == 1){

               index = ThreadLocalRandom.current().nextInt(0, slaveSize);

         }

         dataSourceName = DataSourceHandler.DYNAMIC_DATA_SOURCE.SLAVE.name()+index;

      }else{

         dataSourceName = DataSourceHandler.DYNAMIC_DATA_SOURCE.MASTER.name();

      }

      logger.info("This data source name is "+dataSourceName);

      return dataSourceName;

   }

   public DataSource getMaster() {

      return master;

   }

 

   public void setMaster(DataSource master) {

      this.master = master;

   }

 

   public List<DataSource> getSlaves() {

      return slaves;

   }

 

   public void setSlaves(List<DataSource> slaves) {

      this.slaves = slaves;

   }

 

   public Integer getStrategy() {

      return strategy;

   }

 

   public void setStrategy(Integer strategy) {

      this.strategy = strategy;

   }

 

}

 

 

1.1.5、application-mybatis.xml配置

bodsite-site-service - application-mybatis.xml

<?xml version="1.0" encoding="UTF-8"?>

<beans xmlns="http://www.springframework.org/schema/beans"

   xmlns:context="http://www.springframework.org/schema/context"

   xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:tx="http://www.springframework.org/schema/tx"

   xmlns:aop="http://www.springframework.org/schema/aop"

   xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans-4.3.xsd

      http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context-4.3.xsd

            http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop-4.3.xsd

      http://www.springframework.org/schema/tx http://www.springframework.org/schema/tx/spring-tx-4.3.xsd">

   <!-- druid 连接池配置信息 http://www.cnblogs.com/SummerinShire/p/5828888.html -->

   <!-- druid 数据库连接池 -->

   <bean id="masterDataSource" class="com.alibaba.druid.pool.DruidDataSource"

      init-method="init" destroy-method="close">

      <!-- 连接数据库信息 -->

      <property name="driverClassName" value="${master.jdbc.driver}" />

      <property name="url" value="${master.jdbc.url}" />

      <property name="username" value="${master.jdbc.username}" />

      <property name="password" value="${master.jdbc.password}" />

 

      <!-- 初始化时建立物理连接的个数。初始化发生在显示调用init方法,或者第一次getConnection时 -->

      <property name="initialSize" value="1" />

      <!-- 最大连接池数量 -->

      <property name="maxActive" value="50" />

      <!-- 最小连接池数量 -->

      <property name="minIdle" value="10" />

      <!-- 获取连接时最大等待时间,单位毫秒 -->

      <property name="maxWait" value="60000" />

      <!-- 配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒 -->

      <property name="timeBetweenEvictionRunsMillis" value="60000" />

      <!-- 配置一个连接在池中最小生存的时间,单位是毫秒 -->

      <property name="minEvictableIdleTimeMillis" value="300000" />

      <!-- 用来检测连接是否有效的sq -->

      <property name="validationQuery" value="SELECT 'x' FROM DUAL" />

      <!-- 建议配置为true,不影响性能,并且保证安全性 -->

      <property name="testWhileIdle" value="true" />

      <!-- 申请连接时执行validationQuery检测连接是否有效,做了这个配置会降低性能。 -->

      <property name="testOnBorrow" value="false" />

      <!-- 归还连接时执行validationQuery检测连接是否有效,做了这个配置会降低性能。 -->

      <property name="testOnReturn" value="false" />

      <!-- 属性类型是字符串,通过别名的方式配置扩展插件,常用的插件有:监控统计用的filter:stat 日志用的filter:log4j 防御sql注入的filter:wall -->

      <property name="filters" value="stat" />

   </bean>

 

   <bean id="slaveDataSource" class="com.alibaba.druid.pool.DruidDataSource"

      init-method="init" destroy-method="close">

      <!-- 连接数据库信息 -->

      <property name="driverClassName" value="${slave1.jdbc.driver}" />

      <property name="url" value="${slave1.jdbc.url}" />

      <property name="username" value="${slave1.jdbc.username}" />

      <property name="password" value="${slave1.jdbc.password}" />

 

      <!-- 初始化时建立物理连接的个数。初始化发生在显示调用init方法,或者第一次getConnection时 -->

      <property name="initialSize" value="1" />

      <!-- 最大连接池数量 -->

      <property name="maxActive" value="50" />

      <!-- 最小连接池数量 -->

      <property name="minIdle" value="10" />

      <!-- 获取连接时最大等待时间,单位毫秒 -->

      <property name="maxWait" value="60000" />

      <!-- 配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒 -->

      <property name="timeBetweenEvictionRunsMillis" value="60000" />

      <!-- 配置一个连接在池中最小生存的时间,单位是毫秒 -->

      <property name="minEvictableIdleTimeMillis" value="300000" />

      <!-- 用来检测连接是否有效的sq -->

      <property name="validationQuery" value="SELECT 'x' FROM DUAL" />

      <!-- 建议配置为true,不影响性能,并且保证安全性 -->

      <property name="testWhileIdle" value="true" />

      <!-- 申请连接时执行validationQuery检测连接是否有效,做了这个配置会降低性能。 -->

      <property name="testOnBorrow" value="false" />

      <!-- 归还连接时执行validationQuery检测连接是否有效,做了这个配置会降低性能。 -->

      <property name="testOnReturn" value="false" />

      <!-- 属性类型是字符串,通过别名的方式配置扩展插件,常用的插件有:监控统计用的filter:stat 日志用的filter:log4j 防御sql注入的filter:wall -->

      <property name="filters" value="stat" />

   </bean>

   <!-- 动态数据源 -->

   <bean id="dataSource" class="com.bodsite.common.datasource.DynamicDataSource">

      <property name="master" ref="masterDataSource" />

      <property name="slaves">

         <list>

            <ref bean="slaveDataSource" />

         </list>

      </property>

      <property name="strategy" value="0"/>

   </bean>

 

 

   <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">

      <property name="dataSource" ref="dataSource" />

      <property name="configLocation" value="classpath:mybatis-config.xml" />

      <property name="mapperLocations" value="classpath:/mappings/**/*.xml" />

   </bean>

   <!-- 定义事务 -->

   <bean id="transactionManager"

      class="com.bodsite.common.datasource.DynamicDataSourceTransactionManager">

      <property name="dataSource" ref="dataSource" />

   </bean>

   <!-- 扫描@Transactional注解的类定义事务 -->

   <tx:annotation-driven transaction-manager="transactionManager"

      proxy-target-class="true" />

   <!-- 自动注册mybatis mapper bean -->

   <!-- 注意,没有必要去指定SqlSessionFactory或SqlSessionTemplate, 因为MapperScannerConfigurer将会创建

      MapperFactoryBean,之后自动装配。 但是,如果你使 用了一个以上的DataSource,那 么自动装配可能会失效。 这种情况下,你可以使用

      sqlSessionFactoryBeanName或sqlSessionTemplateBeanName 属性来设置正确的 bean名称来使用。 -->

   <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">

      <property name="basePackage" value="com.bodsite.**.dao" />

   </bean>

</beans>

1.1.6、mybatis-generator.xml 配置-添加拦截器

bodsite-site-service - mybatis- generator.xml

<?xml version="1.0" encoding="UTF-8"?>

<!DOCTYPE configuration PUBLIC "-//mybatis.org//DTD Config 3.0//EN" "http://mybatis.org/dtd/mybatis-3-config.dtd">

<configuration>

   <settings>

      <!--对在此配置文件下的所有cache 进行全局性开/关设置。 -->

      <setting name="cacheEnabled" value="true" />

      <!-- 全局性设置懒加载。如果设为‘false’,则所有相关联的都会被初始化加载。 -->

      <setting name="lazyLoadingEnabled" value="true" />

      <!-- 允许和不允许单条语句返回多个数据集(取决于驱动需求) -->

      <setting name="multipleResultSetsEnabled" value="true" />

      <!-- 使用列标签代替列名称。 -->

      <setting name="useColumnLabel" value="true" />

      <!-- 允许JDBC 生成主键。需要驱动器支持。如果设为了true,这个设置将强制使用被生成的主键,有一些驱动器不兼容不过仍然可以执行。 -->

      <setting name="useGeneratedKeys" value="false" />

      <!-- 这是默认的执行类型 (SIMPLE: 简单; REUSE: 执行器可能重复使用prepared statements语句;BATCH:

         执行器可以重复执行语句和批量更新) -->

      <setting name="defaultExecutorType" value="SIMPLE" />

      <!-- 配置Java属性-数据库表字段对应驼峰规则 -->

      <setting name="mapUnderscoreToCamelCase" value="true" />

   </settings>

   <plugins>

   <!-- 读写分离拦截器 -->

   <plugin interceptor="com.bodsite.common.datasource.DynamicDataSourceInterceptor">

   </plugin>

   </plugins>

</configuration>

1.2、测试

1、启动bodsite-site-service,在bodsite-site中的DemoConsumer类,执行查询

  

2、在bodsite-site中的DemoConsumer类,执行插入

项目地址:https://git.oschina.net/bodsite/bodsite

© 著作权归作者所有

共有 人打赏支持
bodomg
粉丝 15
博文 7
码字总数 5810
作品 0
徐汇
高级程序员
私信 提问
分布式事务系列(1.2)Spring的事务体系

1 系列目录 - 分布式事务系列(开篇)提出疑问和研究过程- 分布式事务系列(1.1)Spring事务管理器PlatformTransactionManager源码分析- 分布式事务系列(1.2)Spring事务体系- 分布式事务系...

乒乓狂魔
2015/05/18
0
4
MiniDao_1.6-SNAPSHOT 版本发布

MiniDao-PE 简介 MiniDao-PE 是一种持久化解决方案,类似mybatis的持久层解决方案,可以轻松集成Hibernate工程,事务统一管理,解决了Hibernate工程想支持mybaits的功能问题。Hibernate的最佳...

多吉Q
2016/07/20
4K
2
Spring编程式和声明式事务

1.编程式事务 1.1 编程式和声明式事务的区别 Spring提供了对编程式事务和声明式事务的支持,编程式事务允许用户在代码中精确定义事务的边界,而声明式事务(基于AOP)有助于用户将操作与事务...

梨加橙
06/19
0
0
分布式系统 - iBase4J

Spring boot,Spring,SpringMVC,Mybatis,mybatis-plus,motan/dubbo分布式,Redis缓存,Shiro权限管理,Spring-Session单点登录,Quartz分布式集群调度,Restful服务,QQ/微信登录,App t...

iBase4J
2016/05/04
0
74
Java面试分享(题目+答案)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/t4i2b10X4c22nF6A/article/details/84535277 1.说下Struts的设计模式 2.拦截器和过滤器的区别? 3.struts2框架...

JAVA高级架构v
11/25
0
0

没有更多内容

加载失败,请刷新页面

加载更多

中国龙-扬科
11分钟前
0
0
假若明天来临——《AI.未来》读后感3900字

假若明天来临——《AI.未来》读后感3900字: 你有没有想过,如果有一天你被确诊为癌症患者,你会做些什么?你有没有想过,在你百年之后,你希望你的墓碑上刻写着什么内容? 在我翻开李开复老...

原创小博客
17分钟前
0
0
tomcat线程模型

Connector结构 BIO模式 NIO模式

grace_233
57分钟前
2
0
Javascript

变量定义 以$,_,字母开头 大小写敏感 var 关键字声明变量 声明未赋值undefined 数据类型 字符串,数字,布尔,数组,NULL,undefined 变量均为对象 函数 无默认值 var声明的是局部变量 函数外声明...

关元
59分钟前
1
0
文件系统、服务、防火墙、SELINUX——安全四大金刚

一提到安全,大家都会想到防火墙,和文件系统权限。而实际工作环境中,我们在Linux的安全配置,会涉及到四个级别。我们思考一个场景,你要在百度盘中存放一个文件,这个动作需要考虑下面四个...

Linux就该这么学
今天
3
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部