Datasource动态切换

spring 使用AbstractRoutingDataSource自定义动态数据源时的事务处理，有需要的朋友可以参考下。

最近在网上看到了一篇博客，继承spring的AbstractRoutingDataSource定义自己的动态数据源，可以根据需要动态的切换不同数据库的数据源，感觉使用起来非常方便。

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
/**
 * 自定义的动态路由数据源 继承自 spring jdbc的AbstractRoutingDataSource
 * 
 * @author 
 *
 */
public class DbRouteDataSource extends AbstractRoutingDataSource {

	/**
	 * 获取与数据源相关的key
	 * 此key是Map resolvedDataSources 中与数据源绑定的key值
	 * 在通过determineTargetDataSource获取目标数据源时使用
	 */
	@Override
	protected Object determineCurrentLookupKey() {
		
		return RouteHolder.getRouteKey();
	}

}

通过容器RouteHolder存储当前线程使用的数据源的key

/**
 * 保存当前线程数据源的key
 * @author 
 * @version 1.0
 *
 */
public class RouteHolder {
	private static ThreadLocal routeKey = new ThreadLocal();
	
	/**
	 * 获取当前线程的数据源路由的key
	 * @return
	 */
	public static String getRouteKey()
	{
		String key = routeKey.get();
		return key;
	}
	/**
	 * 绑定当前线程数据源路由的key
	 * 使用完成后必须调用removeRouteKey()方法删除
	 * @param key
	 */
	public static void  setRouteKey(String key)
	{
		routeKey.set(key);
	}
	
	/**
	 * 删除与当前线程绑定的数据源路由的key
	 */
	public static void removeRouteKey()
	{
		routeKey.remove();
	}
}

使用spring 的aop编程在业务逻辑方法运行前将当前方法使用数据源的key从业务逻辑方法上自定义注解@DataSource中解析数据源key并添加到RouteHolder中

/**
 * 执行dao方法之前的切面
 * 获取datasource对象之前往RouteHolder中指定当前线程数据源路由的key
 * @author Administrator
 *
 */
public class DataSourceAspect
{
	/**
	 * 在dao层方法之前获取datasource对象之前在切面中指定当前线程数据源路由的key
	 */
	public void beforeDaoMethod(JoinPoint point)
	{
		//dao方法上配置的注解
		DataSource datasource = ((MethodSignature)point.getSignature()).getMethod().getAnnotation(DataSource.class);
		RouteHolder.setRouteKey(datasource.value());
		
	}
}

业务逻辑方法

@Named("userService")
public class UserService 
{
	@Inject
	private UserDao userDao;
	
	@DataSource("master")
	@Transactional(propagation=Propagation.REQUIRED)
	public void updatePasswd(int userid,String passwd)
	{
		User user = new User();
		user.setUserid(userid);
		user.setPassword(passwd);
		userDao.updatePassword(user);
	}
	@DataSource("slave")
	@Transactional(propagation=Propagation.REQUIRED)
	public User getUser(int userid)
	{
		User user = userDao.getUserById(userid);
		System.out.println("username------:"+user.getUsername());
		return user;
	}
}

spring的配置文件

事务管理配置一定要配置在往RouteHolder中注入数据源key之前 否则会报 Could not open JDBC Connection for transaction; nested exception is java.lang.IllegalStateException: Cannot determine target DataSource for lookup key [null] 找不到数据源错误

最后测试业务逻辑层的方法发现 可以根据方法上的@DataSource("master") 注解配置不同的数据源key 使用动态数据源。 不需要再业务逻辑方法层中定义不同的数据源对象，人为的使用不同的数据源操作数据。