better logname

dasbd72 · Nov 21, 2023 · d75318e · d75318e
1 parent 30f761a
commit d75318e
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/benchmark/cmd/benchmark/benchmark.go b/benchmark/cmd/benchmark/benchmark.go
@@ -31,6 +31,12 @@ func init() {
 	flag.StringVar(&flags.LogDir, "log-dir", "logs", "log directory")
 
 	flag.Parse()
+
+	if flags.ForceRemote {
+		flags.LogName = fmt.Sprintf("%s-remote-%s", flags.WorkflowType, time.Now().Format("06010-21504"))
+	} else {
+		flags.LogName = fmt.Sprintf("%s-%s", flags.WorkflowType, time.Now().Format("06010-21504"))
+	}
 }
 
 func main() {
@@ -43,7 +49,7 @@ func main() {
 		panic(err)
 	}
 	// open log file
-	f, err := os.OpenFile(fmt.Sprintf("%s/%s-%d.log", flags.LogDir, flags.WorkflowType, time.Now().Unix()), os.O_RDWR|os.O_CREATE, 0666)
+	f, err := os.OpenFile(fmt.Sprintf("%s/%s.log", flags.LogDir, flags.LogName), os.O_RDWR|os.O_CREATE, 0666)
 	if err != nil {
 		panic(err)
 	}
@@ -219,7 +225,7 @@ func benchmark(flags Flags) {
 	log.Println(string(p))
 
 	// Write json file
-	f, err := os.OpenFile(fmt.Sprintf("%s/%s-%d.json", flags.LogDir, flags.WorkflowType, time.Now().Unix()), os.O_RDWR|os.O_CREATE, 0666)
+	f, err := os.OpenFile(fmt.Sprintf("%s/%s.json", flags.LogDir, flags.LogName), os.O_RDWR|os.O_CREATE, 0666)
 	if err != nil {
 		panic(err)
 	}

diff --git a/benchmark/cmd/benchmark/types.go b/benchmark/cmd/benchmark/types.go
@@ -10,6 +10,7 @@ type Flags struct {
 	Warmup       bool    `json:"warmup"`
 	WorkflowType string  `json:"workflow_type"`
 	LogDir       string  `json:"log_dir"`
+	LogName      string  `json:"log_name"`
 }
 
 type BenchmarkResult struct {

diff --git a/benchmark/run.sh b/benchmark/run.sh
@@ -5,4 +5,4 @@ go mod download
 go run cmd/benchmark/* -tasks 5 -distribution seq -workflow-type ImageProcessing -force-remote -url-postfix default.192.168.100.0.sslip.io
 go run cmd/benchmark/* -tasks 5 -distribution seq -workflow-type ImageProcessing -url-postfix default.192.168.100.0.sslip.io
 go run cmd/benchmark/* -tasks 5 -distribution seq -workflow-type VideoProcessing -force-remote -url-postfix default.192.168.100.0.sslip.io
-go run cmd/benchmark/* -tasks 5 -distribution seq -workflow-type VideoProcessing -url-postfix default.192.168.100.0.sslip.io
+go run cmd/benchmark/* -tasks 5 -distribution seq -workflow-type VideoProcessing -url-postfix default.192.168.100.0.sslip.io