feat: handle SIGTERM and SIGINT gracefully

cartesi · Jan 12, 2024 · 124c318 · 124c318
1 parent 210d0ec
commit 124c318
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cmd/cartesi-rollups-node/main.go b/cmd/cartesi-rollups-node/main.go
@@ -5,6 +5,8 @@ package main
 
 import (
 	"context"
+	"os/signal"
+	"syscall"
 	"time"
 
 	"github.com/cartesi/rollups-node/internal/config"
@@ -15,8 +17,8 @@ func main() {
 	startTime := time.Now()
 	var s []services.Service
 
-	ctx, cancel := context.WithCancel(context.Background())
-	defer cancel()
+	ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
+	defer stop()
 
 	// add Redis first
 	s = append(s, newRedis())