首页 > 其他分享> > SpringBoot2项目中(JPA + Druid)使用多数据源

SpringBoot2项目中(JPA + Druid)使用多数据源

2021-01-02 06:32:06 作者：互联网

SpringBoot项目中JPA使用多数据源（举例用Database和Druid两种配置方式注：我仅写Druid的基础数据库配置）

注：代码部分因为影响阅读我将它们折叠起来惹，注意前面有小箭头的文本嗷

本文代码篇幅较长，我愿意写，你愿意听看嘛？

技术栈（仅说一些必要的，记着要对症下药，避免因为环境不对而不能使用）

mysql-connector-java 8.0.22 // 说真的有了druid之后我有段怀疑有没有必要要这个东西了... 等看明白了druid之后我回来再更新以下。
druid-spring-boot-starter 1.2.3
spring-boot 2.4
spring-boot-starter-data-jpa

P.S.

需要的jar包可以直接在MavenRepository里搜索下载

这里的配置是基于注册中心的

业务背景只是些杂谈，具体实现直接跳转到实现过程中

预先说明

本项目内容我是使用Kotlin编写的，如果你用的IDE是IJ的产品，那么可以直接复制到Java代码中，IDE会自动编译成Java代码，但是不能保证所有的代码都是正确的，所以需要自己手动修复一部分，放心，不会太多的。

_{喵的用了三天时间才完成这个模块...kotlin害龙不浅呐（小声bb）}

对于这部分代码我计划加上其它功能后封装一下，作为一个模版项目开源，不过短时间里并没有足够的时间去做它。

业务背景

在写了对方三个管理系统之后，展开了一次新的关于数据整合的业务，在这个业务中，我们需要拿到多个项目后台的数据集。在这里我想到了两种解决方案，分别介绍一下其优劣。

以下内容我将跑业务的服务器统称为`业务后台`，将整合数据使用的后台称为`数据后台`

通过不同业务中的后台中提供相对应的api来获取所需要的数据
- ★ 可以更快地实现（添加接口）而无需重新配置一个项目（懒人专用）
- ★ 对于数据后台来说能够更好的管理接口（通用的东西很多，可以很好地实现模块化）
- ☆ 权限的对接要单独写一个模块
- ☆ 如果图表有更变的话，需要修改所对应的业务，这样会让项目变得很乱
- ☆ 除了查询的网络请求延时之外，中间还会再加一段网络数据请求（几乎可以无视，除非——）
一个后台进行多个数据库的链接，自己拉取得所需要的数据
- ★ 修改时不容易影响到其它的业务（独立服务）
- ★ 减少中间的数据请求过程，让工厂与卖家少一层代理（你们都知道代理是要赚钱的吧？）
- ★
  数据整合统一在一个地方，易于处理，方便中间的数据测试（不需要再改大量的配置文件，不过我确定现在有办法解决这个问题，貌似阿里的学习套件里就包含了test和prod的运行环境部署，或者是部署为docker镜像，不过我还没尝试过，暂时）

既然是做了数据的整合，对于多数据库的访问就是必不可少的了，接下来的就是这篇文章的正题。

实现过程

SpringBoot配置数据库有两个阶段（2、3）：

配置文件中加入数据信息（注册中心的方式配置）
DataSourceConfig（入口注入一些基本信息，类似于对象生成）
DataBaseConfig（数据库配置目的在于指定数据库所服务的区域）

在这里，分为两个步骤实现，第一步实现通用方法，第二步是实现分库配置的方法

P.S.

这里面两个板块的方法都是可以直接使用的（直接将通用方法或者定制方法的代码全部复制进去使用），定制化的配置相当于通用方法的添加内容，我会表明哪些是添加的内容，具体方便自己写。

虽然我比较讨厌这么做，因为太过冗余...不过我也做过一个使用者，对于我们用户来说，我们更喜欢这样的拿来即用的东西。

通用方法

先上项目结构，快速认清局势（为了生成一个树状图，专门下了个brew，各种恶心的问题...）：

 origin  # 因为前面的一堆东西太长，干扰视线，所以也就没有加进去了，你们能明白就行
 ├── config
 │   ├── DataSourceConfig.kt  # 入口文件，这里用了Druid
 │   ├── ServiceAConfig.kt  # 业务A使用的数据库配置
 │   └── ServiceBConfig.kt  # 业务B使用的数据库配置
 ├── serviceA
 │   └── dao
 │       └── ServiceADao.kt  # 这是个Dao，不用我解释了吧？
 └── serviceB
     └── dao
         └── ServiceBDao.kt  # 我记得他们写JPA的喜欢命名为Repository？？

DataSourceConfig.kt

// DataSourceConfig.kt

