Remove os.exit in Run (kube-burner#704)

kube-burner-ocp workloads are using workload helpers for Run command. However this Run() has os.exit(), which is blocking kube-burner-ocp to run any PostRun() commands. As we removed os.exit() here, kube-burner-ocp workloads explicitly need to define PostRun() with os.exit(). kube-burner-ocp PR kube-burner/kube-burner-ocp#115 which adds PostRun() in all workloads Signed-off-by: venkataanil <[email protected]>
rsevilla87 · Oct 1, 2024 · 3a46b55 · 3a46b55
1 parent ead45c5
commit 3a46b55
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/workloads/helpers.go b/pkg/workloads/helpers.go
@@ -44,7 +44,7 @@ func NewWorkloadHelper(config Config, embedConfig embed.FS, kubeClientProvider *
 	return wh
 }
 
-func (wh *WorkloadHelper) Run(workload string) {
+func (wh *WorkloadHelper) Run(workload string) int {
 	var f io.Reader
 	var rc int
 	var err error
@@ -88,8 +88,8 @@ func (wh *WorkloadHelper) Run(workload string) {
 	if err != nil {
 		log.Error(err.Error())
 	}
-	log.Info("👋 Exiting kube-burner ", wh.UUID)
-	os.Exit(rc)
+	log.Infof("👋 kube-burner run completed with rc %d for UUID %s", rc, wh.UUID)
+	return rc
 }
 
 // ExtractWorkload extracts the given workload and metrics profile to the current directory