java-durid、mybatis、spring 整合基于 AbstractRoutingDataSource 的多数据源读写分离配置

╰+攻爆jí腚メ 2024-02-17 21:14 8阅读 0赞

java-durid、mybatis、spring 整合基于 AbstractRoutingDataSource 的多数据源读写分离配置  
  
  
一、定义用于标记使用那个数据源的注解

package me.utils.database;
    
    
    import java.lang.annotation.ElementType;  
    import java.lang.annotation.Target;  
    import java.lang.annotation.Retention;  
    import java.lang.annotation.RetentionPolicy;  
    /** 
     * 定义 @DataSource 注解，标记当前使用的数据源
     * @version 创建时间：2017年4月17日 上午11:02:07
     */
    @Retention(RetentionPolicy.RUNTIME)  
    @Target(ElementType.METHOD)  
    public @interface DataSource {
    	String value();
    }

二、定义一个切入点，用于在执行数据连接前通过读取 @DataSource 注解，动态选择数据源

package me.utils.database;
    
    
    import java.lang.reflect.Method;
    import org.aspectj.lang.JoinPoint;
    
    
    import org.aspectj.lang.reflect.MethodSignature;
    import org.aspectj.weaver.ast.Var;
    
    
    import com.sun.tools.javac.resources.javac;
    
    
    /**
     * 多数据源切换切入点。解析  @DataSource 注解
     * @version 创建时间：2017年4月17日 上午11:08:08
     */
    public class DataSourceAspect {
    
    
    	/**
    	 * 在dao层方法之前调用，获取  @DataSource 注解标记的数据源名称
    	 */
    	public void before(JoinPoint point) {
    		Object target = point.getTarget();
    		String methodName = point.getSignature().getName();
            MethodSignature methodSignature = (MethodSignature) point.getSignature();  
            Method method = methodSignature.getMethod();  
    		//获取方法上的 @DataSource 注解
    		DataSource annotation = (DataSource) method.getAnnotation(DataSource.class);
    		String dataSourceName = null;
            if (annotation != null) {
            	//获得注解标记的数据源名称
            	dataSourceName = annotation.value();
    	        HandleDataSource.putDataSource(dataSourceName);
            }
    		printMsg("方法={0};使用数据源={1}",methodName,annotation);
            
    	}
    	
    	void printMsg(String pattern, Object... arguments){
    		System.out.println(java.text.MessageFormat.format(pattern,arguments));
    	}
    }

三、定义一个记录当前线程使用的数据源的对象，管理多线程下的变量存储（线程安全）

package me.utils.database;
    /** 
    *保存当前线程数据源使用的数据源名
    * @version 创建时间：2017年4月17日 上午11:07:25
    */
    public class HandleDataSource {
    	/**
    	 * 保存当前线程使用的数据源名称
    	 */
    	public static final ThreadLocal<String> holder = new ThreadLocal<String>();  
        /**  
         * 绑定当前线程数据源路由的key     
         * @param key  
         */  
        public static void putDataSource(String datasource) {  
            holder.set(datasource);
    
    
    		System.out.println(java.text.MessageFormat.format("线程[{0}],设置使用的数据源={1}"
    				,Thread.currentThread().getName()
    				,datasource));
        }  
          
        /**  
         * 获取当前线程的数据源路由的key  
         * @return  
         */  
        public static String getDataSource() {  
    		System.out.println(java.text.MessageFormat.format("线程[{0}],获取使用的数据源={1}"
    				,Thread.currentThread().getName()
    				,holder.get()));
            return holder.get();  
        }     
    }

四、基于 spring 的多数据源选择对象

package me.utils.database;
    
    
    import java.util.Map;
    import java.util.function.BiConsumer;
    
    
    import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
    
    
    /**
     * 继承 AbstractRoutingDataSource 实现自己的路由数据源
     * @version 创建时间：2017年4月17日 上午11:05:20
     */
    public class ChooseDataSource extends AbstractRoutingDataSource  {
    	
    	@Override
    	protected Object determineCurrentLookupKey() {
    		//返回当前线程使用的数据源名称
    		return  HandleDataSource.getDataSource();  
    	}
    	
    	@Override
    	public void setTargetDataSources(Map<Object, Object> targetDataSources) {
    		super.setTargetDataSources(targetDataSources);
    		
    		System.out.println("配置读写分离数据源：");
    		//设置目标数据源
    		if(targetDataSources!=null){
    			targetDataSources.forEach((key,val)->{
    				System.out.println(java.text.MessageFormat.format("数据源：{0}={1}", key,val));
    			});
    		}
    	}
    	
    }

五、 mybatis 的配置（其中的顺序很关键）

