Handle termination signals

yankeexe · Sep 5, 2021 · 2791943 · 2791943
1 parent d4e4f83
commit 2791943
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/cmd/search.go b/cmd/search.go
@@ -17,6 +17,8 @@ package cmd
 import (
 	"fmt"
 	"os"
+	"os/signal"
+	"syscall"
 
 	"github.com/gookit/color"
 	fuzzyfinder "github.com/ktr0731/go-fuzzyfinder"
@@ -48,6 +50,15 @@ var searchCmd = &cobra.Command{
 			os.Exit(1)
 		}
 
+		killSignal := make(chan os.Signal, 1)
+		signal.Notify(killSignal, os.Interrupt, syscall.SIGTERM)
+
+		go func() {
+			<-killSignal
+			fmt.Println("Search stopped")
+			os.Exit(130)
+		}()
+
 		searchQuery := args[0]
 		res, err := aqi.HTTPClient.Fetch(searchQuery)
 		if err != nil {