Initial commit

wangrong

2026-08-30 4c949c0dc9bcae785c281c9d8125295f09810b15

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
package io.dataease.flyway;
 
import org.flywaydb.core.Flyway;
import org.flywaydb.core.api.configuration.ClassicConfiguration;
import org.flywaydb.core.api.configuration.Configuration;
 
import javax.sql.DataSource;
import java.nio.charset.Charset;
 
public class TenantFlywayUtil {
 
    private static final String FLYWAY_MANAGE_TABLE_NAME = "de_manage_version";
 
    private static final String FLYWAY_TABLENAME_FORMAT = "de_tenant_%s_version";
    private static final String FLYWAY_LOCATION = "classpath:db/migration";
    private static final String FLYWAY_MANAGE_LOCATION = "classpath:db/distributed/manage";
 
 
    private static Configuration buildConfiguration(DataSource dataSource, boolean isManager, String appName) {
        ClassicConfiguration configuration = new ClassicConfiguration();
        configuration.setValidateOnMigrate(true);
        configuration.setBaselineOnMigrate(true);
        configuration.setTable(isManager ? FLYWAY_MANAGE_TABLE_NAME : getVersionTableName(appName));
        configuration.setEncoding(Charset.forName("utf-8"));
        configuration.setOutOfOrder(true);
        configuration.setBaselineVersionAsString("1");
        configuration.setLocationsAsStrings(isManager ? FLYWAY_MANAGE_LOCATION : FLYWAY_LOCATION);
        configuration.setDataSource(dataSource);
        return configuration;
    }
 
    public static void executeFlyway(DataSource dataSource, boolean isManage, String appName) throws Exception {
        Configuration configuration = buildConfiguration(dataSource, isManage, appName);
        Flyway flyway = new Flyway(configuration);
        flyway.migrate();
    }
 
    private static String getVersionTableName(String appName) {
        return String.format(FLYWAY_TABLENAME_FORMAT, appName);
    }
 
}