From addeb9a0a1114136c5bc5773724a19d6be377c9d Mon Sep 17 00:00:00 2001
From: Rafael Passos <rafael@rcpassos.me>
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

(limited to 'cleaner')

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