springboot多数据源使用

多数据源

一、多数据源的典型使用场景
在实际开发中,经常可能遇到在一个应用中可能需要访问多个数据库的情况。以下是两种典型场景:
1 业务复杂(数据量大)

数据分布在不同的数据库中,数据库拆了, 应用没拆。 一个公司多个子项目,各用各的数据库,涉及数据共享…
springboot多数据源使用_第1张图片

2 读写分离
为了解决数据库的读性能瓶颈(读比写性能更高, 写锁会影响读阻塞,从而影响读的性能)。 很多数据库拥主从架构。也就是,一台主数据库服务器,是对外提供增删改业务的生产服务器;另一(多)台从数据库服务器,主要进行 读的操作。
可以通过中间件(ShardingSphere、mycat、mysql-proxy 、TDDL …), 但是有一些规模较小的公司,没有专门的中间件团队搭建读写分 离基础设施,因此需要业务开发人员自行实现读写分离。
springboot多数据源使用_第2张图片
二、 如何实现多数据源
原理:
对于大多数的java应用,都使用了spring框架,spring-jdbc模块提供了AbstractRoutingDataSource,其内部可以包含了多个 DataSource,然后在运行时来动态的访问哪个数据库。这种方式访问数据库的架构图如下所示:
springboot多数据源使用_第3张图片
应用直接操作的是AbstractRoutingDataSource的实现类,告诉AbstractRoutingDataSource访问哪个数据库,然后由 AbstractRoutingDataSource从事先配置好的数据源(ds1、ds2)选择一个,来访问对应的数据库。
springboot多数据源使用_第4张图片
1.当执行数据库持久化操作,只要集成了Spring就一定会通过DataSourceUtils获取Connection
2. 通过Spring注入的DataSource获取Connection 即可执行数据库操作
所以思路就是:只需配置一个实现了DataSource的Bean, 然后根据业务动态提供Connection即可 3.其实Spring已经提供一个DataSource实现类用于动态切换数据源——AbstractRoutingDataSource 4.分析AbstractRoutingDataSource即可实现动态数据源切换:

2.1、通过AbstractRoutingDataSource实现动态数据源
通过这个类可以实现动态数据源切换。如下是这个类的成员变量

	private Map<Object, Object> targetDataSources;
    private Object defaultTargetDataSource;
  	private Map<Object, DataSource>  resolvedDataSources;

targetDataSources保存了key和数据库连接的映射关系
defaultTargetDataSource标识默认的连接 resolvedDataSources这个数据结构是通过targetDataSources构建而来,存储结构也是数据库标识和数据源的
映射关系

而AbstractRoutingDataSource实现了InitializingBean接口,并实现了afterPropertiesSet方法。afterPropertiesSet方法 是初始化bean的时候执行,通常用作数据初始化。
resolvedDataSources就是在这里赋值

springboot多数据源使用_第5张图片
5.so! 我们只需创建AbstractRoutingDataSource实现类DynamicDataSource然后 始化targetDataSources和key为 数据源标识(可以是字符串、枚举、都行,因为标识是Object)、defaultTargetDataSource即可
6.后续当调用AbstractRoutingDataSource.getConnection 会接着调用提供的模板方法:
determineTargetDataSource
7.通过determineTargetDataSource该方法返回的数据库标识 从resolvedDataSources 中拿到对应的数据源
8.so!我们只需DynamicDataSource中实现determineTargetDataSource为其提供一个数据库标识

总结: 在整个代码中我们只需做4件大事:
1.定义AbstractRoutingDataSource实现类DynamicDataSource
2.初始化时为targetDataSources设置 不同数据源的DataSource和标识、及defaultTargetDataSource
3.在determineTargetDataSource中提供对应的数据源标识即可 4、切换数据源识即

什么到这还不会? 附上代码:

  1. 配置多数据源 和 AbstractRoutingDataSource的自定义实现类:DynamicDataSource
    配置多数据
    springboot多数据源使用_第6张图片
    springboot多数据源使用_第7张图片
    springboot多数据源使用_第8张图片
    DynamicDataSource 代码:
