Przeglądaj źródła

add KUIPER_BASE_LOCATION_KEY into env

RockyJin 5 lat temu
rodzic
commit
66754a7db5
1 zmienionych plików z 7 dodań i 0 usunięć
  1. 7 0
      common/util.go

+ 7 - 0
common/util.go

@@ -21,6 +21,8 @@ const (
 	log_dir = "/log/"
 )
 
+const KUIPER_BASE_LOCATION_KEY string = "KUIPER_BASE_LOCATION_KEY"
+
 var (
 	Log *logrus.Logger
 	Config *XStreamConf
@@ -219,6 +221,11 @@ func GetLoc(subdir string)(string, error) {
 	if err != nil {
 		return "", err
 	}
+
+	if loc := os.Getenv(KUIPER_BASE_LOCATION_KEY); loc != "" {
+		dir = loc
+	}
+
 	confDir := dir + subdir
 	if _, err := os.Stat(confDir); os.IsNotExist(err) {
 		lastdir := dir