<?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"
    	xmlns:tx="http://www.springframework.org/schema/tx" 
    	xmlns:aop="http://www.springframework.org/schema/aop"
    	xmlns:beans="http://www.springframework.org/schema/beans"
    	xsi:schemaLocation="
            http://www.springframework.org/schema/beans 
            http://www.springframework.org/schema/beans/spring-beans-4.2.xsd        
            http://www.springframework.org/schema/context 
            http://www.springframework.org/schema/context/spring-context-4.2.xsd
            http://www.springframework.org/schema/tx 
         	http://www.springframework.org/schema/tx/spring-tx-4.2.xsd
         	http://www.springframework.org/schema/aop 
    		http://www.springframework.org/schema/aop/spring-aop-4.2.xsd">
    
    
    	
    	
    	<bean id="dataSource_druid_write" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
          
          <property name="url" value="${jdbc.url}" />
          <property name="username" value="${jdbc.username}" />
          <property name="password" value="${jdbc.password}" />
            
          
          <property name="initialSize" value="1" />
          <property name="minIdle" value="1" /> 
          <property name="maxActive" value="20" />
       
          
          <property name="maxWait" value="60000" />
       
          
          <property name="timeBetweenEvictionRunsMillis" value="60000" />
       
          
          <property name="minEvictableIdleTimeMillis" value="300000" />
        
          <property name="validationQuery" value="SELECT 'x'" />
          <property name="testWhileIdle" value="true" />
          <property name="testOnBorrow" value="false" />
          <property name="testOnReturn" value="false" />
       
          
          <property name="poolPreparedStatements" value="true" />
          <property name="maxPoolPreparedStatementPerConnectionSize" value="20" />
       
          
          <property name="filters" value="mergeStat" /> 
          
    	  <property name="useGlobalDataSourceStat" value="true" />
    	  
    	  <property name="connectionProperties" value="druid.stat.slowSqlMillis=5000" />
      	</bean>
    	
    	<bean id="dataSource_druid_read" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
          
          <property name="url" value="${jdbc.url}" />
          <property name="username" value="${jdbc.username}" />
          <property name="password" value="${jdbc.password}" />
            
          
          <property name="initialSize" value="1" />
          <property name="minIdle" value="1" /> 
          <property name="maxActive" value="20" />
       
          
          <property name="maxWait" value="60000" />
       
          
          <property name="timeBetweenEvictionRunsMillis" value="60000" />
       
          
          <property name="minEvictableIdleTimeMillis" value="300000" />
        
          <property name="validationQuery" value="SELECT 'x'" />
          <property name="testWhileIdle" value="true" />
          <property name="testOnBorrow" value="false" />
          <property name="testOnReturn" value="false" />
       
          
          <property name="poolPreparedStatements" value="true" />
          <property name="maxPoolPreparedStatementPerConnectionSize" value="20" />
       
          
          <property name="filters" value="mergeStat" /> 
          
    	  <property name="useGlobalDataSourceStat" value="true" />
    	  
    	  <property name="connectionProperties" value="druid.stat.slowSqlMillis=5000" />
      	</bean>
      	
        <bean id="demoDataSource" class="me.utils.database.ChooseDataSource">  
            <property name="targetDataSources">      
              <map key-type="java.lang.String">      
                     
                 <entry key="write" value-ref="dataSource_druid_write"/>      
                     
                 <entry key="read" value-ref="dataSource_druid_read"/>      
              </map>
            </property>   
            
            <property name="defaultTargetDataSource" ref="dataSource_druid_write"/>      
          
        </bean> 	
    	
    	
    	<bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
    		
    		<property name="basePackage" value="me.daos" />
    		<property name="sqlSessionFactoryBeanName" value="sqlSessionFactory" />
    	</bean>
    	
        
        <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
        	
            <property name="dataSource" ref="demoDataSource" />
            
            <property name="configLocation"  value="classpath:mybatis-config/sqlMapConfig.xml"/>
            
            <property name="mapperLocations" value="classpath:mappers/**/*Mapper.xml" />
        </bean>
        
        
        <bean id="sqlSessionTemplate" class="org.mybatis.spring.SqlSessionTemplate">
            <constructor-arg index="0" ref="sqlSessionFactory"></constructor-arg>
        </bean>
    
    
    	
        
        <bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        	
            <property name="dataSource" ref="demoDataSource" />
        </bean>
        
    	<tx:annotation-driven transaction-manager="transactionManager" proxy-target-class="true"/>
    	
    	
    	   
        <bean id="dataSourceAspect" class="me.utils.database.DataSourceAspect" />
            
        <aop:config proxy-target-class="true">    
        	
            <aop:aspect id="dataSourceAspect" ref="dataSourceAspect" order="1">
            	
                <aop:pointcut id="tx" expression="execution(* me.dals..*.*(..)) "/>
                
                <aop:before pointcut-ref="tx" method="before" />                
            </aop:aspect>    
        </aop:config>  
    </beans>

六、调用测试

package me.dals;
    
    
    import java.util.List;
    
    
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.stereotype.Repository;
    import org.springframework.transaction.annotation.Propagation;
    import org.springframework.transaction.annotation.Transactional;
    import org.springframework.transaction.interceptor.TransactionAspectSupport;
    
    
    import me.daos.UserInfoDao;
    import me.exceptions.RollbackException;
    import me.models.UserInfoModel;
    
    
    /** 
    *
    * @version 创建时间：2017年4月11日 下午4:16:54
    */
    @Repository
    public class UserInfoDal {
    	/**
    	 * 注入 UserInfoDao 实例
    	 */
    	@Autowired
    	UserInfoDao dao;
    	/**
    	 * 注入 SqlSessionTemplate 实例
    	 */
    	@Autowired
    	org.mybatis.spring.SqlSessionTemplate sqlSession;
    	
    	/**
    	 * 使用接口形式查询数据库
    	 * @return
    	 */
    	@me.utils.database.DataSource(value="write")
    	public List<UserInfoModel> getAllUseMapper() {
    		List<UserInfoModel> list = dao.getAll();
    		return list;
    	}
    	/**
    	 * 使用 sqlsession 形式查询数据库
    	 * @return
    	 */
    	@me.utils.database.DataSource(value="read")
    	public List<UserInfoModel> getAllUseSqlSession() {
    		List<UserInfoModel> list = sqlSession.selectList("me.daos.UserInfoDao.getAll");
    		
    		return list;
    	}
    }