springboot flyway加载多数据源的方式

今天碰到一个需求是对多个数据源使用flyway进行数据库迁移，正常配置完flyway后发现只会迁移第一个数据库(主数据库)。

百度了半天也没找到个靠谱的... 而且翻来覆去就那么两篇文章 大多都是重复的且用的低版本flyway，于是去看官网的文档，发现其实很简单就能实现。

首先加入flyway的依赖然后去配置文件中配置一下（flyway版本为7.10.0），样子大概如下：

然后需要使用ConfigurationProperties的方式去获取一下配置文件(application.yml)中datasource的内容：

import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Configuration;
import org.springframework.stereotype.Component;

import java.util.Map;

@Component
@Configuration
@ConfigurationProperties(prefix = "spring")
public class DB {
    private Map<String, Map<String,String>> datasource;

    public Map<String, Map<String, String>> getDatasource() {
        return datasource;
    }

    public void setDatasource(Map<String, Map<String, String>> datasource) {
        this.datasource = datasource;
    }

    public DB() {
    }

    public DB(Map<String, Map<String, String>> datasource) {
        this.datasource = datasource;
    }
}

之后再创建一个用于遍历数据源的类，每遍历一个数据源就创建一个flyway的实体类然后migrate()

import com.example.demo.bean.DB;
import org.flywaydb.core.Flyway;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;

import java.util.Map;

@Component
public class FlywayLoad {
    @Resource
    private DB db_temp;

    public static FlywayLoad flywayLoad;
    public static DB db;
    @PostConstruct
    public void init() {
        db = db_temp;
        flywayLoad = this;
    }

    public static void dbMigrate(){
        Map<String, Map<String, String>> datasource = db.getDatasource();
        datasource.forEach((Key,value)->{
            Flyway flyway = Flyway.configure().dataSource(value.get("url"),value.get("username"),value.get("password")).load();
            flyway.migrate();
        });
    }
}

最后在启动类里面在run之后去调用一下这个FlywayLoad中的 dbMigrate方法就可以了：

import com.example.demo.config.FlywayLoad;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;

@SpringBootApplication
public class DemoApplication {
    public static void main(String[] args) {
        SpringApplication.run(DemoApplication.class, args);
        FlywayLoad.dbMigrate();
    }
}

项目结构如下：

总结一下就是一共分为两步：

        1. 循环遍历到所有的DataSource

        2. 每一个DataSource创建一个Flyway 去 migrate

Flyway flyway = Flyway.configure().dataSource(value.get("url"),value.get("username"),value.get("password")).load();
flyway.migrate();