@Configuration
class DataSourceConfig {
    @Primary  //  默认数据库要加Primary关键词修饰
    @Bean("serviceADataSource")  // Bean名称，还是起一下的好
    @Qualifier("serviceADataSource")  // 数据源的分类标记（就像公狗在树下撒尿）

    // yml or properties下的配置内容，将内容通过控制中心直接注入
    @ConfigurationProperties(prefix = "spring.datasource.serviceA")
    fun serviceADataSource(): DataSource {
        return DruidDataSource()
        // 这里，如果用原生的数据库的话，用下面注释掉的内容即可（我在下面的properties配置中仅配置了Druid的写法，原生的需要你自己去改写）
//      return DataSourceBuilder.create().build()
    }

    @Bean("serviceBDataSource")
    @Qualifier("serviceBDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.serviceB")
    fun serviceBDataSource(): DataSource {
        return DruidDataSource()
    }
}

ServiceAConfig.kt

// ServiceAConfig.kt

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(
        entityManagerFactoryRef = "serviceAEntityManagerFactory",
        transactionManagerRef = "serviceATransactionManager",
        basePackages = ["com.arunoido.origin.serviceA"]  // 这里是数据库指向的包名，我这里用的是我自己的包名。愿意的话你可以具体指向到自己的Dao层（[com.arunoido.origin.serviceA.dao]）
)
class ServiceAConfig {

    @Autowired
    @Qualifier("serviceADataSource")
    private lateinit var dataSource: DataSource

    @Primary
    @Bean(name = ["serviceAEntityManager"])
    fun entityManager(builder: EntityManagerFactoryBuilder): EntityManager? {
        return entityManagerFactory(builder).getObject()?.createEntityManager()
    }

    @Primary
    @Bean(name = ["serviceAEntityManagerFactory"])
    fun entityManagerFactory(builder: EntityManagerFactoryBuilder): LocalContainerEntityManagerFactoryBean {

        return builder
                .dataSource(dataSource)
                .packages("com.arunoido.origin.serviceA.model")  // 设置实体类所在位置
                .build()
    }

    @Primary
    @Bean(name = ["serviceATransactionManager"])
    fun transactionManager(builder: EntityManagerFactoryBuilder): PlatformTransactionManager? {
        return JpaTransactionManager(entityManagerFactory(builder).getObject()!!)
    }
}

ServiceBConfig.kt

// ServiceBConfig.kt

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(
        entityManagerFactoryRef = "serviceBEntityManagerFactory",
        transactionManagerRef = "serviceBTransactionManager",
        basePackages = ["com.arunoido.origin.serviceB"]  // 可以指向多个包名，你懂的
)
class ServiceBConfig {

    @Autowired
    @Qualifier("serviceBDataSource")
    private lateinit var dataSource: DataSource

    @Bean(name = ["serviceBEntityManager"])
    fun entityManager(builder: EntityManagerFactoryBuilder): EntityManager? {
        return entityManagerFactory(builder).getObject()?.createEntityManager()
    }

    @Bean(name = ["serviceBEntityManagerFactory"])
    fun entityManagerFactory(builder: EntityManagerFactoryBuilder): LocalContainerEntityManagerFactoryBean {

        return builder
                .dataSource(dataSource)
                .packages("com.arunoido.origin.serviceB.model")
                .build()
    }

    @Bean(name = ["serviceBTransactionManager"])
    fun transactionManager(builder: EntityManagerFactoryBuilder): PlatformTransactionManager? {
        return JpaTransactionManager(entityManagerFactory(builder).getObject()!!)
    }
}

配置文件：

└── resources
   ├── application-dev.properties
   └── application.properties

P.S.

这里我使用的配置文件是跑在开发环境的properties，如果你习惯写yml的话可以自己改过去，关键词相同，只是结构不同了（我其实挺喜欢yml的结构的，主要是想尝试下新东西，嗯）

然后使用dev的配置是在application.properties中的spring.profiles.active=dev

application-dev.properties

