morev0.0.0

1 files changed, 7 insertions, 3 deletions

diff --git a/main.go b/main.go
index e791628..fb2ea38 100644
--- a/main.go
+++ b/main.go
@@ -8,11 +8,12 @@ import (
 
 func main() {
 	f, _ := os.OpenFile("test.sort.pb.go", os.O_WRONLY|os.O_CREATE, 0600)
-	header(f, "GitRefs")
+	header(f, "autogenpb")
+	syncLock(f, "godeplock")
 }
 
 func header(w io.Writer, name string) {
-	fmt.Fprintln(w, "package gitpb")
+	fmt.Fprintln(w, "package " + name)
 	fmt.Fprintln(w, "")
 	fmt.Fprintln(w, "// this is becoming a standard format")
 	fmt.Fprintln(w, "// todo: autogenerate this from the .proto file?")
@@ -24,7 +25,10 @@ func header(w io.Writer, name string) {
 	fmt.Fprintln(w, "	\"sync\"")
 	fmt.Fprintln(w, ")")
 	fmt.Fprintln(w, "")
+}
+
+func syncLock(w io.Writer, name string) {
 	fmt.Fprintln(w, "// bad global lock until I figure out some other plan")
-	fmt.Fprintln(w, "var godeplock sync.RWMutex")
+	fmt.Fprintln(w, "var " + name + " sync.RWMutex")
 	fmt.Fprintln(w, "")
 }