Flyway针对多数据源配置实战方案

介绍

这篇文章主要介绍Flyway、Druid和多数据源结合起来如何配置和使用的方案。网上一些文章似乎都没有达到一开始我自己琢磨想要的程度,还是自己琢磨和调试出来了一套方案,欢迎大家斧正。

引入pom文件

首先引入对应的pom文件

<!-- flyway -->
<dependency>
    <groupId>org.flywaydb</groupId>
    <artifactId>flyway-core</artifactId>
    <version>6.4.4</version>
</dependency>
<!--Druid数据库连接池-->
<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>druid-spring-boot-starter</artifactId>
    <version>1.2.4</version>
</dependency>
<!--多数据源-->
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
    <version>3.4.1</version>
</dependency>

application.yml文件配置

spring:
  autoconfigure:
    # 排除 Druid 自动配置
    exclude: com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceAutoConfigure
  datasource:
    # 指定使用 Druid 数据源
    type: com.alibaba.druid.pool.DruidDataSource
    dynamic:
      datasource:
        goods:
          driverClassName: com.mysql.cj.jdbc.Driver
          url: 
          username: 
          password: 
        order:
          driverClassName: com.mysql.cj.jdbc.Driver
          url: 
          username: 
          password: 
        seckill:
          driverClassName: com.mysql.cj.jdbc.Driver
          url: 
          username: 
          password: 
      primary: order # 设置默认的数据源或者数据源组,默认值即为master
      strict: true # 严格匹配数据源,默认false. true未匹配到指定数据源时抛异常,false使用默认数据源
    druid:
      # 初始连接数
      initialSize: 5
      # 最小连接池数量
      minIdle: 10
      # 最大连接池数量
      maxActive: 20
      # 配置获取连接等待超时的时间
      maxWait: 60000
      # 配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒
      timeBetweenEvictionRunsMillis: 60000
      # 配置一个连接在池中最小生存的时间,单位是毫秒
      minEvictableIdleTimeMillis: 300000
      # 配置一个连接在池中最大生存的时间,单位是毫秒
      maxEvictableIdleTimeMillis: 900000
      # 配置检测连接是否有效
      validationQuery: SELECT 1 FROM DUAL
  flyway:
    enabled: true
    # 版本更新历史记录表
    table: t_db_version
    # 非空数据库初始化Flyway时需要打开此开关进行Baseline操作
    baseline-on-migrate: true
    # 是否可以无序执行。开发环境最好开启 outOfOrder, 生产环境关闭 outOfOrder
    out-of-order: true
    # 迁移前校验 SQL 文件是否存在问题
    validate-on-migrate: true
    locations: classpath:db/order # 这里写主库路径,然后在配置类中转换

这里的主库为order,从库为goods和seckill。

这里解释下,为什么网上的很多方案都不行的原因就是,如果这里的localtions配置为db或者db/migration这样存在所有flyway版本sql的路径,则最终结果会导致项目跑起来,flysqy会在主库中出现所有其他从库的表和自己主库本该有的表。

所以这里要结合flyway配置类来,配置类代码下面介绍。

项目文件夹结构

展示下项目文件夹结构。
项目文件夹结构图.jpg

Flyway配置类

关于配置类的代码如下。结合上面的图,其实本质就是将application文件的flyway的locations设置为db/order这个主库的路径,让主库只去执行order下的sql文件,然后在配置类中拆解locations,重新组成为db这个路径,然后再组合对应的库的名字。这个方案也是自己调试和探索出来的。

import com.baomidou.dynamic.datasource.DynamicRoutingDataSource;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.flywaydb.core.Flyway;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.transaction.annotation.EnableTransactionManagement;

import javax.annotation.PostConstruct;
import javax.sql.DataSource;
import java.util.Map;

@Slf4j
@Configuration
@RequiredArgsConstructor
@EnableTransactionManagement
public class FlywayConfig {

    private final DataSource dataSource;

    @Value("${spring.flyway.locations}")
    private String SQL_LOCATION;

    @Value("${spring.flyway.table}")
    private String VERSION_TABLE;

    @Value("${spring.flyway.baseline-on-migrate}")
    private boolean BASELINE_ON_MIGRATE;

    @Value("${spring.flyway.out-of-order}")
    private boolean OUT_OF_ORDER;

    @Value("${spring.flyway.validate-on-migrate}")
    private boolean VALIDATE_ON_MIGRATE;

    @Bean
    @PostConstruct
    public void migrateOrder() {
        log.info("调用数据库生成工具");
        SQL_LOCATION = SQL_LOCATION.split("/")[0]; // 将路径转换
        DynamicRoutingDataSource ds = (DynamicRoutingDataSource) dataSource;
        Map<String, DataSource> dataSources = ds.getDataSources();
        dataSources.forEach((k, v) -> {
            log.info("正在执行多数据源生成数据库文件 " + k);
            Flyway flyway = Flyway.configure()
                    .dataSource(v)
                    .locations(SQL_LOCATION + "/" + k)
                    .baselineOnMigrate(BASELINE_ON_MIGRATE)
                    .table(VERSION_TABLE)
                    .outOfOrder(OUT_OF_ORDER)
                    .validateOnMigrate(VALIDATE_ON_MIGRATE)
                    .load();
            flyway.migrate();
        });
    }

}

完成。