better cleanup

CHANGES.md

@@ -13,6 +13,7 @@
 + read `GARGS_PROCESS_BUFFER` to let user set size of data before a tempfile is used.
 + read `GARGS_WAIT_MULTIPLIER` to determine how many finished processes will wait for single slow processes
   higher values improve concurrency at the expense of memory.
++ better cleanup of tmp files in case of process halt.
 
 0.3.4
 =====

process/cleanup.go

@@ -0,0 +1,43 @@
+package process
+
+import (
+	"fmt"
+	"log"
+	"os"
+	"os/signal"
+	"path/filepath"
+	"syscall"
+)
+
+// Cleanup is a best-effort to remove all temporary files
+// created by process. Users can call it manually to remove them.
+func Cleanup() {
+	matches, err := filepath.Glob(filepath.Join(os.TempDir(), fmt.Sprintf("%s.*", prefix)))
+	if err != nil {
+		log.Fatal(err)
+	}
+	for _, f := range matches {
+		os.Remove(f)
+	}
+}
+
+func init() {
+	c := make(chan os.Signal, 1)
+	signal.Notify(c,
+		syscall.SIGINT,
+		syscall.SIGTERM,
+		syscall.SIGHUP,
+		syscall.SIGQUIT)
+	go func() {
+		s := <-c
+		Cleanup()
+		fmt.Fprintln(os.Stderr, s)
+		os.Exit(2)
+	}()
+	go func() {
+		if err := recover(); err != nil {
+			Cleanup()
+			panic(err)
+		}
+	}()
+}