diff options
-rw-r--r-- | cleaner/cleaner.go | 38 |
1 files changed, 38 insertions, 0 deletions
diff --git a/cleaner/cleaner.go b/cleaner/cleaner.go new file mode 100644 index 00000000..9bce4c03 --- /dev/null +++ b/cleaner/cleaner.go @@ -0,0 +1,38 @@ +package cleaner + +import ( + "fmt" + "os" + "os/signal" + "syscall" +) + +type t func() error + +var cleaners []t +var inactive bool + +// Register a cleaner function. When a function is registered, the Signal watcher is started in a goroutine. +func Register(f t) { + cleaners = append(cleaners, f) + if !inactive { + inactive = false + go func() { + ch := make(chan os.Signal, 1) + signal.Notify(ch, syscall.SIGINT, syscall.SIGTERM, os.Interrupt) + <-ch + fmt.Println("") + clean() + os.Exit(1) + }() + } +} + +// Clean invokes all registered cleanup functions +func clean() { + fmt.Println("Cleaning") + for _, f := range cleaners { + _ = f() + } + cleaners = []t{} +} |