kongzy
2023-11-24 ebe94e19812a1b24257d60831ec932756855e94b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
package com.gkhy.assess.admin.config;
 
import com.gkhy.assess.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");
        }
    }
}