如何在Flyway社区版中针对特定环境运行SQL脚本？

Flyway社区版原生不支持类似Jinja的模板语法，但可以通过以下几种方式实现针对特定环境执行CREATE USER/GRANT这类操作：

1. 环境特定SQL文件目录 #

Flyway允许为不同环境创建独立的SQL文件目录，通过配置指定对应环境的加载路径：

• 构建如下目录结构：

  src/main/resources/db/migration
  ├── common
  │   ├── V1__initial_schema.sql
  ├── stg
  │   ├── V1__grant_stg_permissions.sql
  ├── prod
  │   ├── V1__grant_prod_permissions.sql
  

• 在不同环境的配置文件中设置flyway.locations：
  • 测试环境（stg）：flyway.locations=classpath:db/migration/common,classpath:db/migration/stg
  • 生产环境（prod）：flyway.locations=classpath:db/migration/common,classpath:db/migration/prod
• 执行时Flyway只会加载对应环境目录下的SQL文件，实现环境专属的权限操作。

2. 占位符+数据库条件语法 #

利用Flyway的占位符替换功能，结合数据库自身的条件判断语句实现：

• 配置中定义环境占位符：flyway.placeholders.env=stg
• 在SQL文件中使用数据库支持的条件逻辑，以PostgreSQL为例：

  DO $$
  BEGIN
    IF current_setting('flyway.env') = 'stg' THEN
      GRANT test_role TO test_user;
    END IF;
  END $$;
  

• 不同数据库语法有差异：MySQL可使用IF()函数或存储过程，Oracle用CASE或PL/SQL块。

3. 自定义Java迁移 #

如果需要更灵活的逻辑，可编写自定义Java迁移类，在代码中判断环境并执行SQL：

• 创建继承BaseJavaMigration的类：

  import org.flywaydb.core.api.migration.BaseJavaMigration;
  import org.flywaydb.core.api.migration.Context;
  import java.sql.Statement;
  
  public class V1__GrantEnvironmentSpecificPermissions extends BaseJavaMigration {
      @Override
      public void migrate(Context context) throws Exception {
          String env = System.getProperty("env");
          if ("stg".equals(env)) {
              try (Statement stmt = context.getConnection().createStatement()) {
                  stmt.execute("GRANT test_role TO test_user;");
              }
          }
      }
  }
  

• 运行Flyway时通过-Denv=stg传递环境参数，类中读取参数后执行对应操作。

社区版无内置模板引擎，但通过以上三种方式可实现类似环境条件化执行的需求。

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