catch signals for shutdown

1 files changed, 22 insertions, 2 deletions

diff --git a/cmd/gorum/main.go b/cmd/gorum/main.go
index 223da9d..e39f620 100644
--- a/cmd/gorum/main.go
+++ b/cmd/gorum/main.go
@@ -4,6 +4,10 @@ import (
 	"context"
 	"flag"
 	"fmt"
+	"log"
+	"os"
+	"os/signal"
+	"syscall"
 
 	"codeberg.org/snonux/gorum/internal"
 )
@@ -21,8 +25,24 @@ func main() {
 		return
 	}
 
-	ctx, cancel := context.WithCancel(context.Background())
+	ctx, cancel := contextWithSignal()
 	defer cancel()
 
-	internal.Run(ctx, *configFile)
+	internal.Start(ctx, *configFile)
+
+	<-ctx.Done()
+	log.Println("Good bye!")
+}
+
+func contextWithSignal() (context.Context, context.CancelFunc) {
+	ctx, cancel := context.WithCancel(context.Background())
+
+	go func() {
+		ch := make(chan os.Signal)
+		signal.Notify(ch, os.Interrupt, syscall.SIGTERM, syscall.SIGINT)
+		<-ch
+		cancel()
+	}()
+
+	return ctx, cancel
 }