在上一篇文章中，我们实现了数据库的读写分离，主数据库发生改变，从数据库会跟着发生改变（https://www.jianshu.com/p/73e75c6abd6c）。这样主从数据库就保持一致了。那么在SSM框架中，我们如何实现读写分离呢？以下就为大家带来SSM框架读写分离的相关配置。
1.创建开一个继承AbstractRoutingDataSource抽象类的类
Spring框架中，为我们提供了一个多数据源的抽象类，叫做AbstractRoutingDataSource，其中有个方法如下：
protected DataSource determineTargetDataSource() {
Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");
// 该类中的一个方法，用来获取数据源
Object lookupKey = this.determineCurrentLookupKey();
DataSource dataSource = (DataSource)this.resolvedDataSources.get(lookupKey);
if (dataSource == null && (this.lenientFallback || lookupKey == null)) {
dataSource = this.resolvedDefaultDataSource;
}
if (dataSource == null) {
throw new IllegalStateException("Cannot determine target DataSource for lookup key [" + lookupKey + "]");
} else {
return dataSource;
}
}
从上面的这个方法中，我们可以看出这个方法会根据我们传入的lookupKey，从而判断是选择哪个数据源，而lookupKey是由该类的determineCurrentLookupKey()获取，所以，要实现多数据源，我们需要创建一个类，用来继承AbstractRoutingDataSource这个类，同时新建的这个类要重写该抽象类中的determineCurrentLookupKey方法。
在这里，为了保证代码的复用性以及新建的类的简单，我们新建了两个类，DynamicDataSource （继承AbstractRoutingDataSource这个抽象类），DynamicDataSourceHolder用来存放以及获取数据源。
package cn.xdw.o2o.dao.split;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
public class DynamicDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
return DynamicDataSourceHolder.getDbType();
}
}
package cn.xdw.o2o.dao.split;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class DynamicDataSourceHolder {
private static Logger logger = LoggerFactory.getLogger(DynamicDataSourceHolder.class);
private static ThreadLocal contextHolder = new ThreadLocal();
public static final String DB_MASTER = "master";
public static final String DB_SLAVE = "slave";
/**
* 获取数据库类型
* @return
*/
public static Object getDbType() {
String db = contextHolder.get();
if (db == null) {
return DB_MASTER;
}
return db;
}
/**
* 设置连接类型
* @param target
*/
public static void setDbType(String target) {
logger.debug("所使用的数据源为：" + target);
contextHolder.set(target);
}
/**
* 清除连接类型
*/
public static void clearDbType() {
contextHolder.remove();
}
}
DynamicDataSourceHolder 中使用了ThreadLocal，这个类是线程安全的，即上述代码中的context对象中保存的参数每个线程相互独立，不会产生影响。
2.创建mybatis拦截器，拦截增删改查操作
创建一个实现Intercepter接口的拦截器，对SQL进行拦截，代码如下：
package cn.xdw.o2o.dao.split;
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.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.transaction.support.TransactionSynchronizationManager;
import java.util.Locale;
import java.util.Properties;
@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 final Logger logger = LoggerFactory.getLogger(DynamicDataSourceInterceptor.class);
private static final String REGEX = ".*update\\u0020.*|.*insert\\u0020.*|.*delete\\u0020.*";
@Override
public Object intercept(Invocation invocation) throws Throwable {
// 首先判断是否是事务
boolean synchronizationActive = TransactionSynchronizationManager.isSynchronizationActive();
Object[] object = invocation.getArgs();
MappedStatement ms = (MappedStatement)object[0];
String lookUpkey = null;
// 如果没有开启事务
if (synchronizationActive != true) {
if (ms.getSqlCommandType().equals(SqlCommandType.SELECT)) {
// 如果是查询最后一条记录（select_last_record）,还是使用主库
if (ms.getId().contains(SelectKeyGenerator.SELECT_KEY_SUFFIX)) {
lookUpkey = DynamicDataSourceHolder.DB_MASTER;
} else {
lookUpkey = DynamicDataSourceHolder.DB_SLAVE;
}
} else {
BoundSql boundSql = ms.getSqlSource().getBoundSql(object[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[{}]..", ms.getId(),
lookUpkey, ms.getSqlCommandType().name());
DynamicDataSourceHolder.setDbType(lookUpkey);
return invocation.proceed();
}
@Override
public Object plugin(Object target) {
// Executor表示增删改查的操作，如果存在增删改查的操作，将其拦截下来
if (target instanceof Executor) {
return Plugin.wrap(target, this);
}
return target;
}
@Override
public void setProperties(Properties properties) {
}
}
代码中的plugin方法，主要作用时拦截增删改查的操作，如果不是增删改查操作，返回传入对象，如果是增删改查操作，返回数据库代理对象。代理对象的处理逻辑，就在intercept这个方法中，详细的逻辑在代码中有注释。
3.相关xml文件的配置
第一步，我们要在mybatis的xml文件中将拦截器配置进去
第二步： 将主从数据库相关信息，配置到我们的jdbc.properites文件当中
jdbc.driver=com.mysql.jdbc.Driver
jdbc.master.url=jdbc:mysql://localhost:3307/o2o?useUnicode=true&characterEncoding=utf8&serverTimezone=GMT%2B8&useSSL=false&allowPublicKeyRetrieval=true
jdbc.slave.url=jdbc:mysql://localhost:3308/o2o?useUnicode=true&characterEncoding=utf8&serverTimezone=GMT%2B8&useSSL=false&allowPublicKeyRetrieval=true
jdbc.username=root
jdbc.password=123456
第三步：在spring-dao.xml中配置数据源,将原先的id=datasource的bean替换为下面的几个bean。
至此，主从数据库的读写分离代码层配置完成！