diff options
Diffstat (limited to 'main.go')
| -rw-r--r-- | main.go | 207 |
1 files changed, 8 insertions, 199 deletions
@@ -1,8 +1,6 @@ package main import ( - "fmt" - "io" "os" "strings" @@ -63,7 +61,7 @@ func main() { } else { sortmap["base"] = argv.LoBase } - sortmap["lock"] = sortmap["base"] + "sLock" // is nonglobal and plural + sortmap["lock"] = sortmap["base"] + "sMu" // is nonglobal and plural if argv.UpBase == "" { sortmap["Base"] = cases.Title(language.English, cases.NoLower).String(protobase) sortmap["Bases"] = sortmap["Base"] + "s" @@ -79,6 +77,13 @@ func main() { os.Exit(0) } + // add mutex + if err := addMutex(sortmap); err == nil { + log.Info("adding mutex to existing protoc-gen-go file worked") + } else { + log.Info("adding mutex to existing protoc-gen-go file did not work") + } + f, _ := os.OpenFile(protobase+".sort.pb.go", os.O_WRONLY|os.O_CREATE, 0600) header(f, sortmap) @@ -126,201 +131,5 @@ func main() { // make the foo.marshal.pb.go file marshal(sortmap) } -} - -func headerComment(w io.Writer) { - fmt.Fprintln(w, "") - fmt.Fprintln(w, "// this file was autogenerated with autogenpb") - fmt.Fprintln(w, "//") - fmt.Fprintln(w, "// you might be able to use it on simple, strictly defined protobuf files") - fmt.Fprintln(w, "//") - fmt.Fprintln(w, "// go install go.wit.com/apps/autogenpb@latest") - fmt.Fprintln(w, "") -} - -func header(w io.Writer, names map[string]string) { - fmt.Fprintln(w, "package "+names["package"]) - headerComment(w) - fmt.Fprintln(w, "import (") - fmt.Fprintln(w, " \"fmt\"") - fmt.Fprintln(w, " \"os\"") - fmt.Fprintln(w, " \"sort\"") - fmt.Fprintln(w, " \"sync\"") - fmt.Fprintln(w, ")") - fmt.Fprintln(w, "") -} - -func syncLock(w io.Writer, names map[string]string) { - fmt.Fprintln(w, "// bad global lock until I figure out some other plan") - fmt.Fprintln(w, "// redo/fix protoc-gen-go 1.35 and do it there?") - fmt.Fprintln(w, "// sync.RWMutex or sync.Mutex?") - fmt.Fprintln(w, "var "+names["lock"]+" sync.RWMutex") - fmt.Fprintln(w, "") -} - -func iterTop(w io.Writer, names map[string]string) { - fmt.Fprintln(w, "type "+names["Base"]+"Iterator struct {") - fmt.Fprintln(w, " sync.RWMutex") - fmt.Fprintln(w, "") - fmt.Fprintln(w, " packs []*"+names["Base"]) - fmt.Fprintln(w, " index int") - fmt.Fprintln(w, "}") - fmt.Fprintln(w, "") - fmt.Fprintln(w, "// New"+names["Base"]+"Iterator initializes a new iterator.") - fmt.Fprintln(w, "func New"+names["Base"]+"Iterator(packs []*"+names["Base"]+") *"+names["Base"]+"Iterator {") - fmt.Fprintln(w, " return &"+names["Base"]+"Iterator{packs: packs}") - fmt.Fprintln(w, "}") - fmt.Fprintln(w, "") - fmt.Fprintln(w, "// Scan moves to the next element and returns false if there are no more packs.") - fmt.Fprintln(w, "// Use Scan() in a loop, similar to a while loop") - fmt.Fprintln(w, "//") - fmt.Fprintln(w, "// for iterator.Scan() ") - fmt.Fprintln(w, "// d := iterator.Next(") - fmt.Fprintln(w, "// fmt.Println(\"found UUID:\", d.Uuid") - fmt.Fprintln(w, "// }") - fmt.Fprintln(w, "func (it *"+names["Base"]+"Iterator) Scan() bool {") - fmt.Fprintln(w, " if it.index >= len(it.packs) {") - fmt.Fprintln(w, " return false") - fmt.Fprintln(w, " }") - fmt.Fprintln(w, " it.index++") - fmt.Fprintln(w, " return true") - fmt.Fprintln(w, "}") - fmt.Fprintln(w, "") -} - -func iterNext(w io.Writer, names map[string]string) { - fmt.Fprintln(w, "// Next() returns the next thing in the array") - fmt.Fprintln(w, "func (it *"+names["Base"]+"Iterator) Next() *"+names["Base"]+" {") - fmt.Fprintln(w, " if it.packs[it.index-1] == nil {") - fmt.Fprintln(w, " for i, d := range it.packs {") - fmt.Fprintln(w, " fmt.Println(\"i =\", i, d)") - fmt.Fprintln(w, " }") - fmt.Fprintln(w, " fmt.Println(\"protobuf autogenpb sort error len =\", len(it.packs))") - fmt.Fprintln(w, " fmt.Println(\"protobuf autogenpb sort error next == nil\", it.index, it.index-1)") - fmt.Fprintln(w, " os.Exit(-1)") - fmt.Fprintln(w, " }") - fmt.Fprintln(w, " return it.packs[it.index-1]") - fmt.Fprintln(w, "}") - fmt.Fprintln(w, "") -} - -func iterSort(w io.Writer, names map[string]string) { - fmt.Fprintln(w, "func (all *"+names["Bases"]+") All() *"+names["Base"]+"Iterator {") - fmt.Fprintln(w, " "+names["base"]+"Pointers := all.selectAll"+names["Base"]+"()") - fmt.Fprintln(w, "") - fmt.Fprintln(w, " iterator := New"+names["Base"]+"Iterator("+names["base"]+"Pointers)") - fmt.Fprintln(w, " return iterator") - fmt.Fprintln(w, "}") - fmt.Fprintln(w, "") - fmt.Fprintln(w, "func (all *"+names["Bases"]+") Sort"+names["sortBy"]+"() *"+names["Base"]+"Iterator {") - fmt.Fprintln(w, " packs := all.selectAll"+names["Base"]+"()") - fmt.Fprintln(w, "") - fmt.Fprintln(w, " sort.Sort("+names["Base"]+""+names["sortBy"]+"(packs))") - fmt.Fprintln(w, "") - fmt.Fprintln(w, " iterator := New"+names["Base"]+"Iterator(packs)") - fmt.Fprintln(w, " return iterator") - fmt.Fprintln(w, "}") - fmt.Fprintln(w, "") - fmt.Fprintln(w, "func (all *"+names["Bases"]+") Len() int {") - fmt.Fprintln(w, " "+names["lock"]+".RLock()") - fmt.Fprintln(w, " defer "+names["lock"]+".RUnlock()") - fmt.Fprintln(w, "") - fmt.Fprintln(w, " return len(all."+names["Bases"]+")") - fmt.Fprintln(w, "}") - fmt.Fprintln(w, "") -} - -func iterEnd(w io.Writer, names map[string]string) { - fmt.Fprintln(w, "type "+names["Base"]+""+names["sortBy"]+" []*"+names["Base"]+"") - fmt.Fprintln(w, "") - fmt.Fprintln(w, "func (a "+names["Base"]+""+names["sortBy"]+") Len() int { return len(a) }") - fmt.Fprintln(w, "func (a "+names["Base"]+""+names["sortBy"]+") Less(i, j int) bool { return a[i]."+names["sortKey"]+" < a[j]."+names["sortKey"]+" }") - fmt.Fprintln(w, "func (a "+names["Base"]+""+names["sortBy"]+") Swap(i, j int) { a[i], a[j] = a[j], a[i] }") - fmt.Fprintln(w, "") - fmt.Fprintln(w, "// safely returns a slice of pointers to the "+names["Base"]+" protobufs") - fmt.Fprintln(w, "func (all *"+names["Bases"]+") selectAll"+names["Base"]+"() []*"+names["Base"]+" {") - fmt.Fprintln(w, " "+names["lock"]+".RLock()") - fmt.Fprintln(w, " defer "+names["lock"]+".RUnlock()") - fmt.Fprintln(w, "") - fmt.Fprintln(w, " // Create a new slice to hold pointers to each "+names["Base"]+"") - fmt.Fprintln(w, " var aStuff []*"+names["Base"]+"") - fmt.Fprintln(w, " aStuff = make([]*"+names["Base"]+", len(all."+names["Bases"]+"))") - fmt.Fprintln(w, " for i, p := range all."+names["Bases"]+" {") - fmt.Fprintln(w, " aStuff[i] = p // Copy pointers for safe iteration") - fmt.Fprintln(w, " }") - fmt.Fprintln(w, "") - fmt.Fprintln(w, " return aStuff") - fmt.Fprintln(w, "}") - fmt.Fprintln(w, "") -} - -func iterAppend(w io.Writer, names map[string]string) { - if names["append"] == "" { - fmt.Fprintln(w, "// does not enforce any unique fields") - } else { - fmt.Fprintln(w, "// enforces "+names["append"]+" is unique") - } - if names["append"] == "" { - fmt.Fprintln(w, "func (all *"+names["Bases"]+") Append(newP *"+names["Base"]+") bool {") - } else { - // fmt.Fprintln(w, "func (all *"+names["Bases"]+") Append(newP *"+names["Base"]+") bool { // todo: make unique name here") - fmt.Fprintln(w, "func (all *"+names["Bases"]+") AppendUnique"+names["append"]+"(newP *"+names["Base"]+") bool {") - } - fmt.Fprintln(w, " "+names["lock"]+".Lock()") - fmt.Fprintln(w, " defer "+names["lock"]+".Unlock()") - fmt.Fprintln(w, "") - if names["append"] != "" { - fmt.Fprintln(w, " for _, p := range all."+names["Bases"]+" {") - fmt.Fprintln(w, " if p."+names["append"]+" == newP."+names["append"]+" {") - fmt.Fprintln(w, " return false") - fmt.Fprintln(w, " }") - fmt.Fprintln(w, " }") - fmt.Fprintln(w, "") - } - fmt.Fprintln(w, " all."+names["Bases"]+" = append(all."+names["Bases"]+", newP)") - fmt.Fprintln(w, " return true") - fmt.Fprintln(w, "}") - fmt.Fprintln(w, "") -} - -func iterReplace(w io.Writer, names map[string]string) { - if names["append"] == "" { - // can't continue without a key field - } - fmt.Fprintln(w, "// enforces "+names["append"]+" is unique") - fmt.Fprintln(w, "func (all *"+names["Bases"]+") Replace"+names["append"]+"(newP *"+names["Base"]+") bool { // todo: make unique name here") - - fmt.Fprintln(w, " "+names["lock"]+".Lock()") - fmt.Fprintln(w, " defer "+names["lock"]+".Unlock()") - fmt.Fprintln(w, "") - fmt.Fprintln(w, " for _, p := range all."+names["Bases"]+" {") - fmt.Fprintln(w, " if p."+names["append"]+" == newP."+names["append"]+" {") - fmt.Fprintln(w, " return false") - fmt.Fprintln(w, " }") - fmt.Fprintln(w, " }") - fmt.Fprintln(w, "") - fmt.Fprintln(w, " all."+names["Bases"]+" = append(all."+names["Bases"]+", newP)") - fmt.Fprintln(w, " return true") - fmt.Fprintln(w, "}") - fmt.Fprintln(w, "") -} -func iterDelete(w io.Writer, names map[string]string) { - fmt.Fprintln(w, "func (all *"+names["Bases"]+") DeleteBy"+names["append"]+"(s string) *"+names["Base"]+" {") - fmt.Fprintln(w, " "+names["lock"]+".Lock()") - fmt.Fprintln(w, " defer "+names["lock"]+".Unlock()") - fmt.Fprintln(w, "") - fmt.Fprintln(w, " var newr "+names["Base"]) - fmt.Fprintln(w, "") - fmt.Fprintln(w, " for i, _ := range all."+names["Bases"]+" {") - fmt.Fprintln(w, " if all."+names["Bases"]+"[i]."+names["append"]+" == s {") - fmt.Fprintln(w, " newr = *all."+names["Bases"]+"[i]") - fmt.Fprintln(w, " all."+names["Bases"]+"[i] = all."+names["Bases"]+"[len(all."+names["Bases"]+")-1]") - fmt.Fprintln(w, " all."+names["Bases"]+" = all."+names["Bases"]+"[:len(all."+names["Bases"]+")-1]") - fmt.Fprintln(w, " return &newr") - fmt.Fprintln(w, " }") - fmt.Fprintln(w, " }") - fmt.Fprintln(w, " return nil") - fmt.Fprintln(w, "}") - fmt.Fprintln(w, "") } |