1  public class DynamicDataSource extends AbstractRoutingDataSource  {
2
3	/**
4	* ThreadLocal 用于提供线程局部变量,在多线程环境可以保证各个线程里的变量独立于其它线程里的变量。
5	* 也就是说 ThreadLocal 可以为每个线程创建一个【单独的变量副本】,相当于线程的 private static 类型变量。
6	*/
7	private static final ThreadLocal<String> CONTEXT_HOLDER = new  ThreadLocal<>();
8
9	/**
10	* 决定使用哪个数据源之前需要把多个数据源的信息以及默认数据源信息配置好
11	*
12	* @param defaultTargetDataSource 默认数据源
13	* @param targetDataSources 目标数据源
14	*/
15	public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object,  Object> targetDataSources) {
16	super.setDefaultTargetDataSource(defaultTargetDataSource);
17	super.setTargetDataSources(targetDataSources);
18	super.afterPropertiesSet();
19	}
20
21	@Override
22	protected Object determineCurrentLookupKey()  {
23	return getDataSource();
24	}
25
26	public static void setDataSource(String dataSource)  {
27	CONTEXT_HOLDER.set(dataSource);
28	}
29
30	public static String getDataSource() {
31	return CONTEXT_HOLDER.get();
32	}
33
34	public static void clearDataSource() {
35	CONTEXT_HOLDER.remove();
36	}
37
38 }

2.2、多数据源切换方式
切换方式看你具体需求:
2.2.1、AOP+自定义注解
不同业务的数据源: 一般利用AOP,结合自定义注解动态切换数据源:
1.自定义注解

1   @Target({ElementType.METHOD,ElementType.TYPE})
2  @Retention(RetentionPolicy.RUNTIME)
3  public @interface WR {
4	String value() default "W";
5 }

2.切面类

1
2  @Component
3  @Aspect
4  public class DynamicDataSourceAspect {
5
6	// 前置通知
7	@Before("within(com.tuling.dynamic.datasource.service.impl.*)  && @annotation(wr)")
8	public void before(JoinPoint joinPoint, WR  wr){
9	System.out.println(wr.value());
10	}
11
12 }

3.使用注解

1  @Service
2  public class FrendImplService implements FrendService  {
3
4	@Autowired
5	FrendMapper frendMapper;
6

7
8	@Override
9	@WR("R") // 库2
10	public List<Frend> list() {
11	return frendMapper.list();
12	}
13
14	@Override
15	@WR("W") // 库1
16	public void save(Frend frend) {
17	frendMapper.save(frend);
18	}
19 }

2.2.2、MyBatis插件
读写分离的数据源:如果是MyBatis可以结合插件实现读写分离动态切换数据源

1
2 @Intercepts({@Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Ob ject.class}),
3	@Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class,   R owBounds.class,
4	ResultHandler.class})})
5  public class DynamicDataSourcePlugin implements Interceptor  {
6

7
8	@Override
9	public Object intercept(Invocation invocation) throws Throwable  {
10
11	Object[] objects = invocation.getArgs();
12	MappedStatement ms = (MappedStatement)  objects[0];
13	// 读方法
14	if  (ms.getSqlCommandType().equals(SqlCommandType.SELECT)) {
15
16	DynamicDataSource.name.set("R");
17	} else {
18	// 写方法
19	DynamicDataSource.name.set("W");
20	}
21	// 修改当前线程要选择的数据源的key
22
23	return invocation.proceed();
24	}
25
26	@Override
27	public Object plugin(Object target) {
28	if (target instanceof Executor) {
29	return Plugin.wrap(target, this);
30	} else {
31	return target;
32	}
33	}
34
35	@Override
36	public void setProperties(Properties properties)  {
37
38	}
39 }

2.3、Spring集成多个MyBatis框架 实现多数据源
springboot多数据源使用_第9张图片
WDataSourceConfig

