SpringBoot的多数据源以及事务解决方案(上篇)

发布时间 2024-01-04 22:49:12作者: 我想去海边

SpringBoot的多数据源以及事务解决方案

  1. 背景

    一个主库和N个应用库的数据源,并且会同时操作主库和应用库的数据,需要解决以下两个问题:

    1. 如何动态管理多个数据源以及切换?
    2. 如何保证多数据源场景下的数据一致性(事务)?
  2. 数据切换原理

    通过扩展Spring提供的抽象类AbstractRoutingDataSource,可以实现切换数据源。其类结构如下图所示

  • targetDataSources&defaultTargetDataSource

项目上需要使用的所有数据源和默认数据源。

  • resolvedDataSources&resolvedDefaultDataSource

当Spring容器创建AbstractRoutingDataSource对象时,通过调用afterPropertiesSet复制上述目标数据源。由此可见,一旦数据源实例对象创建完毕,业务无法再添加新的数据源。

  • determineCurrentLookupKey

    此方法为抽象方法,通过扩展这个方法来实现数据源的切换。目标数据源的结构为:Map<Object, DataSource>其key为lookup key

lookup key通常是绑定在线程上下文中,根据这个key去resolvedDataSources中取出DataSource。

根据目标数据源的管理方式不同,可以使用基于配置文件和数据库表两种方式。基于配置文件管理方案无法后续添加新的数据源,而基于数据库表方案管理,则更加灵活

  1. 配置文件解决方案

    根据上面分析,我们可以按照下面的步骤去实现:

    • 定义DynamicDataSource 继承AbstractRoutingDataSource ,重写 determineCurrentLookupKey 方法。

    • 配置多个数据源注入setTargetDataSourcesdefaultTargetDataSource ,通过 afterPropertiesSet 方法将数据源写入resolvedDataSourcesresolvedDefaultDataSource

    • 调用AbstractRoutingDataSourcegetConnection()方法时,determineTargetDataSource()方法返回DataSource执行底层的getConnection()

    3.1 设置数据源

    我们的数据源配置可能是yml文件或者xml文件形式配置

    yml 配置

    xml

    定义数据源配置读取接口 IDataSource

    在代码中配置Bean

  2. 数据库表解决方案

    要想动态的实现数据源的加载上面的方案还是不行,所以我们应将数据源的配置文件放到数据库表中。

    4.1 根据需求设置数据源表

    略。。。

    4.2 定义数据源 管理接口

​ 实现接口 sava 关键代码

​ 在项目启动时候,加载数据表中的所有数据源

​ 在DataSourceConfigure配置文件中需要在加入一个bena

​ 关键代码

在项目运行过程中,可以使用定时任务对数据源进行保活,为了提升性能再添加一层缓存

设置监听器 DataSourceInitListener 加载数据源

AbstractRoutingDataSource 只支持单库事务,切换数据源是在开启事务之前执行。 Spring使用 DataSourceTransactionManager进行事务管理。开启事务,会将数据源缓存到DataSourceTransactionObject对象中,后续的commit和 rollback事务操作实际上是使用的同一个数据源。

如何解决切库事务问题?借助Spring的声明式事务处理,我们可以在多次切库操作时强制开启新的事务:

下篇讲解