转载

动态切换数据源原理

动态切换数据源原理

有一种场景,调用ServiceA时,使用A数据源,调用ServiceB时,使用B数据源。在不改变原有代码的情况下,这个如何来实现呢。

下面是一种实现方式,供参考:

  • 将各个数据源用key-value形式保存到map中,如下面这个类:
class RouteDatasource {
	Map<String, Datasource> map = new HashMap<>();
	
	map.push("goods", goodsDatasource);
	map.push("order", orderDatasource);
	
	
	获取Connection
	public Connection getConnection() {
		...
	}

}
  • 在操作某个Service类的时候选择对应的数据源

比如在操作com.project.service.goods.GoodsService.java时,key应该设置成goods。这里有个问题是,service类跟数据源是分开的,怎么样才能通知到

RouteDatasource类使用goods的key呢。

方法简单,使用 ThreadLocal 类,在使用GoodsService类时,ThreadLocal中放入key,然后RouteDatasource类从ThreadLocal中取。

class RouteDatasource {
	Map<String, Datasource> map...
	
	map.push("goods", goodsDatasource);
	map.push("order", orderDatasource);
	
	获取Connection
	public Connection getConnection() {
		String key = ThreadLocal.get();
		Datasource ds = map.get(kay);
		return ds.getConnection();
	}

}

获取key完成了,如何设置key呢?不可能在每次使用前put一下吧。容易出错,代码也不美观。有一种方案是使用spring的AOP方式实现。

AOP可以在执行某个Service方法前做一些操作,那么我们就把设置key的操作放在AOP的before行为中。

大致思路是:判断当前方法所在类的package的名称来区分需要设置哪个key。

比如:com.project.service.goods.GoodsService.save()方法的package,是com.project.service.goods开头

那么AOP代码可以这样写

String methodInfo = ...
if(methodInfo.startWith("com.project.service.goods")) {
	ThreadLocal.set("goods")
}

完整的代码如下:

@Component
@Aspect
public class DynamicDataSourceAspect {
	
	@Before("execution(* com.project.service..*.*(..))")
	public void before(JoinPoint jp) {
		String methodName = jp.getSignature().getDeclaringTypeName();
		if (StringUtils.isBlank(methodName)) {
			return;
		}
		if (methodName.startsWith("com.project.service.goods")) {
			ThreadLocal.set("goods")
		} else if (methodName.startsWith("com.project.service.order")) {
			ThreadLocal.set("order")
		}

	}

}

这样,以后只需要配置DynamicDataSourceAspect类即可。

数据源动态切换,主要思想是通过事先定义好的Key-Value进行动态获取。主要使用了ThreadLocal和Spring AOP方式去串连起来。

spring为我们提供了数据源切换类 org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource ,这个类的作用就是一开始提到的RouteDatasource类。感兴趣的同学可以看下代码,很简单。

原文  https://blog.csdn.net/thc1987/article/details/86714562
正文到此结束
Loading...