Merge pull request #1508 from rfyiamcool/main

fix: add sigterm signal
spidernet-io · Oct 31, 2024 · ffff9c9 · ffff9c9
2 parents 81907e9 + 9965b42
commit ffff9c9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/cmd/agent/cmd/root.go b/cmd/agent/cmd/root.go
@@ -6,12 +6,14 @@ package cmd
 import (
 	"context"
 	"fmt"
-	"github.com/spf13/cobra"
-	"github.com/spidernet-io/egressgateway/pkg/agent"
-	"github.com/spidernet-io/egressgateway/pkg/config"
 	"os"
 	"os/signal"
 	"path/filepath"
+	"syscall"
+
+	"github.com/spf13/cobra"
+	"github.com/spidernet-io/egressgateway/pkg/agent"
+	"github.com/spidernet-io/egressgateway/pkg/config"
 )
 
 var binName = filepath.Base(os.Args[0])
@@ -21,7 +23,7 @@ var rootCmd = &cobra.Command{
 	Use:   binName,
 	Short: "run egress gateway agent",
 	Run: func(cmd *cobra.Command, args []string) {
-		ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt)
+		ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
 		defer stop()
 
 		cfg, err := config.LoadConfig(true)

diff --git a/cmd/controller/cmd/root.go b/cmd/controller/cmd/root.go
@@ -9,6 +9,7 @@ import (
 	"os"
 	"os/signal"
 	"path/filepath"
+	"syscall"
 
 	"github.com/spf13/cobra"
 
@@ -23,7 +24,7 @@ var rootCmd = &cobra.Command{
 	Use:   binName,
 	Short: "run egress gateway controller",
 	Run: func(cmd *cobra.Command, args []string) {
-		ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt)
+		ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
 		defer stop()
 
 		cfg, err := config.LoadConfig(false)