数据库读写分离（aop方式完整实现）

最近项目要支持读写分离， 网上找了很多，但都是不太完整，我自己整理了下供大家参考。
我的项目使用的框架: springMvc+spring+hibernate+springJPA+maven， 数据库连接池用阿里的druid。

1. 新建一个DynamicDataSource类， 继承spring的AbstractRoutingDataSource 类，

并重写determineCurrentLookupKey()方法，如下： 

package com.dataSource;

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

/**
 * 数据源动态切换类
 * <p>
 * Copyright: Copyright (c) 2015-3-9 下午3:15:19
 * <p>
 * Company: 
 * <p>
 * 
 * @author macl@c-platform.com
 * @version 1.0.0
 */
public class DynamicDataSource extends AbstractRoutingDataSource {

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

}

2. 新建DataSourceSwitcher类， 实现数据源选择：

package com.dataSource;

import org.springframework.util.Assert;

/**
 * 数据源选择类
 * <p>
 * Copyright: Copyright (c) 2015-3-9 下午3:14:55
 * <p>
 * Company:  
 * <p>
 * 
 * @author macl@c-platform.com
 * @version 1.0.0
 */
public class DataSourceSwitcher {

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

	/** 主库（写库） **/
	public static final String MASTER_DATA_SOURCE = "master";

	/** 从库(读库) **/
	public static final String SLAVE_DATA_SOURCE = "slave";

	public static void setDataSource(String dataSource) {
		Assert.notNull(dataSource, "dataSource cannot be null");
		contextHolder.set(dataSource);
	}

	public static void setMaster() {
		clearDataSource();
	}

	public static void setSlave() {
		setDataSource(SLAVE_DATA_SOURCE);
	}

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

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

 

3. 新建切面类， 判断如果调用方法以query,list,get方法开头的， 切换为查询库： 

package com.dataSource;

import java.lang.reflect.Method;
import java.util.List;

import org.apache.log4j.Logger;
import org.springframework.aop.AfterReturningAdvice;
import org.springframework.aop.MethodBeforeAdvice;
import org.springframework.aop.ThrowsAdvice;

/**
 * 配置AOP切面类，动态切换读/写数据库。
 * <p>
 * Copyright: Copyright (c) 2015-3-9 下午3:16:51
 * <p>
 * Company:  
 * <p>
 * 
 * @author macl@c-platform.com
 * @version 1.0.0
 */
public class DataSourceAdvice implements MethodBeforeAdvice, AfterReturningAdvice, ThrowsAdvice {
 
	private static final Logger log = Logger.getLogger(DataSourceAdvice.class);

	private String logInfo;

	// 需要切换到从库(读库)的方法名前缀
	private List<String> slaveMethods;

	/**
	 * service方法执行之前被调用.
	 */
	@Override
	public void before(Method method, Object[] args, Object target) throws Throwable {
		logInfo = String.format("before切入点:%s-->%s(),切换到:", target.getClass().getName(), method.getName());

		String methodName = method.getName();
		boolean hasSwitchedSlave = false;

		for (String slaveMethod : slaveMethods) {
			if (methodName.startsWith(slaveMethod)) {
				if (log.isDebugEnabled()) {
					log.debug(logInfo + DataSourceSwitcher.SLAVE_DATA_SOURCE);
				}
				hasSwitchedSlave = true;
				DataSourceSwitcher.setSlave();
				break;
			}
		}

		if (!hasSwitchedSlave) {
			if (log.isDebugEnabled()) {
				log.debug(logInfo + "切换到:" + DataSourceSwitcher.MASTER_DATA_SOURCE);
			}
			DataSourceSwitcher.setMaster();
		}
	}

	/**
	 * service方法执行完之后被调用.
	 */
	@Override
	public void afterReturning(Object returnValue, Method method, Object[] args, Object target) throws Throwable {

	}

	/**
	 * 抛出Exception之后被调用。
	 * 
	 * @param method
	 * @param args
	 * @param target
	 * @param ex
	 * @throws Throwable
	 */
	public void afterThrowing(Method method, Object[] args, Object target, Exception ex) throws Throwable {
		logInfo = String.format("after throwing:%s类中%s方法,", target.getClass().getName(), method.getName());
		log.error(logInfo + "发生异常:" + ex.getMessage() + ",切换到:" + DataSourceSwitcher.SLAVE_DATA_SOURCE);
		DataSourceSwitcher.setSlave();
	}

	public List<String> getSlaveMethods() {
		return slaveMethods;
	}

	public void setSlaveMethods(List<String> slaveMethods) {
		this.slaveMethods = slaveMethods;
	}
}

4.  spring配置文件中， 配置数据源：

 

 

 

5. AOP配置（注意， 可以在service层切换（推荐）， 也可以在控制器层切换）：

截图为在控制层切换配置， 附件中的context.xml中给的是在service/dao层切换的配置方式。

6.  mvc层支持aop（如果aop切换配置在控制层， 则需要配置这步）:

7. 特殊类可以硬编码：

      切换到主库 ： DataSourceSwitcher.setMaster();

      比如 ， 登录拦截器RoleGenInterceptor.java --> genLocalSession()

 附件下载地址