Add check for setting HasProbeCompleted

Signed-off-by: nagesh bansal <[email protected]>
litmuschaos · Dec 15, 2023 · d17a02d · d17a02d
1 parent 9ec39bd
commit d17a02d
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/pkg/probe/cmdprobe.go b/pkg/probe/cmdprobe.go
@@ -337,6 +337,11 @@ loop:
 		select {
 		case <-chaosDetails.ProbeContext.Ctx.Done():
 			log.Info("Chaos Execution completed. Stopping Probes")
+			for index := range chaosresult.ProbeDetails {
+				if chaosresult.ProbeDetails[index].Name == probe.Name {
+					chaosresult.ProbeDetails[index].HasProbeCompleted = true
+				}
+			}
 			break loop
 		default:
 			err := triggerInlineCmdProbe(probe, chaosresult)

diff --git a/pkg/probe/k8sprobe.go b/pkg/probe/k8sprobe.go
@@ -141,6 +141,11 @@ loop:
 		select {
 		case <-chaosDetails.ProbeContext.Ctx.Done():
 			log.Info("Chaos Execution completed. Stopping Probes")
+			for index := range chaosresult.ProbeDetails {
+				if chaosresult.ProbeDetails[index].Name == probe.Name {
+					chaosresult.ProbeDetails[index].HasProbeCompleted = true
+				}
+			}
 			break loop
 
 		default:

diff --git a/pkg/probe/promProbe.go b/pkg/probe/promProbe.go
@@ -249,6 +249,11 @@ loop:
 		select {
 		case <-chaosDetails.ProbeContext.Ctx.Done():
 			log.Info("Chaos Execution completed. Stopping Probes")
+			for index := range chaosresult.ProbeDetails {
+				if chaosresult.ProbeDetails[index].Name == probe.Name {
+					chaosresult.ProbeDetails[index].HasProbeCompleted = true
+				}
+			}
 			break loop
 		default:
 			err = triggerPromProbe(probe, chaosresult)