jpa和JdbcTemplate类的后期动态多数据源实现java源码

weixin_42183500

已于 2022-06-11 15:26:31 修改

阅读量1.5k

点赞数

文章标签： java 开发语言数据库 database sql

于 2022-06-11 13:11:22 首次发布

本文链接：https://blog.csdn.net/weixin_42183500/article/details/125231734

版权

本文介绍了一种通过自定义`DyDataSource`类，继承`DataSource`接口，实现在运行时动态切换多数据源的方法。在Java应用中，通过Aop切面编程和ThreadLocal来实现按需选择数据源，适用于数据库隔离级别的SAAS系统。示例代码展示了如何在Controller和Service层使用不同数据源，包括通过注解和回调函数指定数据源的方式。

摘要由CSDN通过智能技术生成

jpa和jdbcTemplate类都使用的是DataSource作为数据源，它只是一个接口类，主要就是对调用方提供了：

public Connection getConnection();

这样一个核心函数，返回一个标准的java.sql的连接对象Connection，以供给jpa或jdbcTemplate类使用来操作数据库表。

所以实现原理很简单，就是自写个dynamicDataSource的类，继承DataSource接口，重写getConnection()主函数即可。在多数据源的情况下，做一个字典Map,以用户选择的数据源标签String作为key，以真实数据源对象作为value,按需返回不同的真实数据源的Connection对象。

本文标题中的后期是指数据源的加入不是在配置文件生成bean的初如化时期，而是进程执行的任何时期。

本文标题中的动态是指数据源可以随时加入，也可以随时卸载。

本文标题中的多数据源是指多个和多种的意思，目前主要举例了:

DruidDataSource，HikariDataSource，PooledDataSource这三种连接池数据源，另外新增了一种jdbc短连接数据源（就是把一个jdbc短连接暂时作为数据源只用一次就关闭释放）。

本源码的作用在于：从一个数据库的表中取得那些读取其它数据库的信息之后，并发对N多个不同主机的数据库进行访问。对于数据库隔离级别的SAAS系统很适用。

本源码用到的关键技术是：

1.Aop切面编程，用它来实现以注解方式按需选择数据源和还原线程状态。

2.ThreadLocal（线程本地存储区）对定义的同一个变量，不同线程各自有一份独立的不同的内存变量的空间和值。

下面列出几处关键调用代码，完整的源码见后。

Controller控制器中的 “调用方法” 举例：

@MyOtherDataSource(value = "my_other_ds_white")//直接指明数据源标签
@GetMapping("/getbooks")//取多本书所有
public List<Book> getBooks() {
    List<Book> books=bookService.getBooks();
    return books;
}//不加注解则使用默认数据源

@MyOtherDataSource(is_use_method_last_para = true)//指明下面方法的最后一个参数是数据源TAG
@GetMapping("/getbook")//只取一本书,按书名
public String getBook(String name,String ds_tag) {
    Book book=bookService.getBook(name);//更建议把上面的注解用在服务层的方法上
    return book.toString();
}

第三种是在@MyOtherDataSource中指定一个回调函数来获取数据源TAG。此处暂略见开源。

下面是第四种：

@GetMapping("/getbooksd")
public List<Book> getBooksd() {
    List<Book> books=null;
    try {
        ThreadDataSourceTag.setThreadDSTag("my_other_ds_white");
        //不用注解，直接修改数据源标签来便捷地选择数据源
        books = bookService.getBooks();//执行JPA数据访问
    }catch (Exception e){e.printStackTrace();}
    finally {
        ThreadDataSourceTag.freeThreadDSTag();
        //但是注意这一条不论上面是否异常都要运行。否则会对后续数据访问形成影响。
    }

    return books;
}

最后一种是短连接数据源调用方式：

@GetMapping("/getbookbyid2")//只取一本书,按id

public String getBookbyid2(Long id) {

//调用服务层的方法，重点是传入后三个参数
book=bookService.getBook2( id, //查书的id

"jdbc:mysql://209.91.145.87:33062/white?useUnicode=true",

"super", //数据库的帐号

"123456" );
return book.toString(); }

//下面这几行代码是服务层service里的方法,在服务层方法上加的注解

@MyShortConnect(value = "com.mysql.cj.jdbc.Driver")
public Book getBook2(Long id,String jdbc_url,String uid,String pwd)
{  //上面后三个参数是AOP切面使用来建立短连接的，与本函数逻辑无关。
   return bookRepository.findBookById(id);//这个是JPA提供的函数了
}

上面是“讲一讲”，下面是"拿来用"

下面是目录结构图：

（本文列出了项目必须的所有源码文件，粘下来即可组成完整项目）

下面是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>
    <groupId>com.jpa</groupId>
    <artifactId>sb-jpa-plus</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <name>sb-jpa-demo</name>
    <description>sb-jpa-plus</description>

    <properties>
        <java.version>1.8</java.version>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
        <project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
        <spring-boot.version>2.3.7.RELEASE</spring-boot.version>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-jpa</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-thymeleaf</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <dependency>
            <groupId>com.zaxxer</groupId>
            <artifactId>HikariCP</artifactId>
            <version>4.0.3</version>
        </dependency>
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>druid</artifactId>
            <version>1.1.17</version>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-aop</artifactId>
        </dependency>
        <!--快重启-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-devtools</artifactId>
            <optional>true</optional> <!-- 可选 -->
        </dependency>


        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <scope>runtime</scope>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </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>
        <dependency>
            <groupId>org.mybatis</groupId>
            <artifactId>mybatis</artifactId>
            <version>3.5.7</version>
        </dependency>
    </dependencies>

    <dependencyManagement>
        <dependencies>