Spring Boot多数据源

Spring Boot多数据源

前面一篇文章Spring Boot多数据源结合自己实际工作经验给大家分享一种简单灵活的SpringBoot多数据源动态切换方案，特点如下：

• 双重数据源初始化方式：支持配置文件和数据库配置表两种。
• 多种数据源切换方式：支持按AOP注解、用户身份、数据源名称三种方式切换。
• 统一事务管控机制：统一封装了事务访问service，事务统一通过@Transactional(rollbackFor = Exception.class)控制（单个数据源访问情况下使用）。
• 灵活实用：相较于常见实现方案，本方案更加灵活和实用 。

该方案存在如下两点不足或需要完善的地方：

切换数据库连接池不灵活： 初始化的默认多数据源数据源对象是HikariDataSource，如果想用其他数据源需要新加实现。

不支持分布式事务： 只能同时在一个数据源下面执行事务操作，不支持跨数据库事务功能。

下面介绍调整后的数据库连接初始化功能和分布式事务（JTA）功能。

一、数据库连接初始化功能

数据源对象参考网上常用方式继承AbstractRoutingDataSource实现，并通过初始化配置类DynamicDataSourceAndTemplateConfig初始化默认数据源和其他数据源到AbstractRoutingDataSource实现类。

1、数据源对象实现示例如下：

public class DynamicDataSourceRouter extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        return DynamicDataSourceKeyHolder.getKey();
    }
}

2、数据源初始化配置类示例如下：

二、添加分布式事务功能

要在多数据源（支持分布式事务的数据库数据源）里实现多个数据库操作事务支持，即分布式事务，主要通过框架里面的JTA实现，统一使用注解@Transactional(rollbackFor = Exception.class)控制事务。主要通过调整配置文件、增加初始化配置类、封装事务查询接口实现。

1、配置文件添加jta，示例如下：

maven配置：

application.yml添加jta开关和参数：

2、增加初始化配置类，示例如下：

3、封装事务查询接口，示例如下：

三、测试

主要在Controller中添加事务测试方法，示例如下：

添加好测试方法后，访问如下接口测试事务：

http://localhost:8080/test/base/testTransaction

测试结果：

1. 如果开启jta，并且模拟异常下面一行代码注释，可以在derby表里成功添加数据，访问一次增加一条。
2. 如果未开启jta，模拟异常下面一行代码注释，不可以在derby表里成功添加数据。
3. 如果开启jta，并且模拟异常下面一行代码未注释，derby表里添加数据失败，返回三条数据。

四、总结

本文分享了一种灵活的同时支持两种多数据源初始化方式、三种多数据源切换方式、多数据源分布式事务控制的SpringBoot多数据源功能实现方案，能满足大部分使用场景。希望对你有所帮助，有疑问欢迎留言交流。