SSM里关于mysql主从配置代码层实现

关于一些书面的东西我就不多写了，直接贴代码

首先

在主从库各增加一个用户并赋予权限

grant select,insert,update,delete on *.* to 'work'@'%' identified by '123456' with grant option;
这里设置%是为了能允许远程访问

刷新权限
flush privileges;

以下三个类，放在同一个包下。

我这里都是放在com.zyx.test.dao.split里的

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


public class DynamicDataSource extends AbstractRoutingDataSource{

	@Override
	protected Object determineCurrentLookupKey() {
		// TODO Auto-generated method stub
		return DynamicDataSourceHolder.getDbType();
	}
	
}

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

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";
	/**
	 * 获取
	 * @return
	 */
	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();
	}
}

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

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.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 = "updata", 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 loockupKey = DynamicDataSourceHolder.DB_MASTER;
		if (synchronizationActive != true) {
			// 读方法
			if (ms.getSqlCommandType().equals(SqlCommandType.SELECT)) {
				// selectKey为自增ID查询主键方法，使用主库
				if (ms.getId().contains(SelectKeyGenerator.SELECT_KEY_SUFFIX)) {
					loockupKey = 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)) {
						loockupKey = DynamicDataSourceHolder.DB_MASTER;
					} else {
						loockupKey = DynamicDataSourceHolder.DB_SLAVE;
					}
				}
			}
		} else {
			loockupKey = DynamicDataSourceHolder.DB_MASTER;
		}
		logger.debug("设置方法[{}] use [{}] Strategy,SqlCommanType [{}]..",
				ms.getId(), loockupKey, ms.getSqlCommandType().name());
		DynamicDataSourceHolder.setDbType(loockupKey);
		return invocation.proceed();
	}

	/**
	 * 返回代理对象,一旦检测到Executor就拦截 Executor有一切增删改查权限，需要给intercept拦截
	 */
	@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里增加一段配置


<plugins>
<plugin interceptor="com.zyx.test.dao.split.DynamicDataSourceInterceptor"></plugin>
</plugins>
这里interceptor="xxx"这里xxx里写的是你DynamicDataSourceInterceptor这个类放在哪里的

在spring-dao.xml里修改配置
<!-- 数据库连接池 -->
<bean id="abstractDataSource" abstract="true" class="com.mchange.v2.c3p0.ComboPooledDataSource" destory-method="close">
<!-- c3p0连接池的私有属性 -->
<property name="maxPoolSize" value="30" />
<property name="minPoolSize" value="10" />
<!-- 关闭连接后不自动commit -->
<property name="autoCommitOnClose" value="false" />
<!-- 获取连接超时时间 -->
<property name="checkoutTimeout" value="10000" />
<!-- 当获取连接失败重试次数 -->
<property name="acquireRetryAttempts" value="2" />
</bean>
增加以下内容
<bean id="master" parent="abstractDataSource">
<!-- 配置连接池属性 -->
<property name="driverClass" value="${jdbc.driver}" />
<property name="jdbcUrl" value="${jdbc.master.url}" />
<property name="user" value="${jdbc.username}" />
<property name="password" value="${jdbc.password}" />
</bean>


<bean id="slave" parent="abstractDataSource">
<!-- 配置连接池属性 -->
<property name="driverClass" value="${jdbc.driver}" />
<property name="jdbcUrl" value="${jdbc.slave.url}" />
<property name="user" value="${jdbc.username}" />
<property name="password" value="${jdbc.password}" />
</bean>
<!-- 配置动态数据源，也就是路由数据源对应的名称--!>
<bean id="dynamicDataSource" class="com.imooc.o2o.dao.split.DynamicDataSource">
<property name="targetDataSources">
<map>
<entry value-ref="master"key="master"></entry>
<entry value-ref="slave"key="slave"></entry>
</map>
</property>
</bean>
<bean id="dataSource"
class="org.springframework.jdbc.datasource.LazyConnectionDataSourceProxy">
<property name="targetDataSource">
<ref bean="dynamicDataSource"/>
</property>
</bean>


对jdbc.properties进行修改
jdbc.driver=com.mysql.jdbc.Driver
jdbc.master.url=jdbc:mysql://192.168.82.228:3306/o2odb?useUnicode=true&characterEncoding=utf8
jdbc.slave.url=jdbc:mysql://192.168.82.196:3306/o2odb?useUnicode=true&characterEncoding=utf8
jdbc.username=work
jdbc.password=123456

这样就ok了。测试的时候用JUnit.如果有一些问题。请自己查看错误信息去进行相应的修改。

本人直接拿来用是没问题的。如果你们出错了，请自行解决。我也不是很清楚