springboot动态多数据源jta事务管理

前言：

公司最近要做一个项目，要求从数据库里读取配置信息，动态切换数据源，这样数据源多的情况下就不用每次修改配置文件了。然后就参考了博客上很多文章，发现很多都是数据源可以动态切换，添加事务后就切换不了，或者是多数据源配置信息写在配置文件里，这些都不是我想要的，经过两天努力，借鉴了很多文章终于解决问题。现把过程记录下来以方便他人和自己。

项目运行从默认主数据库读取所有数据源配置信息，然后交给JTA事务管理器统一管理事务。

正文开始

为了方便观看和了解项目，可以先看下项目目录结构

首先 pom.xml 文件

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>
    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.3.1.RELEASE</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>
    <groupId>com.z</groupId>
    <artifactId>demo</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <name>demo</name>
    <description>Demo project for Spring Boot</description>

    <properties>
        <java.version>1.8</java.version>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <!-- mysql -->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>5.1.38</version>
        </dependency>
        <!-- druid -->
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>druid</artifactId>
            <version>1.1.6</version>
        </dependency>
        <!-- mybatis -->
        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter</artifactId>
            <version>1.3.0</version>
        </dependency>
        <!-- lombok -->
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>
        <!-- jta -->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-jta-atomikos</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
            <exclusions>
                <exclusion>
                    <groupId>org.junit.vintage</groupId>
                    <artifactId>junit-vintage-engine</artifactId>
                </exclusion>
            </exclusions>
        </dependency>
    </dependencies>

    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
            </plugin>
        </plugins>
    </build>

</project>

然后 application.yml (这里配置的是主数据源)

server:
  port: 8090
spring:
  datasource:
    type: com.alibaba.druid.pool.DruidDataSource
    druid:
      # 默认主数据源
      system:
        username: root
        password: 123
        url: jdbc:mysql://localhost:3306/demo?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=UTC
        driverClassName: com.mysql.jdbc.Driver

准备完毕，接下来就是代码

DataSourceContextHolder.java

package com.z.demo.config.utils;

/*
* 数据源上下文切换
* */
public class DataSourceContextHolder {
   

    private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();

    /*
    * 设置当前数据源
    * */
    public static void setDatasourceType(String dbType) {
   
        contextHolder.set(dbType);
    }

    /*
    * 取得当前数据源
    * */
    public static String getDatasourceType() {
   
        return contextHolder.get();
    }

    /*
    * 清除上下文数据源
    * */
    public static void clearDatasourceType() {
   
        contextHolder.remove();
    }
}

CustomSqlSessionTemplate.java

package com.z.demo.config.utils;