Added SIGTERM signal handling for graceful shutdown of Litmus-Checker (…

…#376) * Added SIGTERM signal handling in checker Signed-off-by: Jonsy13 <[email protected]>
litmuschaos · Jun 19, 2023 · c1bf5dd · c1bf5dd
1 parent 5fb73d5
commit c1bf5dd
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/custom/litmus-checker/main.go b/custom/litmus-checker/main.go
@@ -4,13 +4,18 @@ import (
 	"flag"
 	"io/ioutil"
 	"log"
+	"os"
+	"os/signal"
+	"syscall"
 
 	chaos_checker "github.com/gdsoumya/resourceChecker/pkg/chaos-checker"
 	"github.com/gdsoumya/resourceChecker/pkg/k8s"
 	"github.com/gdsoumya/resourceChecker/pkg/util"
 )
 
 func main() {
+	signalChannel := make(chan os.Signal)
+	signal.Notify(signalChannel, os.Interrupt, syscall.SIGTERM)
 	kubeconfig := flag.String("kubeconfig", "", "absolute path to the kubeconfig file")
 	file := flag.String("file", "", "absolute path to the chaosengine yaml")
 	engineFile := flag.String("saveName", "", "absolute path to the output file")
@@ -41,6 +46,14 @@ func main() {
 		Namespace: resp.GetNamespace(),
 		Selectors: "",
 	}
+
+	// Required, While aborting a Chaos Experiment, wait-container (argo-exec) sends SIGTERM signal to other (main) containers for aborting Argo-Workflow Pod
+	go func() {
+		<-signalChannel
+		log.Print("SIGTERM SIGNAL RECEIVED, Shutting down litmus-checker...")
+		os.Exit(0)
+	}()
+
 	log.Print("Created Resource Details: \n", resDef)
 	chaos_checker.CheckChaos(kubeconfig, resDef)
 }