# ServiceA的数据库
spring.datasource.serviceA.url=jdbc:mysql://ipaddress:port/serviceA?useUnicode=true&characterEncoding=UTF-8&serverTimezone=Asia/Shanghai
spring.datasource.serviceA.username=username
spring.datasource.serviceA.password=pwd
spring.datasource.serviceA.driver-class-name=com.mysql.cj.jdbc.Driver
# ServiceB的数据库
spring.datasource.serviceB.url=jdbc:mysql://ipaddress:port/serviceB?useUnicode=true&characterEncoding=UTF-8&serverTimezone=Asia/Shanghai
spring.datasource.serviceB.username=username
spring.datasource.serviceB.password=pwd
spring.datasource.serviceB.driver-class-name=com.mysql.cj.jdbc.Driver
# 通用的JPA配置
spring.jpa.hibernate.ddl-auto=create-drop
spring.jpa.show-sql=false
spring.jpa.properties.hibernate.hbm2ddl.auto=update
spring.jpa.properties.hibernate.dialect=org.hibernate.dialect.MySQL5InnoDBDialect
# Druid的配置，如果不用Druid的话自己配置一下
# 初始化大小，最小，最大
spring.datasource.druid.initial-size=10
spring.datasource.druid.max-active=100
spring.datasource.druid.min-idle=10
#配置获取连接等待超时的时间
spring.datasource.druid.max-wait=60000
#打开PSCache，并且指定每个连接上PSCache的大小
spring.datasource.druid.pool-prepared-statements=true
spring.datasource.druid.max-pool-prepared-statement-per-connection-size=20
#配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒
spring.datasource.druid.time-between-eviction-runs-millis=60000
#配置一个连接在池中最小生存的时间，单位是毫秒
spring.datasource.druid.min-evictable-idle-time-millis=300000
spring.datasource.druid.test-while-idle=true
spring.datasource.druid.test-on-borrow=false
spring.datasource.druid.test-on-return=false
spring.datasource.druid.stat-view-servlet.enabled=true
spring.datasource.druid.stat-view-servlet.url-pattern=/druid/*
spring.datasource.druid.filter.stat.log-slow-sql=true
spring.datasource.druid.filter.stat.slow-sql-millis=1000
spring.datasource.druid.filter.stat.merge-sql=false
spring.datasource.druid.filter.wall.config.multi-statement-allow=true

通用方法覆写（定制化配置）

说明：

一般情况下，我们需要用两种jpa的策略的时候才会用到这里的内容，否则上面的默认配置完全足够使用。

我举一个最简单的可以用到这种方式配置的场景——业务A的数据库使用的是Mysql，业务B使用的数据库是Oracle，这个时候就需要把他们的Driver分别配置了。

老规矩，先上项目结构（M -> Modify）：

└── origin
   ├── config
   │   ├── DataSourceConfig.kt
 M │   ├── ServiceAConfig.kt
 M │   ├── ServiceBConfig.kt
 + │   └── VendorPropertiesConfig.kt
 + ├── global
 + │   └── JpaProperties.kt
   ├── serviceA
   │   └── dao
   │       └── ServiceADao.kt
   └── serviceB
       └── dao
           └── ServiceBDao.kt

+ VendorPropertiesConfig.kt

// VendorPropertiesConfig.kt

@Configuration
class VendorPropertiesConfig {
    /**
     *
     * @return {JpaProperties} jpaProperties
     * 这个类可以覆盖通用属性
     */
    @Bean
    @ConfigurationProperties(prefix = "spring.jpa.properties.serviceA")  // 地址可以随意点，只要不和框架的地址冲突就好
    fun getServiceAProperties(): JpaProperties {
        return JpaProperties()  // 这里用自己写的JpaProperties类，注意不要导错包
    }
    /**
     *
     * @return {JpaProperties} jpaProperties
     * ServiceB的属性
     */
    @Bean
    @ConfigurationProperties(prefix = "spring.jpa.properties.serviceB") 
    fun getServiceBProperties(): JpaProperties {
        return JpaProperties()  
    }
}

JpaProperties我需要说明一下，这里我只列举了几个我用到的配置项，所以只写了四个，你需要以此类推的去写自己用到的选项。

这里的格式我参考了Druid的写法。

+ JpaProperties.kt

// JpaProperties.kt

/**
 *
 * 说明一下，这就是个kotlin版的JavaBean，你只需要把它作为JavaBean写，然后加上两个内部的处理方法就好了，该写getter/setter的写getter/setter。
 * 写lombok的嘛...我不建议写lombok，本龙是亲身体验过lombok版本问题导致的项目无法运行，别问我为什么不改版本，因为EAP和Ultimate的lombok版本本来就不同步。
 */
data class JpaProperties(
        var ddl_auto: String?,
        var dialect: String?,
        var physical_naming_strategy: String?,
        var implicit_naming_strategy: String?,
) {
    constructor() : this(null, null, null, null)

    private fun setConfig(): HashMap<String, *> {
        val properties = HashMap<String, Any>()
        val prefix = "hibernate."

        if (!ddl_auto.isNullOrBlank())
            properties["${prefix}ddl-auto"] = ddl_auto!!

        if (!dialect.isNullOrBlank())
            properties["${prefix}dialect"] = dialect!!

        if (!physical_naming_strategy.isNullOrBlank())
            properties["${prefix}physical_naming_strategy"] = physical_naming_strategy!!

        if (!implicit_naming_strategy.isNullOrBlank())
            properties["${prefix}implicit_naming_strategy"] = implicit_naming_strategy!!

        return properties

    }

    fun getProperties(): HashMap<String, *> {
        return setConfig()
    }

}