move autogen sort.pb.go into seperate file
Signed-off-by: Jeff Carr <jcarr@wit.com>
This commit is contained in:
parent
678cb67c9b
commit
a6035cb90f
|
|
@ -0,0 +1,46 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
// eh, screw it, just add the damned mutex to the pb.go file
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"go.wit.com/log"
|
||||||
|
)
|
||||||
|
|
||||||
|
func addMutex(names map[string]string) error {
|
||||||
|
fullname := names["protobase"] + ".pb.go"
|
||||||
|
log.Info("fullname:", fullname)
|
||||||
|
data, err := os.ReadFile(fullname)
|
||||||
|
if err != nil {
|
||||||
|
// log.Info("open config file :", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
w, _ := os.OpenFile(names["protobase"]+".pb.go", os.O_WRONLY|os.O_CREATE, 0600)
|
||||||
|
|
||||||
|
var found bool
|
||||||
|
|
||||||
|
lines := strings.Split(string(data), "\n")
|
||||||
|
for _, line := range lines {
|
||||||
|
// log.Info("line:", line)
|
||||||
|
start := "type " + names["Bases"] + " struct {"
|
||||||
|
if strings.HasSuffix(line, start) {
|
||||||
|
found = true
|
||||||
|
log.Info("FOUND line:", line)
|
||||||
|
fmt.Fprintln(w, line)
|
||||||
|
fmt.Fprintln(w, "\tsync.RWMutex // auto-added by go.wit.com/apps/autogenpb")
|
||||||
|
fmt.Fprintln(w, "")
|
||||||
|
} else {
|
||||||
|
fmt.Fprintln(w, line)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// os.Exit(-1)
|
||||||
|
if found {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return errors.New("addMutex() parse didn't work")
|
||||||
|
}
|
||||||
207
main.go
207
main.go
|
|
@ -1,8 +1,6 @@
|
||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"os"
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
|
@ -63,7 +61,7 @@ func main() {
|
||||||
} else {
|
} else {
|
||||||
sortmap["base"] = argv.LoBase
|
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 == "" {
|
if argv.UpBase == "" {
|
||||||
sortmap["Base"] = cases.Title(language.English, cases.NoLower).String(protobase)
|
sortmap["Base"] = cases.Title(language.English, cases.NoLower).String(protobase)
|
||||||
sortmap["Bases"] = sortmap["Base"] + "s"
|
sortmap["Bases"] = sortmap["Base"] + "s"
|
||||||
|
|
@ -79,6 +77,13 @@ func main() {
|
||||||
os.Exit(0)
|
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)
|
f, _ := os.OpenFile(protobase+".sort.pb.go", os.O_WRONLY|os.O_CREATE, 0600)
|
||||||
|
|
||||||
header(f, sortmap)
|
header(f, sortmap)
|
||||||
|
|
@ -126,201 +131,5 @@ func main() {
|
||||||
// make the foo.marshal.pb.go file
|
// make the foo.marshal.pb.go file
|
||||||
marshal(sortmap)
|
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, "")
|
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,203 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
)
|
||||||
|
|
||||||
|
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, "")
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue