mirror of https://github.com/jeecgboot/jeecg-boot
EightMonth
8 months ago
13 changed files with 216 additions and 76 deletions
@ -0,0 +1,136 @@
|
||||
package org.jeecg.config.flyway; |
||||
|
||||
import com.baomidou.dynamic.datasource.DynamicRoutingDataSource; |
||||
import lombok.extern.slf4j.Slf4j; |
||||
import org.flywaydb.core.Flyway; |
||||
import org.flywaydb.core.api.FlywayException; |
||||
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
||||
import org.springframework.beans.factory.annotation.Value; |
||||
import org.springframework.context.annotation.Bean; |
||||
import org.springframework.context.annotation.Configuration; |
||||
import org.springframework.core.env.Environment; |
||||
|
||||
import javax.sql.DataSource; |
||||
import java.util.Map; |
||||
|
||||
/** |
||||
* @Description: 初始化flyway配置 修改之后支持多数据源，当出现异常时打印日志，不影响项目启动 |
||||
* |
||||
* @author: wangshuai |
||||
* @date: 2024/3/12 10:03 |
||||
*/ |
||||
@Slf4j |
||||
@Configuration |
||||
public class FlywayConfig { |
||||
|
||||
@Autowired |
||||
private DataSource dataSource; |
||||
|
||||
@Autowired |
||||
private Environment environment; |
||||
|
||||
/** |
||||
* 是否开启flyway |
||||
*/ |
||||
@Value("${spring.flyway.enabled:false}") |
||||
private Boolean enabled; |
||||
|
||||
/** |
||||
* 编码格式，默认UTF-8 |
||||
*/ |
||||
@Value("${spring.flyway.encoding:UTF-8}") |
||||
private String encoding; |
||||
|
||||
/** |
||||
* 迁移sql脚本文件存放路径，官方默认db/migration |
||||
*/ |
||||
@Value("${spring.flyway.locations:}") |
||||
private String locations; |
||||
|
||||
/** |
||||
* 迁移sql脚本文件名称的前缀，默认V |
||||
*/ |
||||
@Value("${spring.flyway.sql-migration-prefix:V}") |
||||
private String sqlMigrationPrefix; |
||||
|
||||
/** |
||||
* 迁移sql脚本文件名称的分隔符，默认2个下划线__ |
||||
*/ |
||||
@Value("${spring.flyway.sql-migration-separator:__}") |
||||
private String sqlMigrationSeparator; |
||||
|
||||
/** |
||||
* 文本前缀 |
||||
*/ |
||||
@Value("${spring.flyway.placeholder-prefix:#(}") |
||||
private String placeholderPrefix; |
||||
|
||||
/** |
||||
* 文本后缀 |
||||
*/ |
||||
@Value("${spring.flyway.placeholder-suffix:)}") |
||||
private String placeholderSuffix; |
||||
|
||||
/** |
||||
* 迁移sql脚本文件名称的后缀 |
||||
*/ |
||||
@Value("${spring.flyway.sql-migration-suffixes:.sql}") |
||||
private String sqlMigrationSuffixes; |
||||
|
||||
/** |
||||
* 迁移时是否进行校验，默认true |
||||
*/ |
||||
@Value("${spring.flyway.validate-on-migrate:true}") |
||||
private Boolean validateOnMigrate; |
||||
|
||||
/** |
||||
* 当迁移发现数据库非空且存在没有元数据的表时，自动执行基准迁移，新建schema_version表 |
||||
*/ |
||||
@Value("${spring.flyway.baseline-on-migrate:true}") |
||||
private Boolean baselineOnMigrate; |
||||
|
||||
/** |
||||
* 是否关闭要清除已有库下的表功能,生产环境必须为true,否则会删库，非常重要！！！ |
||||
*/ |
||||
@Value("${spring.flyway.clean-disabled:true}") |
||||
private Boolean cleanDisabled; |
||||
|
||||
@Bean |
||||
public void migrate() { |
||||
if(!enabled){ |
||||
return; |
||||
} |
||||
|
||||
DynamicRoutingDataSource ds = (DynamicRoutingDataSource) dataSource; |
||||
Map<String, DataSource> dataSources = ds.getDataSources(); |
||||
dataSources.forEach((k, v) -> { |
||||
if("master".equals(k)){ |
||||
String databaseType = environment.getProperty("spring.datasource.dynamic.datasource." + k + ".url"); |
||||
if (databaseType != null && databaseType.contains("mysql")) { |
||||
try { |
||||
Flyway flyway = Flyway.configure() |
||||
.dataSource(v) |
||||
.locations(locations) |
||||
.encoding(encoding) |
||||
.sqlMigrationPrefix(sqlMigrationPrefix) |
||||
.sqlMigrationSeparator(sqlMigrationSeparator) |
||||
.placeholderPrefix(placeholderPrefix) |
||||
.placeholderSuffix(placeholderSuffix) |
||||
.sqlMigrationSuffixes(sqlMigrationSuffixes) |
||||
.validateOnMigrate(validateOnMigrate) |
||||
.baselineOnMigrate(baselineOnMigrate) |
||||
.cleanDisabled(cleanDisabled) |
||||
.load(); |
||||
flyway.migrate(); |
||||
log.info("【升级提示】平台集成了MySQL库的Flyway，数据库版本自动升级! "); |
||||
} catch (FlywayException e) { |
||||
log.error("【升级提示】flyway执行sql脚本失败", e); |
||||
} |
||||
} else { |
||||
log.warn("【升级提示】平台只集成了MySQL库的Flyway，实现了数据库版本自动升级! 其他类型的数据库，您可以考虑手工升级~"); |
||||
} |
||||
} |
||||
}); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue