aboutsummaryrefslogtreecommitdiffstats
path: root/cleaner/cleaner.go
diff options
context:
space:
mode:
authorRafael Passos <rafael@rcpassos.me>2018-10-23 18:00:55 -0300
committerRafael Passos <rafael@rcpassos.me>2018-10-23 18:00:55 -0300
commitaddeb9a0a1114136c5bc5773724a19d6be377c9d (patch)
tree57b32b87a056e9cde940334894a8164b61f46b6d /cleaner/cleaner.go
parente79476a9627226de3dee80e9a19b7f66c4ef68ff (diff)
downloadgit-bug-addeb9a0a1114136c5bc5773724a19d6be377c9d.tar.gz
Created Cleaner package
Diffstat (limited to 'cleaner/cleaner.go')
-rw-r--r--cleaner/cleaner.go38
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{}
+}