1 @MapperScan(basePackages = "com.tuling.dynamic.datasource.mapper.w", sqlSessionFactoryRef = "wSqlSe ssionFactory")
1 @Bean
2  @Primary
3   public SqlSessionFactory wSqlSessionFactory(@Qualifier("dataSource1") DataSource  dataSource1)
4	throws Exception {
5	final SqlSessionFactoryBean sessionFactory = new  SqlSessionFactoryBean();
6	sessionFactory.setDataSource(dataSource1);
7	sessionFactory.setMapperLocations(new  PathMatchingResourcePatternResolver()
8	.getResources("classpath:mapper/w/*.xml"));
9	/*主库设置sql控制台打印*/
10	org.apache.ibatis.session.Configuration configuration = new   org.apache.ibatis.session.Configurati n();
11	configuration.setLogImpl(StdOutImpl.class);
12	sessionFactory.setConfiguration(configuration);
13	return sessionFactory.getObject();
14 }

RDataSourceConfig

1 @MapperScan(basePackages = "com.tuling.dynamic.datasource.mapper.r", sqlSessionFactoryRef = "rSqlSe ssionFactory")
1 @Bean
2   public SqlSessionFactory rSqlSessionFactory(@Qualifier("dataSource2") DataSource  dataSource2)
3	throws Exception {
4	final SqlSessionFactoryBean sessionFactory = new  SqlSessionFactoryBean();
5	sessionFactory.setDataSource(dataSource2);
6	sessionFactory.setMapperLocations(new  PathMatchingResourcePatternResolver()
7	.getResources("classpath:mapper/r/*.xml"));
8	/*从库设置sql控制台打印*/
9	org.apache.ibatis.session.Configuration configuration = new   org.apache.ibatis.session.Configuratio n();
10	configuration.setLogImpl(StdOutImpl.class);
11	sessionFactory.setConfiguration(configuration);
12	return sessionFactory.getObject();
13 }

三、多数据源事务控制
在多数据源下,由于涉及到数据库的多个读写。一旦发生异常就可能会导致数据不一致的情况, 在这种情况希望使用事务 进行回退。
但是Spring的声明式事务在一次请求线程中只能使用一个数据源进行控制

但是对于多源数据库:
1.单一事务管理器(TransactionManager)无法切换数据源,需要配置多个TransactionManager。 2.@Transactionnal是无法管理多个数据源的。 如果想真正实现多源数据库事务控制,肯定是需要分布式事务。 这里讲解 多源数据库事务控制的一种变通方式。

1 @Bean
2   public DataSourceTransactionManager transactionManager1(DynamicDataSource  dataSource){
3	DataSourceTransactionManager dataSourceTransactionManager = new   DataSourceTransactionManager();
4	dataSourceTransactionManager.setDataSource(dataSource);
5	return dataSourceTransactionManager;
6 }
7
8 @Bean
9   public DataSourceTransactionManager transactionManager2(DynamicDataSource  dataSource){
10	DataSourceTransactionManager dataSourceTransactionManager = new   DataSourceTransactionManager();
11	dataSourceTransactionManager.setDataSource(dataSource);
12	return dataSourceTransactionManager;
13 }

1. 只使用主库TransactionManger
使用主库事务管理器,也就是说事务中产生异常时,只能回滚主库数据。但是因为数据操作顺序是先主后从,所以分一下三种情况: 1. 主库插入时异常,主库未插成功,这时候从库还没来及插入,主从数据是还是一致的
2. 主库插入成功,从库插入时异常,这时候在主库事务管理器监测到事务中存在异常,将之前插入的主库数据插入,主从数据还是 一致的
3. 主库插入成功,从库插入成功,事务结束,主从数据一致。

