NDEV-18337 different file for custom job defn

nirmata · Feb 26, 2024 · f8402fd · f8402fd
1 parent 7d19783
commit f8402fd
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/pkg/kubebench/embed.go b/pkg/kubebench/embed.go
@@ -2,6 +2,7 @@ package kubebench
 
 import (
 	"embed"
+	"os"
 )
 
 //go:embed jobs/*.yaml
@@ -14,10 +15,13 @@ func embedYAMLs(benchmark, clusterType, customJobFile string) ([]byte, error) {
 	jobFile := "jobs/job.yaml"
 	if customJobFile != "" {
 		jobFile = customJobFile
-	} else if clusterType != "" {
-		jobFile = "jobs/job-" + clusterType + ".yaml"
+		data, err = os.ReadFile(customJobFile)
+	} else {
+		if clusterType != "" {
+			jobFile = "jobs/job-" + clusterType + ".yaml"
+		}
+		data, err = yamlDir.ReadFile(jobFile)
 	}
-	data, err = yamlDir.ReadFile(jobFile)
 	if err != nil {
 		return nil, err
 	}