From addeb9a0a1114136c5bc5773724a19d6be377c9d Mon Sep 17 00:00:00 2001 From: Rafael Passos Date: Tue, 23 Oct 2018 18:00:55 -0300 Subject: Created Cleaner package --- cleaner/cleaner.go | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) create mode 100644 cleaner/cleaner.go 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{} +} -- cgit