Merge pull request #71 from auyer/master

Cleaning lock files on Interruption

1 files changed, 49 insertions, 0 deletions

diff --git a/util/interrupt/cleaner.go b/util/interrupt/cleaner.go
new file mode 100644
index 00000000..76c9d04d
--- /dev/null
+++ b/util/interrupt/cleaner.go
@@ -0,0 +1,49 @@
+package interrupt
+
+import (
+	"fmt"
+	"os"
+	"os/signal"
+	"syscall"
+)
+
+// Cleaner type referes to a function with no inputs that returns an error
+type Cleaner func() error
+
+var cleaners []Cleaner
+var active = false
+
+// RegisterCleaner is responsible for regisreting a cleaner function. When a function is registered, the Signal watcher is started in a goroutine.
+func RegisterCleaner(f ...Cleaner) {
+	for _, fn := range f {
+		cleaners = append([]Cleaner{fn}, cleaners...)
+		if !active {
+			active = true
+			go func() {
+				ch := make(chan os.Signal, 1)
+				signal.Notify(ch, syscall.SIGINT, syscall.SIGTERM, os.Interrupt)
+				<-ch
+				// Prevent un-terminated ^C character in terminal
+				fmt.Println()
+				fmt.Println("Cleaning")
+				errl := Clean()
+				for _, err := range errl {
+					fmt.Println(err)
+				}
+				os.Exit(1)
+			}()
+		}
+	}
+}
+
+// Clean invokes all registered cleanup functions, and returns a list of errors, if they exist.
+func Clean() (errorlist []error) {
+	for _, f := range cleaners {
+		err := f()
+		if err != nil {
+			errorlist = append(errorlist, err)
+		}
+	}
+	cleaners = []Cleaner{}
+	return
+}