stub.go

package main

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

func main() {
	var (
		survivalTime time.Duration
		waitTime     time.Duration
		exitCode     int
	)
	flag.DurationVar(&survivalTime, "survival-time", 0, "survival time")
	flag.DurationVar(&waitTime, "wait-time", 0, "time to wait for a signal to be sent")
	flag.IntVar(&exitCode, "exit-code", 0, "exit code")
	flag.Parse()
	sigchan := make(chan os.Signal)
	signal.Notify(sigchan, syscall.SIGTERM)
	signal.Notify(sigchan, syscall.SIGINT)
	if survivalTime > 0 {
		select {
		case <-time.After(survivalTime):
			break
		case <-sigchan:
			time.Sleep(waitTime)
			break
		}
	} else if waitTime > 0 {
		select {
		case <-sigchan:
			time.Sleep(waitTime)
			break
		}
	} else {
	}
	os.Exit(exitCode)
}