package com.gkhy.exam.admin.config;

import com.gkhy.exam.common.config.BaseFlywayConfig;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.DependsOn;

import javax.annotation.PostConstruct;

@Configuration
@DependsOn({"druidConfig"})
public class FlywayConfig extends BaseFlywayConfig {
    @Autowired
    private ApplicationContext applicationContext;

    @PostConstruct
    public void migrate() {
        initDataSource("masterDataSource");
        if(applicationContext.containsBean("slaveDataSource")){
            initDataSource("slaveDataSource");
        }
    }
}