tsurumure 1 month ago
parent
commit
1599cc7bc2
1 changed files with 13 additions and 6 deletions
  1. 13 6
      src/main/java/com/backendsys/modules/PublicController.java

+ 13 - 6
src/main/java/com/backendsys/modules/PublicController.java

@@ -27,14 +27,21 @@ public class PublicController {
     @GetMapping("/api/version")
     public String getVersion() {
 
-        Integer level = 1;
         String activeProfile = env.getActiveProfiles()[0];
-        if ("local".equals(activeProfile)) level = 2;
+        if ("local".equals(activeProfile)) {
+            // jar 包所在目录 + 文件名
+            String path = FileUtil.getParent(
+                    this.getClass().getProtectionDomain().getCodeSource().getLocation().getFile(), 2) + "./version.txt";
+            return FileUtil.readUtf8String(path).trim();
+        } else {
+            // 获取 jar 启动时所在的目录(即工作目录)
+            String basePath = System.getProperty("user.dir");
+            String versionFile = basePath + "./version.txt"; // 上一级目录
+            return FileUtil.exist(versionFile)
+                    ? FileUtil.readUtf8String(versionFile).trim()
+                    : "0.0.0";
+        }
 
-        // jar 包所在目录 + 文件名
-        String path = FileUtil.getParent(
-                this.getClass().getProtectionDomain().getCodeSource().getLocation().getFile(), level) + "./version.txt";
-        return FileUtil.readUtf8String(path).trim();
     }
 
 }