1  @Override
2  @WR("W")
3  public void save(Frend frend) {
4	frendMapper.save(frend);
5	//int a=1/0; 1.主库插入时异常,主库未插成功,这时候从库还没来及插入,主从数据是还是一致的
6 }
7
8  @Override
9  @WR("R")
10  @Transactional(
11	transactionManager = "transactionManager2",
12	propagation= Propagation.REQUIRES_NEW)
13  public void saveRead(Frend frend) {
14	frend.setName("xushu");
15	frendMapper.save(frend);
16	// int a=1/0; 2.主库插入成功,从库插入时异常,这时候在主库事务管理器监测到事务中存在异常,将之前插入的主库数 据插入,主从数据还是一致的
17 }
18
19  @Override
20   @Transactional(transactionManager = "transactionManager1")
21  public void saveAll(Frend frend) {
22	// 3. 无异常情况:主库插入成功,从库插入成功,事务结束,主从数据一致。
23	FrendService  self= (FrendService)AopContext.currentProxy();
24	self.save(frend);
25	self.saveRead(frend);
26	//int a=1/0; 从库插入之后出现异常, 只能回滚主库数据 ,从库数据是无法回滚的 , 数据将不一致
27 }

当然这只是理想情况,例外情况:
4.从库插入之后出现异常, 只能回滚主库数据 ,从库数据是无法回滚的 , 数据将不一致
5.从库数据插入成功后,主库提交,这时候主库崩溃了,导致数据没插入,这时候从库数据也是无法回滚的。这种方式可以简单实现多源数 据库的事务管理,但是无法处理上述情况。

  1. 一个方法开启2个事务
    spring编程式事务 :
1  // 读‐‐ 写库
2  @Override
3  public void saveAll(Frend frend) {
4	wtransactionTemplate.execute(wstatus ‐> {
5	rtransactionTemplate.execute(rstatus ‐> {
6	try{
7	saveW(frend);
8	saveR(frend);
9	int a=1/0;
10	return true;
11	}
12	catch (Exception e){
13	wstatus.setRollbackOnly();
14	rstatus.setRollbackOnly();
15	return false;
16	}
17	});
18	return true;
19	});
20 }

spring声明式事务:
@Transactional

1   @Transactional(transactionManager = "wTransactionManager")
2  public void saveAll(Frend frend) throws Exception  {
3	FrendService frendService = (FrendService)  AopContext.currentProxy();
4	frendService.saveAllR(frend);
5 }
6
7   @Transactional(transactionManager = "rTransactionManager"
8	,propagation = Propagation.REQUIRES_NEW )
9  public void saveAllR(Frend frend) {
10	saveW(frend);
11	saveR(frend); 12	int a = 1 / 0; 13 }

四、dynamic-datasource多数源组件
两三个数据源、事务场景比较少
基于 SpringBoot 的多数据源组件,功能强悍,支持 Seata 分布式事务。

支持 数据源分组 ,适用于多种场景 纯粹多库 读写分离 一主多从 混合模式。
支持数据库敏感配置信息 加密 ENC()。 支持每个数据库独立初始化表结构schema和数据库database。 支持无数据源启动,支持懒加载数据源(需要的时候再创建连接)。 支持 自定义注解 ,需继承DS(3.2.0+)。 提供并简化对Druid,HikariCp,BeeCp,Dbcp2的快速集成。
提供对Mybatis­Plus,Quartz,ShardingJdbc,P6sy,Jndi等组件的集成方案。
提供 自定义数据源来源 方案(如全从数据库加载)。
提供项目启动后 动态增加移除数据源 方案。
提供Mybatis环境下的 纯读写分离 方案。
提供使用 spel动态参数 解析数据源方案。内置spel,session,header,支持自定义。 支持 多层数据源嵌套切换 。(ServiceA >>> ServiceB >>> ServiceC)。
提供 基于seata的分布式事务方案
提供 本地多数据源事务方案。 附:不能和原生spring事务混用。

