Revert chaos when error during drain for node-drain experiments (#668)

- Added an call to uncordonNode in case of an error of the drainNode function Signed-off-by: Calvin Audier <[email protected]>
litmuschaos · Sep 21, 2023 · b710216 · b710216
1 parent 392ea29
commit b710216
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/chaoslib/litmus/node-drain/lib/node-drain.go b/chaoslib/litmus/node-drain/lib/node-drain.go
@@ -76,6 +76,10 @@ func PrepareNodeDrain(experimentsDetails *experimentTypes.ExperimentDetails, cli
 
 	// Drain the application node
 	if err := drainNode(experimentsDetails, clients, chaosDetails); err != nil {
+		log.Info("[Revert]: Reverting chaos because error during draining of node")
+		if uncordonErr := uncordonNode(experimentsDetails, clients, chaosDetails); uncordonErr != nil {
+			return cerrors.PreserveError{ErrString: fmt.Sprintf("[%s,%s]", stacktrace.RootCause(err).Error(), stacktrace.RootCause(uncordonErr).Error())}
+		}
 		return stacktrace.Propagate(err, "could not drain node")
 	}