时间戳

首页 > 解决方案 > 使用 FlywayMigrationStrategy 的多个 Flyway

spring-boot - 使用 FlywayMigrationStrategy 的多个 Flyway

问题描述

我有多个Flyway数据源，需要对FlywayMigrationStrategy它们实施。每个数据源都有自己的flyway_migration表等。

但是当我创建FlywayMigrationStrategy它时不会被调用。

这不起作用：

@Bean
public FlywayMigrationStrategy cleanMigrateStrategy() { ...

这有效：

  @PostConstruct
  public void cleanBeforeMigrate(
      @Qualifier("dpaFlyway") Flyway dpaflyway,
      @Qualifier("flyway") Flyway flyway) {

    dpaflyway.clean();
    dpaflyway.migrate();

    flyway.clean();
    flyway.migrate();
  }

有更好的选择吗？

标签： spring-bootflyway

解决方案

为什么要实现@PostContruct方法？如我所见，FlywayMigrationStrategy是一个功能接口，我猜您的代码应该如下所示：

@Bean
public FlywayMigrationStrategy cleanMigrateStrategy() {
    FlywayMigrationStrategy strategy = new FlywayMigrationStrategy() {
        @Override
        public void migrate(Flyway flyway) {
            flyway.clean();
            flyway.migrate();
        }
    };

    return strategy;
}

好的，我有一些研究，现在可以解释你的问题的答案。

首先，让我们检查一下FlywayAutoConfiguration

当您使用通过属性文件配置 flyway 迁移时，配置工作，创建和配置 Flyway 实例并迁移基础。所有这些都配置在FlywayConfiguration. 再来看看配置上的条件注解

@ConditionalOnMissingBean(Flyway.class)

这意味着如果类的 beanFlyway.class已经存在于 spring 上下文中，则不会创建配置。好的，接下来，配置只创建到bean

@Bean
public Flyway flyway(FlywayProperties properties, DataSourceProperties dataSourceProperties,
        ResourceLoader resourceLoader, ObjectProvider<DataSource> dataSource,
        @FlywayDataSource ObjectProvider<DataSource> flywayDataSource,
        ObjectProvider<FlywayConfigurationCustomizer> fluentConfigurationCustomizers,
        ObjectProvider<JavaMigration> javaMigrations, ObjectProvider<Callback> callbacks)
.....
@Bean
@ConditionalOnMissingBean
public FlywayMigrationInitializer flywayInitializer(Flyway flyway,
        ObjectProvider<FlywayMigrationStrategy> migrationStrategy) {
        return new FlywayMigrationInitializer(flyway, migrationStrategy.getIfAvailable());
}

配置如何配置flywaybean 并不重要，但重要的是它创建了FlywayMigrationInitializer启动迁移的 bean。如您所见，FlywayMigrationStrategy设置FlywayMigrationInitializer为构造函数参数。如果设置了 MigrationStrategy ，则由他使用，否则只需执行flyway.migrate()。（见源代码）

好的，现在我们知道它一般是如何工作的，让我们看看你的代码。

您在主配置中创建 flyway 实例：

@Bean(initMethod = "migrate")
@FlywayDataSource
public Flyway firstFlyway(DataSource dataSource) {
  return new Flyway(
      new FluentConfiguration()
          .locations("db/first-migration")
          .schemas("first")
          .outOfOrder(true)
          .dataSource(dataSource)
  );
}

@Bean(initMethod = "migrate")
@FlywayDataSource
public Flyway secondFlyway(@Qualifier("secondDataSource") DataSource dataSource) {
  return new Flyway(
      new FluentConfiguration()
          .dataSource(dataSource)
          .schemas("second")
          .outOfOrder(true)
          .locations("db/second-migration")
  );
}

由于您已经创建了 Flyway 实例，FlywayConfiguration因此不会创建（由于条件）并且Flyway也不会创建您的 bean 的 Initializer bean。因此迁移不会执行，您需要添加(initMethod = "migrate")到 bean 声明以开始迁移。另外，我认为这@FlywayDataSource不是必需的，什么也不做。

现在让我们转到您的测试配置。

创建时FlywayMigrationStrategy，它可以正常工作，但没有人使用该策略（在您的主配置文件中，您调用的Flyway.migrate只是 initMethod 和 MigrationInitializer 未创建）。结果，该策略没有被执行。

TestConfiguration在您的工作示例中，您在postconstruct 方法中添加了调用 clean 和 migrate 。它之所以有效，是因为 postconstruct 方法是在创建配置后执行的。但是，如果您调试代码，您将看到该migrate方法为每个 Flyway 实例执行两次：作为 beaninitMethod和来自 Testconfiguration postcontruct 方法。我不确定这是你想要的。

好的，为了修复它，我建议删除(initMethod = "migrate")and ，为您的主配置中的每个bean@FlywayDataSource创建bean 并在您的测试配置中实现。FlywayMigrationInitializerFlywayFlywayMigrationStrategy

@Bean
public FlywayMigrationInitializer flywayInitializer(@Qualifier(...) Flyway flyway,
        ObjectProvider<FlywayMigrationStrategy> migrationStrategy) {
        return new FlywayMigrationInitializer(flyway, migrationStrategy.getIfAvailable());
}

推荐阅读