约定

  1. 本框架只做 切换数据源 这件核心的事情,并不限制你的具体操作,切换了数据源可以做任何CRUD。
  2. 配置文件所有以下划线 _ 分割的数据源 首部 即为组的名称,相同组名称的数据源会放在一个组下。
  3. 切换数据源可以是组名,也可以是具体数据源名称。组名则切换时采用负载均衡算法切换,默认是轮询的。
  4. 默认的数据源名称为 master ,你可以通过 spring.datasource.dynamic.primary 修改。
  5. 方法上的注解优先于类上注解。
  6. DS支持继承抽象类上的DS,暂不支持继承接口上的DS。
    使用方法
    引入dynamic­datasource­spring­boot­starter。
    maven­central v3.5.0
1  <dependency>
2	<groupId>com.baomidou</groupId>
3	<artifactId>dynamic‐datasource‐spring‐boot‐starter</artifactId>
4	<version>${version}</version>
5  </dependency>

2.配置数据源。

  spring:
2	datasource:
3	dynamic:
4	#设置默认的数据源或者数据源组,默认值即为master
5	primary: master
6	#严格匹配数据源,默认false. true未匹配到指定数据源时抛异常,false使用默认数据源
7	strict: false
8	datasource:
9	master:
10	url: jdbc:mysql://xx.xx.xx.xx:3306/dynamic
11	username: root
12	password: 123456
13	driver‐class‐name: com.mysql.jdbc.Driver #  3.2.0开始支持SPI可省略此配置
14	slave_1:
15	url: jdbc:mysql://xx.xx.xx.xx:3307/dynamic
16	username: root
17	password: 123456
18	driver‐class‐name: com.mysql.jdbc.Driver
19	slave_2:
20	url: ENC(xxxxx) # 内置加密,使用请查看详细文档
21	username: ENC(xxxxx)
22	password: ENC(xxxxx)
23	driver‐class‐name: com.mysql.jdbc.Driver
24
25	#......省略
26	#以上会配置一个默认库master,一个组slave下有两个子库slave_1,slave_2
1  # 多主多从 纯粹多库(记得设置primary) 混合配置
2  spring: spring: spring:
3	datasource: datasource: datasource:
4	dynamic: dynamic: dynamic:
5	datasource: datasource: datasource:
6	master_1: mysql: master:
7	master_2: oracle: slave_1:
8	slave_1: sqlserver: slave_2:
9	slave_2: postgresql: oracle_1:
10	slave_3: h2: oracle_2:

3.使用 @DS 切换数据源。
@DS 可以注解在方法上或类上,同时存在就近原则 方法上注解 优先于 类上注解。
在这里插入图片描述

1  @Service
2  @DS("slave")
3  public class UserServiceImpl implements UserService  {
4
5	@Autowired
6	private JdbcTemplate jdbcTemplate;
7
8	public List selectAll() {
9	return jdbcTemplate.queryForList("select * from  user");
10	}
11
12	@Override
13	@DS("slave_1")
14	public List selectByCondition() {
15	return jdbcTemplate.queryForList("select * from user where age  >10");
16	}
17 }

本地事务:

使用@DSTransactional即可, 不能和Spring@Transactional混用!

1 最外层的方法添加 @DSTransactional,底下调用的各个类该切数据源就正常使用DS切换数据源即可。 就是这么简单。~
2  //如AService调用BService和CService的方法,A,B,C分别对应不同数据源。
3
4  public class AService {
5
6	@DS("a")//如果a是默认数据源则不需要DS注解。
7	@DSTransactional
8	public void dosomething(){
9	BService.dosomething();
10	CService.dosomething();
11	}
12 }
13
14  public class BService {
15
16	@DS("b")
17	public void dosomething(){
18	//dosomething
19	}
20 }
21
22  public class CService {
23
24	@DS("c")
25	public void dosomething(){
26	//dosomething
27	}
28 }
29

只要@DSTransactional注解下任一环节发生异常,则全局多数据源事务回滚。 如果BC上也有@DSTransactional会有影响吗?答:没有影响的。

动态添加删除数据源:

通过DynamicRoutingDataSource 类即可,它就相当于我们之前自定义的那个DynamicDataSource

1

