Mybatis快速实现读写分离

首先我们来了简单介绍一下mybatis的架构图

mybatis 通过 解析 mybatis.xml 拿到 configuration 返回一个SqlSessionFactory 

然后通过 SqlSessionFactory 拿到一个SqlSession

 注意这里 创建 Excutor的同时把他加入到了 intreceptorChain 里面。这里用到了观察者模式 类似于 zookeeper 监听机制

代码细节如下

这里注意的是Mybatis拦截器只能拦截Executor、ParameterHandler、StatementHandler、ResultSetHandler四个对象里面的方法。

重点来了，我们实现数据库读写分离也是基于拦截器来实现的。这里拦截的是 Executor.代码如下 

import com.gooagoo.crm.datasources.DataSourceNames;
import com.gooagoo.crm.datasources.DynamicDataSource;
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.*;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
import org.springframework.transaction.support.TransactionSynchronizationManager;

import java.util.Locale;
import java.util.Properties;

//指定拦截哪些方法,update包括增删改
@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 DataSourceInterceptor implements Interceptor{
    private static final String REGEX=".*insert\\u0020.*|.*delete\\u0020.*|.*update\\u0020.*";
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        boolean synchronizationActive= TransactionSynchronizationManager.isActualTransactionActive();
        String lookupKey= DataSourceNames.DB_MASTER;
        if(!synchronizationActive){
            Object[] objects=invocation.getArgs();
            MappedStatement ms=(MappedStatement)objects[0];
            if(ms.getSqlCommandType().equals(SqlCommandType.SELECT)){
                BoundSql boundSql=ms.getSqlSource().getBoundSql(objects[1]);
                String sql=boundSql.getSql().toLowerCase(Locale.CHINA).replaceAll("[\\t\\n\\r]"," ");
                if(sql.matches(REGEX)){
                    lookupKey= DataSourceNames.DB_MASTER;
                }else{
                    //这里如果有多个从数据库，则添加挑选过程
                    lookupKey= DataSourceNames.DB_SLAVE;
                }
            }
        }else{
            lookupKey=DataSourceNames.DB_MASTER;
        }
        DynamicDataSource.setDataSource(lookupKey);
        return invocation.proceed();
    }

    @Override
    public Object plugin(Object target) {
        //增删改查的拦截，然后交由intercept处理
        if(target instanceof Executor){
            return Plugin.wrap(target,this);
        }else{
            return target;
        }
    }

    @Override
    public void setProperties(Properties properties) {

    }
}

这里面有意思的是 Executor里面 没有 add  或者 save方法。 也不知道是不是有意为之。

public interface DataSourceNames {
    public static final String DB_MASTER = "master";
    public static final String DB_SLAVE = "slave";

}

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

import javax.sql.DataSource;
import java.util.Map;

/**
 * 作者：
 * 更新日期：2019-10-29 14:18
 * 方法描述：配置数据源
 * 审查者：
 * 审查意见：
 */
public class DynamicDataSource extends AbstractRoutingDataSource {
    private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();

    @Override
    protected Object determineCurrentLookupKey() {
        return getDataSource();
    }

    public static void setDataSource(String dataSource) {
        contextHolder.set(dataSource);
    }

    public static String getDataSource() {
        return contextHolder.get();
    }

    public static void clearDataSource() {
        contextHolder.remove();
    }

}

再然后就是 xml 文件了 crmWriteDataSource  和 crmReadDataSource 指向两个不同的数据源即可

<bean id="dynamicDataSource" class="com.gooagoo.crm.datasources.DynamicDataSource">
   <property name="targetDataSources">
      <map key-type="java.lang.String">
         <entry value-ref="crmWriteDataSource" key="master"></entry>
         <entry value-ref="crmReadDataSource" key="slave"></entry>
      </map>
   </property>
   <!-- 默认使用server的数据源 -->
   <property name="defaultTargetDataSource" ref="crmWriteDataSource"></property>
</bean>