refactor: rewrite signal with go-signalcontext

Author: CMGS

README.md

@@ -76,7 +76,7 @@ Make sure you can clone code. After the fresh image was named and tagged, it wil
 2. Deploy core itself
 
 ```shell
-<cli_execute_path> container deploy --pod <pod_name> [--node <node_name>] --entry core --network <network_name> --image <projecteru2/core>|<your_own_image> --file <core_config_yaml>:/core.yaml [--count <count_num>] [--cpu 0.3 | --mem 1024000000] http://bit.ly/EruCore
+<cli_execute_path> workloads deploy --pod <pod_name> [--node <node_name>] --entry core --network <network_name> --image <projecteru2/core>|<your_own_image> --file <core_config_yaml>:/core.yaml [--count <count_num>] [--cpu 0.3 | --mem 1024000000] http://bit.ly/EruCore
 ```
 
 Now you will find core was started in nodes.

core.go

@@ -6,8 +6,6 @@ import (
 	"net/http"
 	_ "net/http/pprof" // nolint
 	"os"
-	"os/signal"
-	"syscall"
 
 	"github.com/projecteru2/core/auth"
 	"github.com/projecteru2/core/cluster/calcium"
@@ -17,6 +15,7 @@ import (
 	"github.com/projecteru2/core/utils"
 	"github.com/projecteru2/core/versioninfo"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
+	"github.com/sethvargo/go-signalcontext"
 	log "github.com/sirupsen/logrus"
 	cli "github.com/urfave/cli/v2"
 	"google.golang.org/grpc"
@@ -111,10 +110,11 @@ func serve(c *cli.Context) error {
 	log.Info("[main] Cluster started successfully.")
 
 	// wait for unix signals and try to GracefulStop
-	sigs := make(chan os.Signal, 1)
-	signal.Notify(sigs, syscall.SIGINT, syscall.SIGHUP, syscall.SIGTERM)
-	sig := <-sigs
-	log.Infof("[main] Get signal %v.", sig)
+	ctx, cancel := signalcontext.OnInterrupt()
+	defer cancel()
+	<-ctx.Done()
+
+	log.Info("[main] Interrupt by signal")
 	close(rpcch)
 	unregisterService()
 	grpcServer.GracefulStop()

go.mod

@@ -31,6 +31,7 @@ require (
 	github.com/projecteru2/libyavirt v0.0.0-20200831030711-2bb3f2ba0cb7
 	github.com/prometheus/client_golang v1.7.1
 	github.com/sanity-io/litter v1.3.0
+	github.com/sethvargo/go-signalcontext v0.1.0 
 	github.com/sirupsen/logrus v1.6.0
 	github.com/stretchr/testify v1.6.1
 	github.com/urfave/cli/v2 v2.2.0

go.sum

@@ -250,6 +250,8 @@ github.com/sanity-io/litter v1.3.0 h1:5ZO+weUsqdSWMUng5JnpkW/Oz8iTXiIdeumhQr1sSj
 github.com/sanity-io/litter v1.3.0/go.mod h1:5Z71SvaYy5kcGtyglXOC9rrUi3c1E8CamFWjQsazTh0=
 github.com/sergi/go-diff v1.1.0 h1:we8PVUC3FE2uYfodKH/nBHMSetSfHDR6scGdBi+erh0=
 github.com/sergi/go-diff v1.1.0/go.mod h1:STckp+ISIX8hZLjrqAeVduY0gWCT9IjLuqbuNXdaHfM=
+github.com/sethvargo/go-signalcontext v0.1.0 h1:3IU7HOlmRXF0PSDf85C4nJ/zjYDjF+DS+LufcKfLvyk=
+github.com/sethvargo/go-signalcontext v0.1.0/go.mod h1:PXu9UmR2f7mmp8kEwgkKmaDbxq/PbqixkiC66WIkkWE=
 github.com/shurcooL/sanitized_anchor_name v1.0.0 h1:PdmoCO6wvbs+7yrJyMORt4/BmY5IYyJwS/kOiWx8mHo=
 github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc=
 github.com/sirupsen/logrus v1.0.4-0.20170822132746-89742aefa4b2/go.mod h1:pMByvHTf9Beacp5x1UXfOR9xyW/9antXMhjMPG0dEzc=