feat: add CUSTOM type to support custom log path set by log-path flag…

…, conveniently saving the log file in custom path, for the logs persistence scene.

Signed-off-by: NigelWu95 <[email protected]>

util/log.go

@@ -29,19 +29,25 @@ import (
 const (
 	Blade = 1
 	Bin   = 2
+	Custom= 3
 )
 
 const BladeLog = "chaosblade.log"
 
 var (
 	Debug    bool
+	LogPath  string
 	LogLevel string
 )
 
 func AddDebugFlag() {
 	flag.BoolVar(&Debug, "debug", false, "set debug mode")
 }
 
+func AddLogPathFlag() {
+	flag.StringVar(&LogPath, "log-path", GetProgramPath(), "the directory path to save chaosblade.log.")
+}
+
 func AddLogLevelFlag() {
 	flag.StringVar(&LogLevel, "log-level", "info", "level of logging wanted.")
 }
@@ -127,6 +133,8 @@ func GetLogPath(programType int) (string, error) {
 		binDir = GetProgramPath()
 	case Bin:
 		binDir = GetProgramParentPath()
+	case Custom:
+		binDir = LogPath
 	default:
 		binDir = GetProgramPath()
 	}