safe_exit.go

package main

// Catches the SIGTERM given by ^C and waits until the end of the loop to quit.

import (
	"fmt"
	"os"
	"os/signal"
	"syscall"
	"time"
)

func runTasks(c chan os.Signal) {

	for {
		fmt.Println("starting")
		time.Sleep(time.Second * 3)
		fmt.Println("ending")

		select {
		case <-c:
			fmt.Println("safe exit!")
			os.Exit(2)
		default:
		}
	}
}

func main() {
	c := make(chan os.Signal, 2)
	signal.Notify(c, os.Interrupt, syscall.SIGTERM)

	runTasks(c)
}