b

root27 · Jun 5, 2024 · 976703d · 976703d
1 parent 75d1661
commit 976703d
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -4,6 +4,9 @@ import (
 	"fmt"
 	"github.com/fatih/color"
 	"github.com/root27/slacknotifyMQ/controllers"
+	"os"
+	"os/signal"
+	"syscall"
 )
 
 var (
@@ -65,6 +68,16 @@ func main() {
 		SlackChannel: slackChannel,
 	}
 
+	stopChan := make(chan os.Signal, 1)
+
+	signal.Notify(stopChan, os.Interrupt, syscall.SIGTERM)
+
+	go func() {
+		<-stopChan
+		fmt.Println(color.RedString("Notifier Stopped"))
+		os.Exit(0)
+	}()
+
 	s.HandleRabbit()
 
 }