具有JPA依赖的Flyway Spring Boot Autowired Bean

Answer 1

由于我对JPA的依赖，我为此挣扎了很长时间。我要稍微修改一下我的问题的标题来反映这一点...

@AutoWiredbean是从ApplicationContext实例化的。我们可以创建一个不同的bean，即ApplicationContextAWare，并使用它“手动连接”我们的bean，以便在迁移中使用。

这里可以找到一个相当干净的方法。不幸的是，在使用JPA时，这会抛出一个未捕获的异常（特别是applicationcontext为null)。幸运的是，我们可以通过使用@dependson注释来解决这个问题，并在设置了ApplicationContext之后强制flyway运行。

首先，我们需要上面avehlies/spring-beans-flyway2中的springutility。

package com.mypackage;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

@Component
public class SpringUtility implements ApplicationContextAware {

    @Autowired
    private static ApplicationContext applicationContext;

    public void setApplicationContext(final ApplicationContext applicationContext) {
        this.applicationContext = applicationContext;
    }

    /*
        Get a class bean from the application context
     */
    public static <T> T getBean(final Class clazz) {
        return (T) applicationContext.getBean(clazz);
    }

    /*
        Return the application context if necessary for anything else
     */
    public static ApplicationContext getContext() {
        return applicationContext;
    }

}

然后，使用SpringUtility的@dependson配置FlywayInitializer。我在这里扩展了FlywayAutoConfiguration，希望保留自动配置功能。这对我来说似乎很管用，只是关闭gradle.build文件中的flyway不再管用了，所以我不得不添加@profile（“！integration”）来防止它在测试期间运行。除此之外，自动配置似乎对我有效，但必须承认，我只运行了一次迁移。如果我错了，希望有人能纠正我。

package com.mypackage;

import org.flywaydb.core.Flyway;
import org.springframework.boot.autoconfigure.flyway.FlywayMigrationInitializer;
import org.springframework.boot.autoconfigure.flyway.FlywayAutoConfiguration.FlywayConfiguration;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Profile;
import org.springframework.context.annotation.Primary;
import org.springframework.context.annotation.DependsOn;

import com.mypackage.SpringUtility;

@Configuration
@Profile("!integration")
class MyFlywayConfiguration extends FlywayConfiguration {
    @Primary
    @Bean(name = "flywayInitializer")
    @DependsOn("springUtility")
    public FlywayMigrationInitializer flywayInitializer(Flyway flyway){
        return super.flywayInitializer(flyway);
        //return new FlywayMigrationInitializer(flyway, null);
    }
}

为了完成这个示例，这里有一个迁移：

package db.migration;

import org.flywaydb.core.api.migration.spring.BaseSpringJdbcMigration;
import org.springframework.jdbc.core.JdbcTemplate;

import org.springframework.stereotype.Component;

import com.mypackage.repository.AccountRepository;
import com.mypackage.domain.Account;

import com.mypackage.SpringUtility;

import java.util.List;

public class V2__account_name_ucase_firstname extends BaseSpringJdbcMigration {

    private AccountRepository accountRepository = SpringUtility.getBean(AccountRepository.class);

    public void migrate(JdbcTemplate jdbcTemplate) throws Exception {

        List<Account> accounts = accountRepository.findAll();

        for (Account account : accounts) {

            String firstName = account.getFirstName();
            account.setFirstName(firstName.substring(0, 1).toUpperCase() + firstName.substring(1));
            account = accountRepository.save(account);

        }
    }
}

感谢github上的avehlies、stack overflow上的Andy Wilkinson和github上的OldIMP一路帮助我。

如果您使用的是最新版本的Flyway，请扩展BaseJavaMigration而不是BaseSpringJDBCMigration，因为后者不推荐使用。另外，看看下面用户Wim Deblauwe的两条评论。