2
3  @RestController
4  @RequestMapping("/datasources")
5  @Api(tags = "添加删除数据源")
6  public class DataSourceController {
7
8	@Autowired
9	private DataSource dataSource;
10	// private final DataSourceCreator dataSourceCreator;  //3.3.1及以下版本使用这个通用
11	@Autowired
12	private  DefaultDataSourceCreator dataSourceCreator;
13	@Autowired
14	private  BasicDataSourceCreator basicDataSourceCreator;
15	@Autowired
16	private  JndiDataSourceCreator jndiDataSourceCreator;
17	@Autowired
18	private  DruidDataSourceCreator druidDataSourceCreator;
19	@Autowired
20	private  HikariDataSourceCreator hikariDataSourceCreator;
21	@Autowired
22	private  BeeCpDataSourceCreator beeCpDataSourceCreator;
23	@Autowired
24	private  Dbcp2DataSourceCreator dbcp2DataSourceCreator;
25
26	@GetMapping
27	@ApiOperation("获取当前所有数据源")
28	public Set<String> now() {
29	DynamicRoutingDataSource ds = (DynamicRoutingDataSource)  dataSource;
30	return ds.getCurrentDataSources().keySet();
31	}
32
33	//通用数据源会根据maven中配置的连接池根据顺序依次选择。
34	//默认的顺序为druid>hikaricp>beecp>dbcp>spring basic
35	@PostMapping("/add")
36	@ApiOperation("通用添加数据源(推荐)")
37	public Set<String> add(@Validated @RequestBody DataSourceDTO dto)  {
38	DataSourceProperty dataSourceProperty = new  DataSourceProperty();
39	BeanUtils.copyProperties(dto,  dataSourceProperty);
40	DynamicRoutingDataSource ds = (DynamicRoutingDataSource)  dataSource;
41	DataSource  dataSource  = dataSourceCreator.createDataSource(dataSourceProperty);
42	ds.addDataSource(dto.getPollName(),  dataSource);
43	return ds.getCurrentDataSources().keySet();
44	}
45
46	@PostMapping("/addBasic(强烈不推荐,除了用了马上移除)")
47	@ApiOperation(value = "添加基础数据源", notes =  "调用Springboot内置方法创建数据源,兼容1,2")
48	public Set<String> addBasic(@Validated @RequestBody DataSourceDTO dto)  {
49	DataSourceProperty dataSourceProperty = new  DataSourceProperty();
50	BeanUtils.copyProperties(dto,  dataSourceProperty);
51	DynamicRoutingDataSource ds = (DynamicRoutingDataSource)  dataSource;
52	DataSource  dataSource  = basicDataSourceCreator.createDataSource(dataSourceProperty);
53	ds.addDataSource(dto.getPollName(),  dataSource);
54	return ds.getCurrentDataSources().keySet();
55	}
56
57	@PostMapping("/addJndi")
58	@ApiOperation("添加JNDI数据源")
59	public Set<String> addJndi(String pollName, String jndiName)  {
60	DynamicRoutingDataSource ds = (DynamicRoutingDataSource)  dataSource;
61	DataSource  dataSource  = jndiDataSourceCreator.createDataSource(jndiName);
62	ds.addDataSource(pollName, dataSource);
63	return ds.getCurrentDataSources().keySet();
64	}
65
66	@PostMapping("/addDruid")
67	@ApiOperation("基础Druid数据源")
68	public Set<String> addDruid(@Validated @RequestBody DataSourceDTO dto)  {
69	DataSourceProperty dataSourceProperty = new  DataSourceProperty();
70	BeanUtils.copyProperties(dto,  dataSourceProperty);
71	dataSourceProperty.setLazy(true);
72	DynamicRoutingDataSource ds = (DynamicRoutingDataSource)  dataSource;
73	DataSource  dataSource  = druidDataSourceCreator.createDataSource(dataSourceProperty);
74	ds.addDataSource(dto.getPollName(),  dataSource);
75	return ds.getCurrentDataSources().keySet();
76	}
77
78	@PostMapping("/addHikariCP")
79	@ApiOperation("基础HikariCP数据源")
80	public Set<String> addHikariCP(@Validated @RequestBody DataSourceDTO dto)  {
81	DataSourceProperty dataSourceProperty = new  DataSourceProperty();
82	BeanUtils.copyProperties(dto,  dataSourceProperty);
83	dataSourceProperty.setLazy(true);//3.4.0版本以下如果有此属性,需手动设置,不然会空指针。
84	DynamicRoutingDataSource ds = (DynamicRoutingDataSource)  dataSource;
85	DataSource  dataSource  = hikariDataSourceCreator.createDataSource(dataSourceProperty);
86	ds.addDataSource(dto.getPollName(),  dataSource);
87	return ds.getCurrentDataSources().keySet();
88	}
89
90	@PostMapping("/addBeeCp")
91	@ApiOperation("基础BeeCp数据源")
92	public Set<String> addBeeCp(@Validated @RequestBody DataSourceDTO dto)  {
93	DataSourceProperty dataSourceProperty = new  DataSourceProperty();
94	BeanUtils.copyProperties(dto,  dataSourceProperty);
95	dataSourceProperty.setLazy(true);//3.4.0版本以下如果有此属性,需手动设置,不然会空指针。
96	DynamicRoutingDataSource ds = (DynamicRoutingDataSource)  dataSource;
97	DataSource  dataSource  = beeCpDataSourceCreator.createDataSource(dataSourceProperty);
98	ds.addDataSource(dto.getPollName(),  dataSource);
99	return ds.getCurrentDataSources().keySet();
100	}
101
102	@PostMapping("/addDbcp")
103	@ApiOperation("基础Dbcp数据源")
104	public Set<String> addDbcp(@Validated @RequestBody DataSourceDTO dto)  {
105	DataSourceProperty dataSourceProperty = new  DataSourceProperty();
106	BeanUtils.copyProperties(dto,  dataSourceProperty);
107	dataSourceProperty.setLazy(true);//3.4.0版本以下如果有此属性,需手动设置,不然会空指针。
108	DynamicRoutingDataSource ds = (DynamicRoutingDataSource)  dataSource;
109	DataSource  dataSource  = dbcp2DataSourceCreator.createDataSource(dataSourceProperty);
110	ds.addDataSource(dto.getPollName(),  dataSource);
111	return ds.getCurrentDataSources().keySet();
112	}
113
114	@DeleteMapping
115	@ApiOperation("删除数据源")
116	public String remove(String name) {
117	DynamicRoutingDataSource ds = (DynamicRoutingDataSource)  dataSource;
118	ds.removeDataSource(name);
119	return "删除成功";
120	}
121 }

