From 2c029eba7c8aceee40b4a8dab32fb452fed97d9e Mon Sep 17 00:00:00 2001 From: Keuin Date: Thu, 8 Sep 2022 18:49:01 +0800 Subject: Print shutdown logs properly. Handle multiple signal channels using in one goroutine. --- main.go | 24 +++++++++++++----------- 1 file changed, 13 insertions(+), 11 deletions(-) (limited to 'main.go') diff --git a/main.go b/main.go index bf12042..6b68ee0 100644 --- a/main.go +++ b/main.go @@ -139,10 +139,7 @@ func main() { logger.Printf("Starting tasks...") wg := sync.WaitGroup{} - defer func() { - wg.Wait() - logger.Println("Stopping YABR...") - }() + ctx, cancelTasks := context.WithCancel(context.Background()) for _, task := range tasks { wg.Add(1) @@ -155,18 +152,23 @@ func main() { syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM) - go func() { - <-chSigStop - cancelTasks() - }() chSigQuit := make(chan os.Signal) signal.Notify(chSigQuit, syscall.SIGQUIT) go func() { - <-chSigQuit - os.Exit(0) + select { + case <-chSigStop: + logger.Println("Stopping all tasks...") + cancelTasks() + case <-chSigQuit: + logger.Println("Aborted.") + os.Exit(0) + } }() // block main goroutine on task goroutines - wg.Wait() + defer func() { + wg.Wait() + logger.Println("YABR is stopped.") + }() } -- cgit v1.2.3