catalyst-node: re-add MistUtilLoad shutdown

livepeer · Nov 18, 2022 · 76cd4a0 · 76cd4a0
1 parent 1859029
commit 76cd4a0
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/cmd/catalyst-node/catalyst-node.go b/cmd/catalyst-node/catalyst-node.go
@@ -12,10 +12,12 @@ import (
 	"net/url"
 	"os"
 	"os/exec"
+	"os/signal"
 	"regexp"
 	"runtime"
 	"strings"
 	"sync"
+	"syscall"
 	"time"
 
 	serfclient "github.com/hashicorp/serf/client"
@@ -375,6 +377,16 @@ func main() {
 		}
 	}()
 
+	go func() {
+		c := make(chan os.Signal, 1)
+		signal.Notify(c, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT)
+		for {
+			s := <-c
+			glog.Errorf("caught signal=%v killing MistUtilLoad", s)
+			killchan <- true
+		}
+	}()
+
 	// Everything past this is booting up Serf
 	tmpFile, err := writeSerfConfig(&serfConfig)
 	if err != nil {