add
heheng
2025-06-10 942bdeee0b6fcc92b35e788c851d39c5182a8e40
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.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");
        }
    }
}