diff options
| -rw-r--r-- | generate.go | 2 | ||||
| -rw-r--r-- | generateHeader.go | 17 |
2 files changed, 18 insertions, 1 deletions
diff --git a/generate.go b/generate.go index c2287b9..f905d9d 100644 --- a/generate.go +++ b/generate.go @@ -16,7 +16,7 @@ func (pb *Files) makeNewSortfile(pf *File) error { wSort, _ := os.OpenFile(pf.Filebase+".sort.pb.go", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644) defer wSort.Close() - header(wSort, pf) + headerSort(wSort, pf) pf.syncLock(wSort) if os.Getenv("PROTOBUF_REGRET") != "true" { pf.addNewFunc(wSort) diff --git a/generateHeader.go b/generateHeader.go index 78fb359..d961646 100644 --- a/generateHeader.go +++ b/generateHeader.go @@ -53,6 +53,23 @@ func header(w io.Writer, pf *File) { fmt.Fprintln(w, "") } +func headerSort(w io.Writer, pf *File) { + // header must come first + headerComment(w) + fmt.Fprintf(w, "package %s\n", pf.Package) + fmt.Fprintln(w, "") + fmt.Fprintln(w, "import (") + fmt.Fprintln(w, " \"fmt\"") + fmt.Fprintln(w, " \"iter\"") + fmt.Fprintln(w, " \"sort\"") + fmt.Fprintln(w, " \"sync\"") + fmt.Fprintln(w, "") + fmt.Fprintln(w, " \"go.wit.com/lib/config\"") + fmt.Fprintln(w, " \"google.golang.org/protobuf/proto\"") + fmt.Fprintln(w, ")") + fmt.Fprintln(w, "") +} + func (pb *File) addNewFunc(w io.Writer) { var STRUCT string = pb.Bases.Name var UUID string = pb.Uuid |