原理
1.通过DynamicDataSourceAutoConfiguration自动配置类,
2. 配置了DynamicRoutingDataSource 它就相当于我们之前自定义的那个DynamicDataSource, 用来动态提供数据源
3. 配置DynamicDataSourceAnnotationAdvisor 就相当于 之前 自定义的一个切面类
4.设置DynamicDataSourceAnnotationInterceptor 当前advisor的拦截器 , 把它理解成之前环绕通知
4. 当执行方法会调用DynamicDataSourceAnnotationInterceptor#invoke 来进行增强:

1  // 获取当前方法的DS注解的value值
2  String dsKey =  determineDatasourceKey(invocation);
3  // 设置当当前数据源的标识TheardLocal中
4   DynamicDataSourceContextHolder.push(dsKey);
5  try {
6	// 执行目标方法
7	return invocation.proceed();
8  } finally {
9	DynamicDataSourceContextHolder.poll();
10 }
  1. 在执行数据库操作时候, 就会调用DataSource.getConnection, 此时的DataSource指的就是DynamicRoutingDataSource
  2. 然后执行模板方法
1  @Override
2  public DataSource determineDataSource() {
3	// 拿到之前切换的数据源标识
4	String dsKey =  DynamicDataSourceContextHolder.peek();
5	// 通过该标识获取对应的数据源
6	return getDataSource(dsKey);
7 }

你可能感兴趣的:(java,java)