From 6ea315ab112b018092573d27d91d8ba785beba4e Mon Sep 17 00:00:00 2001
From: Jordan Orelli <jorelli@etsy.com>
Date: Fri, 4 Jul 2014 15:46:36 +0000
Subject: [PATCH] added pidfile, hup handling, truncate option

---
 randomizr.go | 56 ++++++++++++++++++++++++++++++++++++++--------------
 1 file changed, 41 insertions(+), 15 deletions(-)

diff --git a/randomizr.go b/randomizr.go
index 0ff24b9..bac22cc 100644
--- a/randomizr.go
+++ b/randomizr.go
@@ -5,14 +5,17 @@ import (
 	"fmt"
 	"math/rand"
 	"os"
+	"os/signal"
+	"syscall"
 	"time"
 )
 
 var (
-	fname   string
-	fappend bool
-	reopen  bool
-	freq    float64
+	fname     string
+	ftruncate bool
+	reopen    bool
+	freq      float64
+	pidfile   string
 )
 
 // generates a pseudorandom string of length n that is composed of alphanumeric
@@ -31,11 +34,11 @@ func outFile() (*os.File, error) {
 		return os.Stdout, nil
 	}
 	options := os.O_WRONLY | os.O_CREATE
-    if fappend {
-        options |= os.O_APPEND
-    } else {
-        options |= os.O_TRUNC
-    }
+	if ftruncate {
+		options |= os.O_TRUNC
+	} else {
+		options |= os.O_APPEND
+	}
 
 	return os.OpenFile(fname, options, 0644)
 }
@@ -55,16 +58,25 @@ func reopenWrite(c chan string) {
 }
 
 func regularWrite(c chan string) {
+	hup := make(chan os.Signal, 1)
+	signal.Notify(hup, syscall.SIGHUP, syscall.SIGUSR1)
+
+START:
 	f, err := outFile()
 	if err != nil {
 		fmt.Printf("ERROR: unable to open outfile: %v", err)
 		return
 	}
-	defer f.Close()
 
-	for line := range c {
-		if _, err := f.WriteString(line); err != nil {
-			fmt.Printf("ERROR: unable to write line: %v", err)
+	for {
+		select {
+		case line := <-c:
+			if _, err := f.WriteString(line); err != nil {
+				fmt.Printf("ERROR: unable to write line: %v", err)
+			}
+		case <-hup:
+			f.Close()
+			goto START
 		}
 	}
 }
@@ -77,8 +89,21 @@ func writeLines(c chan string) {
 	}
 }
 
+func writePid() {
+	if pidfile == "" {
+		return
+	}
+	f, err := os.OpenFile(pidfile, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "ERROR unable to open pidfile: %v", err)
+		return
+	}
+	fmt.Fprintf(f, "%d\n", os.Getpid())
+}
+
 func main() {
 	flag.Parse()
+	writePid()
 	c := make(chan string)
 
 	writeLines(c)
@@ -90,8 +115,9 @@ func main() {
 
 func init() {
 	flag.StringVar(&fname, "file", "", "destination file to which random data will be written")
-	flag.BoolVar(&fappend, "append", false, "append to file instead of truncating file on open")
+	flag.StringVar(&pidfile, "pidfile", "", "file to which a pid is written")
+	flag.BoolVar(&ftruncate, "truncate", false, "truncate file on opening instead of appending")
 	flag.BoolVar(&reopen, "reopen", false, "reopen file handle on every write instead of using a persistent handle")
-	flag.Float64Var(&freq, "freq", 1, "frequency in hz at which lines will be written")
+	flag.Float64Var(&freq, "freq", 10, "frequency in hz at which lines will be written")
 	rand.Seed(time.Now().UnixNano())
 }