MySQL读写分离应用层实现

柔情只为你懂 2022-05-15 04:49 172阅读 0赞

### 基于Spring的AOP ###

在应用层，实现读写分离  
在请求进入Service之前，使用AOP来做出判断

判断方式  
是使用写库还是读库，判断依据可以根据方法名判断  
比如，以query、find、get等开头的就走读库，其他的走写库  
![这里写图片描述][70]

### DynamicDataSource ###

定义动态数据源

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
    
    /** * 定义动态数据源，实现通过集成Spring提供的AbstractRoutingDataSource，只需要实现determineCurrentLookupKey方法即可 * * 由于DynamicDataSource是单例的，线程不安全的，所以采用ThreadLocal保证线程安全，由DynamicDataSourceHolder完成。 * */
    public class DynamicDataSource extends AbstractRoutingDataSource{ 
    
        @Override
        protected Object determineCurrentLookupKey() {
            // 使用DynamicDataSourceHolder保证线程安全，并且得到当前线程中的数据源key
            return DynamicDataSourceHolder.getDataSourceKey();
        }
    }

### DynamicDataSourceHolder ###

使用ThreadLocal记录数据源

/** * * 使用ThreadLocal技术来记录当前线程中的数据源的key * */
    public class DynamicDataSourceHolder { 
    
        //写库对应的数据源key
        private static final String MASTER = "master";
    
        //读库对应的数据源key
        private static final String SLAVE = "slave";
    
        //使用ThreadLocal记录当前线程的数据源key
        private static final ThreadLocal<String> holder = new ThreadLocal<String>();
    
        /** * 设置数据源key * @param key */
        public static void putDataSourceKey(String key) {
            holder.set(key);
        }
    
        /** * 获取数据源key * @return */
        public static String getDataSourceKey() {
            return holder.get();
        }
    
        /** * 标记写库 */
        public static void markMaster(){
            putDataSourceKey(MASTER);
        }
    
        /** * 标记读库 */
        public static void markSlave(){
            putDataSourceKey(SLAVE);
        }
    }

### DataSourceAspect ###

定义数据源的AOP切面

import org.apache.commons.lang3.StringUtils;
    import org.aspectj.lang.JoinPoint;
    
    /** * 定义数据源的AOP切面，通过该Service的方法名判断是应该走读库还是写库 * */
    public class DataSourceAspect { 
    
        /** * 在进入Service方法之前执行 * * @param point 切面对象 */
        public void before(JoinPoint point) {
            // 获取到当前执行的方法名
            String methodName = point.getSignature().getName();
            if (isSlave(methodName)) {
                // 标记为读库
                DynamicDataSourceHolder.markSlave();
            } else {
                // 标记为写库
                DynamicDataSourceHolder.markMaster();
            }
        }
    
        /** * 判断是否为读库 * * @param methodName * @return */
        private Boolean isSlave(String methodName) {
            // 方法名以query、find、get开头的方法名走从库
            return StringUtils.startsWithAny(methodName, "query", "find", "get");
        }
    }

### 数据源 ###

配置2个数据源  
jdbc.properties

jdbc.master.driver=com.mysql.jdbc.Driver
    jdbc.master.url=jdbc:mysql://127.0.0.1:3306/mybatis_1128?useUnicode=true&characterEncoding=utf8&autoReconnect=true&allowMultiQueries=true
    jdbc.master.username=root
    jdbc.master.password=123456
    
    jdbc.slave01.driver=com.mysql.jdbc.Driver
    jdbc.slave01.url=jdbc:mysql://127.0.0.1:3307/mybatis_1128?useUnicode=true&characterEncoding=utf8&autoReconnect=true&allowMultiQueries=true
    jdbc.slave01.username=root
    jdbc.slave01.password=123456

### 定义连接池 ###

applicationContext.xml

<bean id="masterDataSource" class="com.jolbox.bonecp.BoneCPDataSource" destroy-method="close">
        
        <property name="driverClass" value="${jdbc.master.driver}" />
        
        <property name="jdbcUrl" value="${jdbc.master.url}" />
        
        <property name="username" value="${jdbc.master.username}" />
        
        <property name="password" value="${jdbc.master.password}" />
        
        <property name="idleConnectionTestPeriod" value="60" />
        
        <property name="idleMaxAge" value="30" />
        
        <property name="maxConnectionsPerPartition" value="150" />
        
        <property name="minConnectionsPerPartition" value="5" />
    </bean>
    
    
    <bean id="slave01DataSource" class="com.jolbox.bonecp.BoneCPDataSource" destroy-method="close">
        
        <property name="driverClass" value="${jdbc.slave01.driver}" />
        
        <property name="jdbcUrl" value="${jdbc.slave01.url}" />
        
        <property name="username" value="${jdbc.slave01.username}" />
        
        <property name="password" value="${jdbc.slave01.password}" />
        
        <property name="idleConnectionTestPeriod" value="60" />
        
        <property name="idleMaxAge" value="30" />
        
        <property name="maxConnectionsPerPartition" value="150" />
        
        <property name="minConnectionsPerPartition" value="5" />
    </bean>

### DataSource ###

定义数据源

<bean id="dataSource" class="cn.itcast.usermanage.spring.DynamicDataSource">
        
        <property name="targetDataSources">
            <map key-type="java.lang.String">
                
                <entry key="master" value-ref="masterDataSource" />
                <entry key="slave" value-ref="slave01DataSource" />
            </map>
        </property>
        
        <property name="defaultTargetDataSource" ref="masterDataSource" />
    </bean>

### 事务管理 ###

定义事务管理器

<bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="dataSource" />
    </bean>

### 事务策略 ###

定义事务策略

<tx:advice id="txAdvice" transaction-manager="transactionManager">
        <tx:attributes>
            
            <tx:method name="query*" read-only="true" />
            <tx:method name="find*" read-only="true" />
            <tx:method name="get*" read-only="true" />
    
            
            <tx:method name="save*" propagation="REQUIRED" />
            <tx:method name="update*" propagation="REQUIRED" />
            <tx:method name="delete*" propagation="REQUIRED" />
    
            
            <tx:method name="*" />
        </tx:attributes>
    </tx:advice>

### 定义切面 ###

自定义数据源在前面  
先应该确定数据源，然后，开启事务  
确保，该切面在最前面

<bean class="cn.itcast.usermanage.spring.DataSourceAspect" id="dataSourceAspect" />
    <aop:config>
        
        <aop:pointcut id="txPointcut" expression="execution(* xx.xxx.xxxxxxx.service.*.*(..))" />
        
        <aop:advisor advice-ref="txAdvice" pointcut-ref="txPointcut" />
    
        
        <aop:aspect ref="dataSourceAspect" order="-9999">
            <aop:before method="before" pointcut-ref="txPointcut" />
        </aop:aspect>
    </aop:config>

[70]: /images/20220515/9dfc21d9ddfa4bf8abb354006ac95e02.png