PBM-397: cancel context on SIGINT and SIGKILL

to force agent status cleanup

Author: defbin

cmd/pbm-agent/main.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	stdlog "log"
 	"os"
+	"os/signal"
 	"runtime"
 	"strconv"
 	"strings"
@@ -86,7 +87,7 @@ func runAgent(mongoURI string, dumpConns int) error {
 	mtLog.SetDateFormat(log.LogTimeFormat)
 	mtLog.SetVerbosity(&options.Verbosity{VLevel: mtLog.DebugLow})
 
-	ctx, cancel := context.WithCancel(context.Background())
+	ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, os.Kill)
 	defer cancel()
 
 	leadConn, err := connect.Connect(ctx, mongoURI, "pbm-agent")

pbm/ctrl/recv.go

@@ -54,6 +54,9 @@ func ListenCmd(ctx context.Context, m connect.Client, cl <-chan struct{}) (<-cha
 			)
 			if err != nil {
 				errc <- errors.Wrap(err, "watch the cmd stream")
+				if errors.Is(err, context.Canceled) {
+					return
+				}
 				continue
 			}