From 97de0b5f0cf14fcf677a40cc4081174bddbb4cda Mon Sep 17 00:00:00 2001
From: JiaJu Zhuang <5152853@qq.com>
Date: Sat, 29 Jul 2023 21:30:30 +0800
Subject: [PATCH] Starting Flyway No More Repeatedly

---
 .../src/main/java/ai/chat2db/server/start/Application.java     | 2 +-
 .../start/config/config/AsyncContextRefreshedListener.java     | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/Application.java b/chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/Application.java
index 0b9c9387..847acf27 100644
--- a/chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/Application.java
+++ b/chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/Application.java
@@ -35,7 +35,7 @@ public class Application {
         String currentVersion = ConfigUtils.getLocalVersion();
         ConfigJson configJson = ConfigUtils.getConfig();
         // Represents that the current version has been successfully launched
-        if (StringUtils.equals(currentVersion, configJson.getLatestStartupSuccessVersion())) {
+        if (StringUtils.isNotBlank(currentVersion) && StringUtils.equals(currentVersion, configJson.getLatestStartupSuccessVersion())) {
             // Flyway doesn't need to start every time to increase startup speed
             args = ArrayUtils.add(args, "--spring.flyway.enabled=false");
             log.info("The current version {} has been successfully launched once and will no longer load Flyway.",
diff --git a/chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/config/config/AsyncContextRefreshedListener.java b/chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/config/config/AsyncContextRefreshedListener.java
index 02a146a1..65eacf3f 100644
--- a/chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/config/config/AsyncContextRefreshedListener.java
+++ b/chat2db-server/chat2db-server-start/src/main/java/ai/chat2db/server/start/config/config/AsyncContextRefreshedListener.java
@@ -23,7 +23,8 @@ public class AsyncContextRefreshedListener implements ApplicationListener<Contex
         // Successfully set up startup
         String currentVersion = ConfigUtils.getLocalVersion();
         ConfigJson configJson = ConfigUtils.getConfig();
-        if (!StringUtils.equals(currentVersion, configJson.getLatestStartupSuccessVersion())) {
+        if (StringUtils.isNotBlank(currentVersion) && !StringUtils.equals(currentVersion,
+            configJson.getLatestStartupSuccessVersion())) {
             configJson.setLatestStartupSuccessVersion(currentVersion);
             ConfigUtils.setConfig(configJson);
         }