graceful shutdown on SIGTERM,SIGQUIT,SIGINT

The-Gleb · Mar 17, 2024 · cc034cc · cc034cc
1 parent 6f46514
commit cc034cc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/cmd/agent/main.go b/cmd/agent/main.go
@@ -19,6 +19,7 @@ import (
 	"runtime"
 	"sync"
 	"sync/atomic"
+	"syscall"
 	"time"
 
 	"github.com/shirou/gopsutil/v3/cpu"
@@ -63,7 +64,7 @@ func main() {
 	var reportInterval = time.Duration(config.ReportInterval * 1000000000)
 
 	c := make(chan os.Signal, 1)
-	signal.Notify(c, os.Interrupt)
+	signal.Notify(c, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
 
 	pollTicker := time.NewTicker(pollInterval)
 	reportTicker := time.NewTicker(reportInterval)

diff --git a/cmd/server/main.go b/cmd/server/main.go
@@ -45,7 +45,7 @@ func main() {
 }
 
 func Run(ctx context.Context) error {
-	ctx, cancel := signal.NotifyContext(ctx, syscall.SIGINT)
+	ctx, cancel := signal.NotifyContext(ctx, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
 	defer cancel()
 
 	go func() {