在本篇文中，我们将研究如何使用Flyway在Spring Boot应用程序中管理SQL数据库模式

Flyway是一个数据库迁移工具，它提供迁移历史和回滚的功能，可以帮助开发人员自动化管理数据库结构和预置数据的变更，减少手动数据迁移，从而提高开发效率和数据库管理的可靠性。

创建Spring Boot

我们将使用的 Spring Boot 应用程序可以使用此Spring Initializr 链接生成。它包含所有必要的依赖项。

下载应用程序并解决依赖关系后，我们将创建一个名为spring-boot-flyway的新 Postgres 数据库，并配置应用程序以连接到它。

application.properties：

spring.datasource.url=jdbc:postgresql://localhost:5432/spring-boot-flyway
spring.datasource.username=demo
spring.datasource.password=demo

默认情况下，Flyway 会在类路径中的db/migration/目录中搜索包含用于管理数据库表和记录的 SQL 语句的迁移文件。

对于旧版本的库，我们可能需要在resources/db/migration/ 中创建一个名为.keep的空文本文件，以确保该目录在应用程序启动期间被编译并可用，以避免错误。

完成此操作后，我们现在可以启动应用程序并且它应该成功运行。

基本用法

Flyway 的工作方式是，我们在resources/db/migration目录中创建一个迁移文件，Spring Boot 会自动执行迁移脚本，因为我们已经在上面的描述中将 Flyway 依赖项添加到了类路径中。

V1__Users.sql：

CREATE TABLE IF NOT EXISTS users
(
    id    SERIAL,
    email VARCHAR(200) NOT NULL,
    name  VARCHAR(200) NOT NULL,
    PRIMARY KEY (id)
);

我们来仔细检查一下此段代码片段。文件名V1__Users.sql遵循一定的约定：

• “ V ”表示这是版本化迁移。
• V 后面的“ 1 ”是实际版本号。它也可以是“ V1_1 ”，这将转换为版本 1.1。
• 后面是分隔符“ __ ”（两个下划线）。这会将版本信息与迁移文件的名称（在本例中为Users ）分开。
• 最后一部分“ .sql ”是扩展名；因此，该文件包含一个简单的 SQL 语句。

此时，重新启动应用程序将在数据库中创建users表。此外，我们还可以看到flyway创建了另一个表--flyway_schema_history。

Flyway_schema_history由 Flyway 本身用来跟踪已应用的迁移。如果该表丢失，Flyway 将假设我们是第一次初始化数据库，并按照版本号的顺序运行所有迁移。

当Flyway_schema_history表存在时，Flyway 将仅应用之前未应用过的较新的迁移文件。这意味着，如果为了添加新表，我们只需创建具有更新版本号的更新的迁移文件并重新启动应用程序。

除了使用 SQL 之外，我们还可以使用Java编写迁移脚本。在Java迁移风格中，我们的迁移文件是Java类，必须扩展抽象BaseJavaMigration类并实现migrate方法。

IDE 通常不希望 Java 类位于resources目录中，因此我们将在src/main/java中创建一个名为db/migration的新包。非常重要的是要知道这个新包db/migration应该位于src/main/jav目录中。

让我们创建一个新的 Java 迁移来添加新表：

V2__Posts.java ：

public class V2__Posts extends BaseJavaMigration {
    @Override
    public void migrate(Context context) throws Exception {
        var sql = """
                CREATE TABLE posts (
                     id SERIAL,
                     author_id INT NOT NULL,
                     post TEXT NOT NULL,
                     created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
                     updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
                     PRIMARY KEY (id)
                );
                """;
        try(var statement = context.getConnection().createStatement()) {
            statement.execute(sql);
        }
    }
}

与 SQL 文件相比，使用 Java 迁移的优点是我们可以添加使用普通 SQL 无法实现的自定义逻辑、条件和验证。例如，我们可以检查另一个表是否存在或从环境中获取某些值等。

正如现在可能猜到的那样，可以在同一个代码库中混合 SQL 和 Java 风格的迁移，只要我们确保两种情况下的 Flyway 位置相同。

Flyway配置和定制

到目前为止，我们一直在使用默认的 Flyway 行为。我们可以进一步调整 Flyway 以满足我们的需求。例如：

我们可以更改迁移文件的默认位置、

配置数据库脚本（也称为表空间）、

