feat: add ctrl+c handler

adhocore · Oct 20, 2024 · 8e9fb39 · 8e9fb39
1 parent a228487
commit 8e9fb39
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/cmd/fast/main.go b/cmd/fast/main.go
@@ -3,7 +3,10 @@ package main
 import (
 	"flag"
 	"log"
+	"os"
+	"os/signal"
 	"sync"
+	"syscall"
 	"time"
 
 	"github.com/adhocore/chin"
@@ -23,9 +26,17 @@ func main() {
 	s := chin.New().WithWait(&wg)
 	go s.Start()
 
+	var sig = make(chan os.Signal, 1)
+	signal.Notify(sig, os.Interrupt, syscall.SIGTERM, syscall.SIGABRT)
+
 	wg.Add(1)
 	go doFast(s, &wg, noUp)
 
+	go func() {
+		<-sig
+		s.Stop()
+	}()
+
 	wg.Wait()
 }