java 主从表维护_Java Mybatis实现主从同步

import java.util.Locale;

import java.util.Properties;

import java.util.concurrent.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.Interceptor;

import org.apache.ibatis.plugin.Invocation;

import org.apache.ibatis.plugin.Plugin;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.transaction.support.TransactionSynchronizationManager;

/**

* @author wls

*

*/

@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 Logger logger=LoggerFactory.getLogger(DynamicDataSourceInterceptor.class);

private static final String REGEX=

".*insert\\u0020.*|.*delete\\u0020.*|.8update\\u0020.*";

@Override

public Object intercept(Invocation invocation) throws Throwable {

//判断当前是不是事物

boolean synchronizatioinActive=TransactionSynchronizationManager.

isActualTransactionActive();

Object[] objects= invocation.getArgs();

MappedStatement ms=(MappedStatement)objects[0];

String lookupKey=DynamicDataSourceHolder.DB_MASTER;

if(synchronizatioinActive!=true){

//读方法

if(ms.getSqlCommandType().equals(SqlCommandType.SELECT)){

//SELECT_KEY为自增id查询主键SELECT LAST_INSERT_ID()，使用主库

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

lookupKey=DynamicDataSourceHolder.DB_MASTER;

}else{

BoundSql boundSql=ms.getSqlSource().getBoundSql(objects[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,SqlCommanType[{}]" ,ms.getId(),lookupKey,

ms.getSqlCommandType().name());;

DynamicDataSourceHolder.setDbType(lookupKey);

return invocation.proceed();

}

@Override //返回封装好的对象

public Object plugin(Object target) {

// TODO Auto-generated method stub

if(target instanceof Executor){

return Plugin.wrap(target, this);

}else{

return target;

}

}

@Override

public void setProperties(Properties arg0) {

// TODO Auto-generated method stub

}

}