将 SQL 迁移前缀从“V”更改为我们想要的任何内容等等。在下面的配置中，我们配置了迁移文件所在的路径并禁用清理数据库（即删除所有表）以防止在生产环境中意外使用。

application.properties：

spring.flyway.locations=classpath:migrations
spring.flyway.clean-disabled=true

该键下还有其他可配置属性spring.flyway，我们可以使用它们来微调库的行为。另外，我们可以查阅Flyway 文档页面以供参考。

Flyway回调事件

Flyway为我们提供了配置回调的能力，这些回调可以在迁移过程的不同阶段调用。回调机制是在迁移生命周期的不同阶段执行某些操作的便捷方法。

假设我们有一些默认数据想要在应用程序启动时运行。我们可以简单地创建一个支持该AFTER_MIGRATE事件的回调。

FlywayDatabaseSeeder.java：

public class FlywayDatabaseSeeder implements Callback {

    @Override
    public boolean supports(Event event, Context context) {
        return event.name().equals(Event.AFTER_MIGRATE.name());
    }
    
    @Override
    public void handle(Event event, Context context) {
        try(var statement = context.getConnection().createStatement()) {

            var ADMIN_EMAIL = "superadmin@example.com";
            
            var checkQuery = "SELECT id FROM users WHERE email = %s"
                    .formatted(ADMIN_EMAIL);
            
            statement.execute(checkQuery);
            ResultSet resultSet = statement.getResultSet();
            resultSet.last();

            //如果已经执行了seeder，则返回
            if(resultSet.getRow() >= 0) return;

            var sql = """
                    INSERT INTO users (email, name) VALUES
                    ('%s', 'Super Admin')
                    """.formatted(ADMIN_EMAIL);
            
            statement.execute(sql);
            
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }
    }

    @Override
    public boolean canHandleInTransaction(Event event, Context context) {
        return true;
    }

    @Override
    public String getCallbackName() {
        return FlywayDatabaseSeeder.class.getName();
    }
}

在上面的清单中，在supports方法中，我们声明只应针对AFTER_MIGRATE事件执行此回调，并且在handle方法中，我们概述了插入默认超级管理员用户（如果尚不存在）的逻辑。

在这之前，我们需要在 SpringBoot 中向 Flyway 注册回调类。我们通过创建一个FlywayMigrationStrategybean 来实现这一点。

SpringBootFlywayApplication.java ：

@Bean
public FlywayMigrationStrategy flywayMigrationStrategy() {
	return (flywayOld) -> {

		/*
		 更新现有自动配置的Flyway bean
     以包含我们的回调类
		*/
		Flyway flyway = Flyway.configure()
				.configuration(flywayOld.getConfiguration())
				.callbacks(new FlywayDatabaseSeeder())
				.load();

		flyway.migrate();

	};
}

org.flywaydb.core.api.callback.Event枚举中还有其他事件 ，我们可以配置Callback类来支持。例如，可以有一个回调来支持该AFTER_MIGRATE_ERROR事件并发送 Slack 通知来提醒工程师。

技巧和窍门

1、在本地环境中进行开发时，可以从Flyway_schema_history表中删除迁移条目。

下次启动应用程序时，删除其历史记录的迁移将再次执行。这样，可以更正错误或更新架构，同时仍在本地计算机上进行开发，而无需删除整个数据库。

2、在 SpringBoot 中，可以控制 Flyway 在应用程序启动时执行迁移脚本。

例如，假设我们不希望在本地环境中自动执行迁移。我们可以执行以下操作：

SpringBootFlywayApplication.java：

Bean
public FlywayMigrationStrategy flywayMigrationStrategy(@Value("${spring.profiles.active}") String activeProfile) {
	return (flywayOld) -> {

		/*
		 更新现有自动配置的Flyway bean
     以包含我们的回调类
		*/
		Flyway flyway = Flyway.configure()
				.configuration(flywayOld.getConfiguration())
				.callbacks(new FlywayDatabaseSeeder())
				.load();

		if(!"local".equalsIgnoreCase(activeProfile)) {
			flyway.migrate();
		}

	};
}

结论

使用数据库迁移工具的优点之一是它使数据库架构成为应用程序代码库的一部分。由于应用程序中有一个中心参考点，因此可以更轻松地跟踪数据库随时间的变化。

Flyway 和 Liquibase 区别

阅读参考：

Flyway官网：https://flywaydb.org/
Flyway文档：https://documentation.red-gate.com/fd/