ssm框架下的多数据源主从数据库配置

主从数据库SSM框架配置

原理：动态的使用不同的datasource我们就可以动态的使用不同的数据源，问题来了怎样动态的使用不同的datasource呢，----在spring中有一个类叫做AbstractRoutingDataSource，顾名思义叫做路由选择datasource，这个类继承AbstractDataSource，AbstractDataSource实现DataSource，因此可以使用AbstractRoutingDataSource来完成我们的需求。AbstractRoutingDataSource中有一个determineTargetDataSource方法，根据注释是用来决定目标数据源的，determineTargetDataSource方法中通过determineCurrentLookupKey来决定lookupKey，然后使用封装好了的map集合resolvedDataSources通过lookupKey为key值取得datasource，因此这里面最重要的就是determineCurrentLookupKey方法获取key值，这里spring大牛是把这个方法抽象出来让我们实现，这就是我们的目的接下来开始实现

1、第一步编写DynamicDataSource，DynamicDataSourceHolder，通过两个类相结合，动态的设置datasource

ssm框架下的多数据源主从数据库配置

public class DynamicDataSource extends AbstractRoutingDataSource{

@Override

protected Object determineCurrentLookupKey() {

// TODO Auto-generated method stub

return DynamicDataSourceHolder.getDbType();

}

ssm框架下的多数据源主从数据库配置

public class DynamicDataSourceHolder {

private static Logger logger=LoggerFactory.getLogger(DynamicDataSourceHolder.class);

private static ThreadLocal<String> contextHolder=new ThreadLocal<String>();

public static final String DB_MASTER="master";

public static final String DB_SLAVE="slave";

public static String getDbType(){

String db=contextHolder.get();

if(db==null)

db=DB_MASTER;

return db;

}

/**

* 设置线程的DbType

* @param str

public static void setDbType(String str){

logger.debug("所使用的数据源为"+str);

contextHolder.set(str);

}

/**

* 清理连接类型

public static void clearDBType(){

contextHolder.remove();

}

2、编写拦截器，并配置拦截器，继承Interceptor分别实现三个方法setProperties方法可以不管，默认就行

plugin方法：因为mybatis底层所做的操作都是Executor进行操作的，因此我们只要判定是否是Executor的类型的执行器就行了（注意：这里的Executor的权限类名org.apache.ibatis.executor.Executor 楼主因为这里被折磨不小），然后进行拦截

intercept方法：这个方法是相当重要的，这个方法通过匹配SqlCommandType类型判断使用不同的数据源，并且这里使用日志功能输出在控制台

@Intercepts注解：在mybatis的底层并不是select，update，insert，delete四个方法，而是把update，insert，delete三个方法都变为update方法，因此我们只需要写两个@signature即可

ssm框架下的多数据源主从数据库配置

import java.util.Locale;

import java.util.Properties;

import org.apache.ibatis.executor.Executor;

import org.apache.ibatis.executor.******.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.Intercepts;

import org.apache.ibatis.plugin.Invocation;

import org.apache.ibatis.plugin.Plugin;

import org.apache.ibatis.plugin.Signature;

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;

@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.*|.*update\\u0020.*";

@Override

public Object intercept(Invocation invocation) throws Throwable {

// TODO Auto-generated method stub

boolean synchronizationActive=TransactionSynchronizationManager.isActualTransactionActive();

Object[] objects=invocation.getArgs();

MappedStatement ms=(MappedStatement) objects[0];

String lookupKey=DynamicDataSourceHolder.DB_MASTER;

if(synchronizationActive!=true)

{

//读方法

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

{

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[{}]startegy,SqlCommanType[{}]..",ms.getId(),lookupKey,ms.getSqlCommandType());

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

}

将拦截器配置在mybatis-config.xml文件中

ssm框架下的多数据源主从数据库配置

</plugin>

</plugins>

3、配置数据源，把公共属性配置在abstractDataSource中，然后配置主库从库数据源，并继承abstractDataSource，配置动态数据源，这儿targetDataSources就是路由数据源对应的名称，编写最终datasource，属性targetDataSource引用dynamicDataSource，大功告成

ssm框架下的多数据源主从数据库配置

数据库连接配置，db.properties

ssm框架下的多数据源主从数据库配置

<bean id="abstractDataSource" abstract="true" class="com.mchange.v2.c3p0.ComboPooledDataSource"

destroy-method="close">

</bean>

</bean>

</bean>

<map>

</map>

</property>

</bean>

</property>

</bean>

ssm框架下的多数据源主从数据库配置

相关推荐