From 50796875f7dda25040666f22d9eed0176f71d883 Mon Sep 17 00:00:00 2001
From: Zzm0809 <934230207@qq.com>
Date: Thu, 15 Aug 2024 22:38:24 +0800
Subject: [PATCH] Refactor the method of obtaining user.dir (#3708)

Signed-off-by: Zzm0809 <934230207@qq.com>
Co-authored-by: Zzm0809 <Zzm0809@users.noreply.github.com>
---
 .../dinky/context/ConsoleContextHolder.java   |  7 +--
 .../dinky/controller/CatalogueController.java |  4 +-
 .../dinky/controller/DownloadController.java  |  5 ++-
 .../dinky/controller/SystemController.java    |  2 +-
 .../dinky/service/impl/SystemServiceImpl.java |  2 +-
 .../java/org/dinky/utils/DirUtilTest.java     |  6 +--
 .../org/dinky/app/flinksql/Submitter.java     |  3 +-
 .../org/dinky/data/constant/DirConstant.java  | 44 +++++++++++++++++--
 .../org/dinky/utils/DockerClientUtils.java    |  3 +-
 .../main/java/org/dinky/utils/URLUtils.java   |  3 +-
 .../dinky/function/constant/PathConstant.java | 21 ++++-----
 .../gateway/kubernetes/KubernetesGateway.java |  3 +-
 .../org/dinky/gateway/yarn/YarnGateway.java   |  3 +-
 script/bin/auto.sh                            |  8 ++--
 14 files changed, 76 insertions(+), 38 deletions(-)

diff --git a/dinky-admin/src/main/java/org/dinky/context/ConsoleContextHolder.java b/dinky-admin/src/main/java/org/dinky/context/ConsoleContextHolder.java
index 409842818c..73f0218363 100644
--- a/dinky-admin/src/main/java/org/dinky/context/ConsoleContextHolder.java
+++ b/dinky-admin/src/main/java/org/dinky/context/ConsoleContextHolder.java
@@ -20,6 +20,7 @@
 package org.dinky.context;
 
 import org.dinky.aop.ProcessAspect;
+import org.dinky.data.constant.DirConstant;
 import org.dinky.data.enums.ProcessStatus;
 import org.dinky.data.enums.ProcessStepType;
 import org.dinky.data.enums.ProcessType;
@@ -98,7 +99,7 @@ public ProcessEntity getProcess(String processName) {
             return logPross.get(processName);
         }
         try {
-            String filePath = String.format("%s/tmp/log/%s.json", System.getProperty("user.dir"), processName);
+            String filePath = String.format("%s/log/%s.json", DirConstant.getTempRootDir(), processName);
             String string = FileUtil.readString(filePath, StandardCharsets.UTF_8);
             ProcessEntity process = JSONObject.parseObject(string, ProcessEntity.class);
             if (process.getStatus().isActiveStatus()) {
@@ -113,7 +114,7 @@ public ProcessEntity getProcess(String processName) {
 
     public boolean clearProcessLog(String processName) {
         // find process and delete
-        String filePath = String.format("%s/tmp/log/%s.json", System.getProperty("user.dir"), processName);
+        String filePath = String.format("%s/log/%s.json", DirConstant.getTempRootDir(), processName);
         if (FileUtil.exist(filePath)) {
             return FileUtil.del(filePath);
         }
@@ -236,7 +237,7 @@ public synchronized void finishedProcess(String processName, ProcessStatus statu
             if (e != null) {
                 appendLog(processName, null, LogUtil.getError(e.getCause()), true);
             }
-            String filePath = String.format("%s/tmp/log/%s.json", System.getProperty("user.dir"), processName);
+            String filePath = String.format("%s/log/%s.json", DirConstant.getTempRootDir(), processName);
             if (FileUtil.exist(filePath)) {
                 Assert.isTrue(FileUtil.del(filePath));
             }
diff --git a/dinky-admin/src/main/java/org/dinky/controller/CatalogueController.java b/dinky-admin/src/main/java/org/dinky/controller/CatalogueController.java
index 18ff3775ae..2e318b139e 100644
--- a/dinky-admin/src/main/java/org/dinky/controller/CatalogueController.java
+++ b/dinky-admin/src/main/java/org/dinky/controller/CatalogueController.java
@@ -23,6 +23,7 @@
 import org.dinky.data.annotations.CheckTaskOwner;
 import org.dinky.data.annotations.Log;
 import org.dinky.data.annotations.TaskId;
+import org.dinky.data.constant.DirConstant;
 import org.dinky.data.dto.CatalogueTaskDTO;
 import org.dinky.data.dto.CatalogueTreeQueryDTO;
 import org.dinky.data.enums.BusinessType;
@@ -30,7 +31,6 @@
 import org.dinky.data.model.Catalogue;
 import org.dinky.data.result.Result;
 import org.dinky.data.vo.TreeVo;
-import org.dinky.function.constant.PathConstant;
 import org.dinky.service.TaskService;
 import org.dinky.service.catalogue.CatalogueService;
 
@@ -77,7 +77,7 @@ public class CatalogueController {
     @ApiOperation("Upload Zip Package And Create Catalogue")
     public Result<String> upload(MultipartFile file, @PathVariable Integer id) {
         // 获取上传的路径
-        String filePath = PathConstant.WORK_DIR;
+        String filePath = DirConstant.getRootPath();
         // 获取源文件的名称
         String fileName = file.getOriginalFilename();
         String zipPath = filePath + File.separator + fileName;
diff --git a/dinky-admin/src/main/java/org/dinky/controller/DownloadController.java b/dinky-admin/src/main/java/org/dinky/controller/DownloadController.java
index b6d0fbf6de..65794a7fb3 100644
--- a/dinky-admin/src/main/java/org/dinky/controller/DownloadController.java
+++ b/dinky-admin/src/main/java/org/dinky/controller/DownloadController.java
@@ -20,6 +20,7 @@
 package org.dinky.controller;
 
 import org.dinky.assertion.Asserts;
+import org.dinky.data.constant.DirConstant;
 import org.dinky.data.exception.BusException;
 import org.dinky.data.model.FlinkUdfManifest;
 import org.dinky.function.constant.PathConstant;
@@ -107,8 +108,8 @@ public void downloadJavaUDF(@PathVariable Integer taskId, HttpServletResponse re
     @GetMapping("downloadAppJar/{version}")
     @ApiOperation("Download App Jar")
     public void downloadAppJar(@PathVariable String version, HttpServletResponse resp) {
-        List<File> files = FileUtil.loopFiles(
-                PathConstant.WORK_DIR + "/jar", pathname -> pathname.getName().contains("dinky-app-" + version));
+        List<File> files = FileUtil.loopFiles(DirConstant.getRootPath() + "/jar", pathname -> pathname.getName()
+                .contains("dinky-app-" + version));
         if (CollUtil.isNotEmpty(files)) {
             ServletUtil.write(resp, files.get(0));
         }
diff --git a/dinky-admin/src/main/java/org/dinky/controller/SystemController.java b/dinky-admin/src/main/java/org/dinky/controller/SystemController.java
index f19bf02905..282242f56a 100644
--- a/dinky-admin/src/main/java/org/dinky/controller/SystemController.java
+++ b/dinky-admin/src/main/java/org/dinky/controller/SystemController.java
@@ -78,7 +78,7 @@ public Result<List<TreeNodeDTO>> listLogDir() {
     @ApiOperation("Get Root Log File Content")
     @SaCheckPermission(PermissionConstants.SYSTEM_SETTING_INFO_ROOT_LOG)
     public Result<String> getRootLog() {
-        return Result.data(systemService.readFile(DirConstant.ROOT_LOG_PATH));
+        return Result.data(systemService.readFile(DirConstant.getRootLog()));
     }
 
     /**
diff --git a/dinky-admin/src/main/java/org/dinky/service/impl/SystemServiceImpl.java b/dinky-admin/src/main/java/org/dinky/service/impl/SystemServiceImpl.java
index 769e32e4a3..e6233fd725 100644
--- a/dinky-admin/src/main/java/org/dinky/service/impl/SystemServiceImpl.java
+++ b/dinky-admin/src/main/java/org/dinky/service/impl/SystemServiceImpl.java
@@ -62,7 +62,7 @@ public class SystemServiceImpl implements SystemService {
 
     @Override
     public List<TreeNodeDTO> listLogDir() {
-        File systemLogFiles = TreeUtil.getFilesOfDir(DirConstant.LOG_DIR_PATH);
+        File systemLogFiles = TreeUtil.getFilesOfDir(DirConstant.getRootLogsPath());
         return TreeUtil.treeNodeData(systemLogFiles, false);
     }
 
diff --git a/dinky-admin/src/test/java/org/dinky/utils/DirUtilTest.java b/dinky-admin/src/test/java/org/dinky/utils/DirUtilTest.java
index 4c1b57e24a..8c150e437a 100644
--- a/dinky-admin/src/test/java/org/dinky/utils/DirUtilTest.java
+++ b/dinky-admin/src/test/java/org/dinky/utils/DirUtilTest.java
@@ -39,21 +39,21 @@ public class DirUtilTest {
     @Ignore
     @Test
     public void testListDirByPath() {
-        List<FileNode> dirList = DirUtil.listDirByPath(DirConstant.LOG_DIR_PATH);
+        List<FileNode> dirList = DirUtil.listDirByPath(DirConstant.getRootLogsPath());
         Assertions.assertThat(dirList).isNotNull();
     }
 
     @Ignore
     @Test
     public void testReadFile() {
-        String result = DirUtil.readFile(DirConstant.LOG_DIR_PATH + "/dinky.log");
+        String result = DirUtil.readFile(DirConstant.getRootLog());
         Assertions.assertThat(result).isNotNull();
     }
 
     @Ignore
     @Test
     public void testReadRootLog() {
-        String result = DirUtil.readFile(DirConstant.ROOT_LOG_PATH);
+        String result = DirUtil.readFile(DirConstant.getRootLog());
         Assertions.assertThat(result).isNotNull();
     }
 }
diff --git a/dinky-app/dinky-app-base/src/main/java/org/dinky/app/flinksql/Submitter.java b/dinky-app/dinky-app-base/src/main/java/org/dinky/app/flinksql/Submitter.java
index 3a1a01c35b..5a5c599c1e 100644
--- a/dinky-app/dinky-app-base/src/main/java/org/dinky/app/flinksql/Submitter.java
+++ b/dinky-app/dinky-app-base/src/main/java/org/dinky/app/flinksql/Submitter.java
@@ -30,6 +30,7 @@
 import org.dinky.constant.FlinkSQLConstant;
 import org.dinky.data.app.AppParamConfig;
 import org.dinky.data.app.AppTask;
+import org.dinky.data.constant.DirConstant;
 import org.dinky.data.enums.GatewayType;
 import org.dinky.data.model.SystemConfiguration;
 import org.dinky.executor.Executor;
@@ -161,7 +162,7 @@ private static String readSql(Executor executor) {
         if (!sqlFile.exists()) {
             sqlFile = new File(confDir, sqlFileName);
             if (!sqlFile.exists()) {
-                log.error("sql file not found,current dir:{},conf dir:{}", System.getProperty("user.dir"), confDir);
+                log.error("sql file not found,current dir:{},conf dir:{}", DirConstant.getRootPath(), confDir);
                 throw new RuntimeException("sql file not found");
             }
         }
diff --git a/dinky-common/src/main/java/org/dinky/data/constant/DirConstant.java b/dinky-common/src/main/java/org/dinky/data/constant/DirConstant.java
index c28011e21f..7b5d4bd4d9 100644
--- a/dinky-common/src/main/java/org/dinky/data/constant/DirConstant.java
+++ b/dinky-common/src/main/java/org/dinky/data/constant/DirConstant.java
@@ -21,6 +21,7 @@
 
 import java.io.File;
 
+import cn.hutool.core.util.StrUtil;
 import cn.hutool.system.SystemUtil;
 
 /**
@@ -29,9 +30,44 @@
  * @since 2022/10/15 18:37
  */
 public class DirConstant {
-
+    public static final String DINKY_ENV_OR_HOME_ROOT_DIR_NAME = "dinky.root.path";
     public static final String FILE_SEPARATOR = File.separator;
-    public static final String ROOT_PATH = System.getProperty(SystemUtil.USER_DIR);
-    public static final String LOG_DIR_PATH = ROOT_PATH + FILE_SEPARATOR + "logs";
-    public static final String ROOT_LOG_PATH = LOG_DIR_PATH + FILE_SEPARATOR + "dinky.log";
+
+    // user dir path, default is project root path
+    public static final String USER_DIR_ENV_ROOT_PATH = System.getProperty(SystemUtil.USER_DIR);
+
+    // dinky env or home root path
+    public static final String DINKY_ENV_OR_HOME_ROOT_DIR = System.getProperty(DINKY_ENV_OR_HOME_ROOT_DIR_NAME);
+
+    /**
+     * get a root path of dinky
+     * @return root path of dinky
+     */
+    public static String getRootPath() {
+        return StrUtil.isEmpty(DINKY_ENV_OR_HOME_ROOT_DIR) ? USER_DIR_ENV_ROOT_PATH : DINKY_ENV_OR_HOME_ROOT_DIR;
+    }
+
+    /**
+     * get a root logs path of dinky
+     * @return root logs path of dinky
+     */
+    public static String getRootLogsPath() {
+        return getRootPath() + FILE_SEPARATOR + "logs";
+    }
+
+    /**
+     * get root log  of dinky
+     * @return root log of dinky
+     */
+    public static String getRootLog() {
+        return getRootLogsPath() + FILE_SEPARATOR + "dinky.log";
+    }
+
+    /**
+     * get tmp dir root of dinky
+     * @return tmp dir root of dinky
+     */
+    public static String getTempRootDir() {
+        return getRootPath() + FILE_SEPARATOR + "tmp";
+    }
 }
diff --git a/dinky-common/src/main/java/org/dinky/utils/DockerClientUtils.java b/dinky-common/src/main/java/org/dinky/utils/DockerClientUtils.java
index d9a90e9a4a..2fc82a58fd 100644
--- a/dinky-common/src/main/java/org/dinky/utils/DockerClientUtils.java
+++ b/dinky-common/src/main/java/org/dinky/utils/DockerClientUtils.java
@@ -20,6 +20,7 @@
 package org.dinky.utils;
 
 import org.dinky.config.Docker;
+import org.dinky.data.constant.DirConstant;
 import org.dinky.docker.DockerClientBuilder;
 
 import java.io.File;
@@ -50,7 +51,7 @@ public class DockerClientUtils {
     public DockerClientUtils(Docker docker) {
         this.docker = docker;
         this.dockerfile = FileUtil.writeUtf8String(
-                docker.getDockerfile(), System.getProperty("user.dir") + "/tmp/dockerfile/" + UUID.randomUUID());
+                docker.getDockerfile(), DirConstant.getTempRootDir() + "/dockerfile/" + UUID.randomUUID());
         dockerClient = DockerClientBuilder.getInstance(DefaultDockerClientConfig.createDefaultConfigBuilder()
                         .withDockerHost(docker.getInstance())
                         .withRegistryUrl(docker.getRegistryUrl())
diff --git a/dinky-common/src/main/java/org/dinky/utils/URLUtils.java b/dinky-common/src/main/java/org/dinky/utils/URLUtils.java
index 35e1f42222..fa92e81844 100644
--- a/dinky-common/src/main/java/org/dinky/utils/URLUtils.java
+++ b/dinky-common/src/main/java/org/dinky/utils/URLUtils.java
@@ -20,6 +20,7 @@
 package org.dinky.utils;
 
 import org.dinky.assertion.Asserts;
+import org.dinky.data.constant.DirConstant;
 
 import java.io.File;
 import java.net.MalformedURLException;
@@ -39,7 +40,7 @@
  * @since 0.7.0
  */
 public class URLUtils {
-    private static final String TMP_PATH = StrUtil.join(File.separator, System.getProperty("user.dir"), "tmp");
+    private static final String TMP_PATH = DirConstant.getTempRootDir();
 
     /**
      * url download file to local
diff --git a/dinky-function/src/main/java/org/dinky/function/constant/PathConstant.java b/dinky-function/src/main/java/org/dinky/function/constant/PathConstant.java
index dd74cf62ea..83ac2f2f55 100644
--- a/dinky-function/src/main/java/org/dinky/function/constant/PathConstant.java
+++ b/dinky-function/src/main/java/org/dinky/function/constant/PathConstant.java
@@ -19,34 +19,29 @@
 
 package org.dinky.function.constant;
 
+import org.dinky.data.constant.DirConstant;
+
 import org.apache.flink.table.catalog.FunctionLanguage;
 
 import java.io.File;
 
 import cn.hutool.core.util.StrUtil;
 
-/**
- * 文件路径常量
- *
- * @since 0.6.8
- */
 public class PathConstant {
 
-    /** 基本路径，dinky 部署的路径 */
-    public static final String WORK_DIR = System.getProperty("user.dir");
-    /** tmp路径 */
-    public static final String TMP_PATH = WORK_DIR + File.separator + "tmp" + File.separator;
+    /** Tmp path */
+    public static final String TMP_PATH = DirConstant.getTempRootDir() + File.separator;
 
-    /** udf路径 */
+    /** UDF path */
     public static final String UDF_PATH = TMP_PATH + "udf" + File.separator;
 
     public static final String COMPILER = "compiler";
     public static final String PACKAGE = "package";
-    /** udf jar规则 */
+    /** UDF jar rules */
     public static final String UDF_JAR_RULE = "udf-\\d+.jar";
-    /** udf版本规则 */
+    /** UDF version rules */
     public static final String UDF_VERSION_RULE = "\\d+";
-    /** udf jar tmp名字 */
+    /**Udf jar tmp name */
     public static final String UDF_JAR_TMP_NAME = "udf-tmp.jar";
 
     public static final String UDF_JAR_NAME = "udf.jar";
diff --git a/dinky-gateway/src/main/java/org/dinky/gateway/kubernetes/KubernetesGateway.java b/dinky-gateway/src/main/java/org/dinky/gateway/kubernetes/KubernetesGateway.java
index 1e735fba46..f0eef28b68 100644
--- a/dinky-gateway/src/main/java/org/dinky/gateway/kubernetes/KubernetesGateway.java
+++ b/dinky-gateway/src/main/java/org/dinky/gateway/kubernetes/KubernetesGateway.java
@@ -20,6 +20,7 @@
 package org.dinky.gateway.kubernetes;
 
 import org.dinky.assertion.Asserts;
+import org.dinky.data.constant.DirConstant;
 import org.dinky.data.enums.Status;
 import org.dinky.gateway.AbstractGateway;
 import org.dinky.gateway.config.FlinkConfig;
@@ -72,7 +73,7 @@ public abstract class KubernetesGateway extends AbstractGateway {
     private Pod defaultPodTemplate;
 
     private K8sClientHelper k8sClientHelper;
-    private String tmpConfDir = String.format("%s/tmp/kubernets/%s", System.getProperty("user.dir"), UUID.randomUUID());
+    private String tmpConfDir = String.format("%s/kubernetes/%s", DirConstant.getTempRootDir(), UUID.randomUUID());
 
     public KubernetesGateway() {}
 
diff --git a/dinky-gateway/src/main/java/org/dinky/gateway/yarn/YarnGateway.java b/dinky-gateway/src/main/java/org/dinky/gateway/yarn/YarnGateway.java
index ef13897ad2..c0bc96ca9d 100644
--- a/dinky-gateway/src/main/java/org/dinky/gateway/yarn/YarnGateway.java
+++ b/dinky-gateway/src/main/java/org/dinky/gateway/yarn/YarnGateway.java
@@ -22,6 +22,7 @@
 import org.dinky.assertion.Asserts;
 import org.dinky.constant.CustomerConfigureOptions;
 import org.dinky.context.FlinkUdfPathContextHolder;
+import org.dinky.data.constant.DirConstant;
 import org.dinky.data.enums.JobStatus;
 import org.dinky.data.model.CustomConfig;
 import org.dinky.data.model.SystemConfiguration;
@@ -98,7 +99,7 @@
 public abstract class YarnGateway extends AbstractGateway {
     private static final String HTML_TAG_REGEX = "<pre>(.*)</pre>";
     private final String TMP_SQL_EXEC_DIR =
-            String.format("%s/tmp/sql-exec/%s", System.getProperty("user.dir"), UUID.randomUUID());
+            String.format("%s/sql-exec/%s", DirConstant.getTempRootDir(), UUID.randomUUID());
 
     protected YarnConfiguration yarnConfiguration;
 
diff --git a/script/bin/auto.sh b/script/bin/auto.sh
index e935eee6fb..55e65745fe 100644
--- a/script/bin/auto.sh
+++ b/script/bin/auto.sh
@@ -62,7 +62,7 @@ fi
 # OOM dump path
 OOM_OPT="-XX:+HeapDumpOnOutOfMemoryError -XX:HeapDumpPath=${APP_HOME}/logs/heapdump.hprof"
 # JVM parameters and log path
-PARAMS_OPT="-Ddinky.logs.path=${DINKY_LOG_PATH} -Ddruid.mysql.usePingMethod=false -Dlog4j2.isThreadContextMapInheritable=true"
+PARAMS_OPT="-Ddinky.logs.path=${DINKY_LOG_PATH} -Ddinky.root.path=${APP_HOME} -Ddruid.mysql.usePingMethod=false -Dlog4j2.isThreadContextMapInheritable=true"
 # JAR parameters
 JAR_PARAMS_OPT="--logging.config=${LOG_CONFIG}"
 # JMX path
@@ -102,7 +102,7 @@ start() {
   assertIsInputVersion
   updatePid
   if [ -z "$pid" ]; then
-    nohup java ${PARAMS_OPT} ${JVM_OPTS} ${OOM_OPT} ${GC_OPT} ${PARAMS_OPT} -Xverify:none -cp "${CLASS_PATH}" org.dinky.Dinky ${JAR_PARAMS_OPT}  > ${DINKY_LOG_PATH}/dinky-start.log 2>&1 &
+    nohup java ${PARAMS_OPT} ${JVM_OPTS} ${OOM_OPT} ${GC_OPT} -Xverify:none -cp "${CLASS_PATH}" org.dinky.Dinky ${JAR_PARAMS_OPT}  > ${DINKY_LOG_PATH}/dinky-start.log 2>&1 &
     echo $! >"${PID_PATH}"/${PID_FILE}
     echo "........................................Start Dinky Done........................................"
     echo "current log path : ${DINKY_LOG_PATH}/dinky-start.log , you can execute tail -fn1000 ${DINKY_LOG_PATH}/dinky-start.log to watch the log"
@@ -115,7 +115,7 @@ startOnPending() {
   assertIsInputVersion
   updatePid
   if [ -z "$pid" ]; then
-    java ${PARAMS_OPT} ${JVM_OPTS} ${OOM_OPT} ${GC_OPT} ${PARAMS_OPT} -Xverify:none -cp "${CLASS_PATH}" org.dinky.Dinky  ${JAR_PARAMS_OPT}
+    java ${PARAMS_OPT} ${JVM_OPTS} ${OOM_OPT} ${GC_OPT} -Xverify:none -cp "${CLASS_PATH}" org.dinky.Dinky  ${JAR_PARAMS_OPT}
     echo "........................................Start Dinky Successfully........................................"
   else
     echo "Dinky pid $pid is in ${PID_PATH}/${PID_FILE}, Please stop first !!!"
@@ -126,7 +126,7 @@ startWithJmx() {
   assertIsInputVersion
   updatePid
   if [ -z "$pid" ]; then
-    nohup java ${PARAMS_OPT} ${JVM_OPTS} ${OOM_OPT} ${GC_OPT} ${PARAMS_OPT} -Xverify:none "${JMX}" -cp "${CLASS_PATH}" org.dinky.Dinky  ${JAR_PARAMS_OPT}  > ${DINKY_LOG_PATH}/dinky-start.log 2>&1 &
+    nohup java ${PARAMS_OPT} ${JVM_OPTS} ${OOM_OPT} ${GC_OPT} -Xverify:none "${JMX}" -cp "${CLASS_PATH}" org.dinky.Dinky  ${JAR_PARAMS_OPT}  > ${DINKY_LOG_PATH}/dinky-start.log 2>&1 &
 #    echo $! >"${PID_PATH}"/${PID_FILE}
     updatePid
     echo "........................................Start Dinky with Jmx Successfully.....................................