如何通过Flyway仅在Local Profile下执行数据库初始化

方法1：利用Flyway的Profiles功能（Spring Boot集成版） #

Spring Boot集成的Flyway支持通过环境配置指定激活的Profile，结合SQL文件命名规则实现条件执行：

• 将初始化SQL文件命名为V1.1__init_data_local.sql（后缀_local对应目标环境Profile）
• 在application-local.yml中添加配置：

  flyway:
    profiles: local
  

• 在application-production.yml中不配置flyway.profiles或设为空，此时Flyway只会执行不带Profile后缀的迁移文件，自动跳过_local结尾的初始化脚本。

方法2：通过Spring Profile控制Flyway脚本目录 #

利用Spring多Profile配置，仅在Local环境下让Flyway加载额外的初始化脚本目录：

• 创建分环境配置文件：
  • application-local.yml：

    flyway:
      locations: classpath:db/migration, classpath:db/migration/local
    

  • application-production.yml：

    flyway:
      locations: classpath:db/migration
    

• 将仅Local环境执行的INSERT脚本放在src/main/resources/db/migration/local目录下（比如命名为V1.1__init_data.sql），常规迁移脚本保留在db/migration目录。这样只有Local环境会加载并执行local子目录下的脚本。

方法3：自定义Flyway回调实现细粒度控制 #

如果需要更灵活的逻辑（比如判断特定条件再执行），可以自定义Flyway回调结合Spring Profile判断：

• 创建Spring管理的回调类：

  import org.flywaydb.core.api.callback.BaseCallback;
  import org.flywaydb.core.api.callback.Context;
  import org.flywaydb.core.api.callback.Event;
  import org.springframework.core.env.Environment;
  import org.springframework.jdbc.core.JdbcTemplate;
  import org.springframework.stereotype.Component;
  import org.springframework.core.io.ClassPathResource;
  import org.apache.commons.io.FileUtils;
  import java.nio.charset.StandardCharsets;
  
  @Component
  public class LocalDataInitCallback extends BaseCallback {
      private final Environment env;
      private final JdbcTemplate jdbcTemplate;
  
      public LocalDataInitCallback(Environment env, JdbcTemplate jdbcTemplate) {
          this.env = env;
          this.jdbcTemplate = jdbcTemplate;
      }
  
      @Override
      public boolean supports(Event event, Context context) {
          // 仅在Flyway迁移完成后触发，且当前是local环境
          return event == Event.AFTER_MIGRATE && env.acceptsProfiles("local");
      }
  
      @Override
      public void handle(Event event, Context context) {
          // 方式1：直接执行SQL语句
          jdbcTemplate.execute("INSERT INTO your_table (col1, col2) VALUES ('val1', 'val2')");
  
          // 方式2：加载外部SQL文件执行（需引入commons-io依赖）
          // try {
          //     ClassPathResource resource = new ClassPathResource("initialization_script.sql");
          //     String sql = FileUtils.readFileToString(resource.getFile(), StandardCharsets.UTF_8);
          //     jdbcTemplate.execute(sql);
          // } catch (Exception e) {
          //     throw new RuntimeException("执行初始化脚本失败", e);
          // }
      }
  }
  

• 确保类被Spring扫描到，Flyway会在迁移完成后自动触发回调，仅Local环境执行初始化逻辑。

关键注意事项 #

• 所有初始化脚本必须保证幂等性（比如用INSERT ... ON DUPLICATE KEY UPDATE或先判断记录存在性），避免重复执行导致数据冲突。
• 严格遵守Flyway的版本号规则，确保脚本执行顺序符合业务预期。

内容的提问来源于stack exchange，提问作者Quilir