Spring多数据源管理实现原理
应用场景:
大部分单一架构项目连接一台数据库服务器,但随着业务的增加数据库数据量不断飙升,数据库达到性能瓶颈,大部分技术人员都会对数据库主从配置;既然读写分离那就需要连接两个不同的数据库,这时候Spring多数据源管理类AbstractRoutingDataSource就要派上用场了(排除使用数据库集群管理工具统一管理的应用场景)
源码分析:
通过源码可以看出该类是一个抽象类,定义了6个属性。
targetDataSources:是一个map类型该属性正是用来维护项目中多个数据源
defaultTargetDataSource:通过属性名很直观的可以理解它的作用(默认数据源)
lenientFallback:默认为true,无需改动
dataSourceLookup:查找数据源接口的名称
该类是一个interface并且只有一个方法getDataSource,通过方法的参数名称应该清楚传入一个字符类型的数据源名称获取DataSource
resolvedDataSources:如果该字段没有赋值,就是targetDataSources
resolvedDefaultDataSource:改变后的数据源
深入理解:
使用数据源的目的就是要获取Connection,接下来就从AbstractRoutingDataSource的getConnection方法一探究竟。
直接进入determineTargetDataSource方法
该方法的作用是检索目标数据源
第一句代码
Assert.notNull(this.resolvedDataSources, “DataSource router not initialized”);
判断resolvedDataSources是否为空,为空就抛出异常。
通过afterPropertiesSet方法看出如果resolvedDataSources为空就遍历targetDataSources并把对应的key,value put到resolvedDataSources
第二句代码
Object lookupKey = determineCurrentLookupKey();
进入determineCurrentLookupKey方法
该方法是一个抽象方法,返回要从resolvedDataSources查找key,该方法还会实现检查线程绑定事务上下文。
第三局代码
DataSource dataSource = this.resolvedDataSources.get(lookupKey);
从resolvedDataSources中取出数据源并返回
代码实现
实现AbstractRoutingDataSource重写determineCurrentLookupKey
定义DataSourceContextHolder
通过ThreadLocal类使每个线程获取独立的数据源,防止并发访问时获取错误的数据源
基于自定义注解+aop实现数据源动态切换
自定义注解
定义aop实现类
Spring配置文件中配置自定义DataSource类DynamicDataSource
所有工作都已完成,只需要在Dao层方法上添加DataSource注解即可
没有加DataSource注解的方法都会使用配置文件中配置的默认数据源
启示
在开发中如果数据库事物是在Service层,这时在Dao层中即使添加DataSoure注解也会使用默认数据源,请各位开发者注意!!!原因:请认真查看源码分析中第二句代码部分的内容