handle some signals

Author: gulducat

hashicorp.hcl

@@ -4,9 +4,9 @@ program "consul" {
 }
 
 program "nomad" {
-	command = "nomad agent -dev"
-	check   = "nomad node status"
-	# seconds = 1 # to test check failures
+  command = "nomad agent -dev"
+  check   = "nomad node status"
+  # seconds = 3 # to test check failures
 }
 
 program "vault" {

main.go

@@ -2,8 +2,11 @@ package main
 
 import (
 	"context"
+	"fmt"
 	"log"
 	"os"
+	"os/signal"
+	"syscall"
 
 	"github.com/gulducat/go-run-programs/program"
 )
@@ -14,17 +17,22 @@ func main() {
 
 func CLI(args []string) int {
 	if len(args) < 2 {
-		log.Println("gotta provide an hcl file ok?")
+		fmt.Println("gotta provide an hcl file ok?")
 		return 1
 	}
 
+	sig := make(chan os.Signal, 1)
+	signal.Notify(sig, os.Interrupt, syscall.SIGTERM)
+
 	stop, err := program.RunFromHCL(context.Background(), args[1])
 	defer stop()
+
 	if err != nil {
-		log.Println("error:", err)
+		fmt.Println("error:", err)
 		return 1
 	}
 
-	// let the good times roll
-	select {}
+	s := <-sig
+	log.Println("signal:", s)
+	return 0
 }