From 0bf868d3cdf9226e178c076d3b588ed5207409a0 Mon Sep 17 00:00:00 2001
From: kongzy <kongzy>
Date: 星期五, 24 十一月 2023 17:51:40 +0800
Subject: [PATCH] merge

---
 assess-admin/src/main/java/com/gkhy/assess/admin/config/FlywayConfig.java |   24 ++++++++++++++++++++++++
 1 files changed, 24 insertions(+), 0 deletions(-)

diff --git a/assess-admin/src/main/java/com/gkhy/assess/admin/config/FlywayConfig.java b/assess-admin/src/main/java/com/gkhy/assess/admin/config/FlywayConfig.java
new file mode 100644
index 0000000..270d756
--- /dev/null
+++ b/assess-admin/src/main/java/com/gkhy/assess/admin/config/FlywayConfig.java
@@ -0,0 +1,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");
+        }
+    }
+}

--
Gitblit v1.9.2