enhance: add hard exit, #78

chainreactors · Nov 1, 2024 · de12d56 · de12d56
1 parent 02162cf
commit de12d56
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/cmd/cmd.go b/cmd/cmd.go
@@ -149,6 +149,15 @@ func Spray() {
 	}
 
 	ctx, canceler := context.WithTimeout(context.Background(), time.Duration(runner.Deadline)*time.Second)
+	go func() {
+		select {
+		case <-ctx.Done():
+			time.Sleep(10 * time.Second)
+			logs.Log.Errorf("deadline and timeout not work, hard exit!!!")
+			os.Exit(0)
+		}
+	}()
+
 	go func() {
 		exitChan := make(chan os.Signal, 2)
 		signal.Notify(exitChan, os.Interrupt, syscall.SIGTERM)