285 lines
9.5 KiB
Go
285 lines
9.5 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
)
|
|
|
|
func (msg *MsgName) iterTop(w io.Writer) {
|
|
var BASE string = msg.Name
|
|
|
|
fmt.Fprintln(w, "type "+BASE+"Iterator struct {")
|
|
fmt.Fprintln(w, " sync.RWMutex")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " things []*"+BASE)
|
|
fmt.Fprintln(w, " index int")
|
|
fmt.Fprintln(w, "}")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, "// New"+BASE+"Iterator initializes a new iterator.")
|
|
fmt.Fprintln(w, "func New"+BASE+"Iterator(things []*"+BASE+") *"+BASE+"Iterator {")
|
|
fmt.Fprintln(w, " return &"+BASE+"Iterator{things: things}")
|
|
fmt.Fprintln(w, "}")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, "// Scan moves to the next element and returns false if there are no more things.")
|
|
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 *"+BASE+"Iterator) Scan() bool {")
|
|
fmt.Fprintln(w, " if it.index >= len(it.things) {")
|
|
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 (msg *MsgName) iterNext(w io.Writer) {
|
|
var BASE string = msg.Name
|
|
|
|
fmt.Fprintln(w, "// Next() returns the next thing in the array")
|
|
fmt.Fprintln(w, "func (it *"+BASE+"Iterator) Next() *"+BASE+" {")
|
|
fmt.Fprintln(w, " if it.things[it.index-1] == nil {")
|
|
fmt.Fprintln(w, " for i, d := range it.things {")
|
|
fmt.Fprintln(w, " fmt.Println(\"i =\", i, d)")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, " return it.things[it.index-1]")
|
|
fmt.Fprintln(w, "}")
|
|
fmt.Fprintln(w, "")
|
|
}
|
|
|
|
func (pf *File) selectAllFunc(w io.Writer) {
|
|
var BASES string = pf.Bases.Name
|
|
var BASE string = pf.Base.Name
|
|
var LOCK string = pf.Bases.Lockname
|
|
|
|
fmt.Fprintln(w, "func (x *"+BASES+") All() *"+BASE+"Iterator {")
|
|
fmt.Fprintln(w, " "+BASE+"Pointers := x.selectAll"+BASE+"()")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " iterator := New"+BASE+"Iterator("+BASE+"Pointers)")
|
|
fmt.Fprintln(w, " return iterator")
|
|
fmt.Fprintln(w, "}")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, "func (x *"+BASES+") Len() int {")
|
|
fmt.Fprintln(w, " "+LOCK+".RLock()")
|
|
fmt.Fprintln(w, " defer "+LOCK+".RUnlock()")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " return len(x."+BASES+")")
|
|
fmt.Fprintln(w, "}")
|
|
fmt.Fprintln(w, "")
|
|
}
|
|
|
|
func (pf *File) sortByFunc(w io.Writer) {
|
|
var BASES string = pf.Bases.Name
|
|
var BASE string = pf.Base.Name
|
|
|
|
for _, SORT := range pf.Base.Sort {
|
|
fmt.Fprintln(w, "func (x *"+BASES+") SortBy"+SORT+"() *"+BASE+"Iterator {")
|
|
fmt.Fprintln(w, " things := x.selectAll"+BASE+"()")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " sort.Sort("+BASE+""+SORT+"(things))")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " iterator := New"+BASE+"Iterator(things)")
|
|
fmt.Fprintln(w, " return iterator")
|
|
fmt.Fprintln(w, "}")
|
|
fmt.Fprintln(w, "")
|
|
|
|
fmt.Fprintln(w, "type "+BASE+""+SORT+" []*"+BASE+"")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, "func (a "+BASE+""+SORT+") Len() int { return len(a) }")
|
|
fmt.Fprintln(w, "func (a "+BASE+""+SORT+") Less(i, j int) bool { return a[i]."+SORT+" < a[j]."+SORT+" }")
|
|
fmt.Fprintln(w, "func (a "+BASE+""+SORT+") Swap(i, j int) { a[i], a[j] = a[j], a[i] }")
|
|
fmt.Fprintln(w, "")
|
|
}
|
|
}
|
|
|
|
func (pf *File) iterSelect(w io.Writer) {
|
|
var BASES string = pf.Bases.Name
|
|
var BASE string = pf.Base.Name
|
|
var LOCK string = pf.Bases.Lockname
|
|
|
|
fmt.Fprintln(w, "// safely returns a slice of pointers to the "+BASE+" protobufs")
|
|
fmt.Fprintln(w, "func (x *"+BASES+") selectAll"+BASE+"() []*"+BASE+" {")
|
|
fmt.Fprintln(w, " "+LOCK+".RLock()")
|
|
fmt.Fprintln(w, " defer "+LOCK+".RUnlock()")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " // Create a new slice to hold pointers to each "+BASE+"")
|
|
fmt.Fprintln(w, " var tmp []*"+BASE+"")
|
|
fmt.Fprintln(w, " tmp = make([]*"+BASE+", len(x."+BASES+"))")
|
|
fmt.Fprintln(w, " for i, p := range x."+BASES+" {")
|
|
fmt.Fprintln(w, " tmp[i] = p // Copy pointers for safe iteration")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " return tmp")
|
|
fmt.Fprintln(w, "}")
|
|
}
|
|
|
|
func (pf *File) appendUniqueOld(w io.Writer) {
|
|
var MSG string = pf.Bases.Name
|
|
var BASE string = pf.Base.Name
|
|
var LOCK string = pf.Bases.Lockname
|
|
|
|
if argv.Mutex {
|
|
// use the mutex lock from the modified protoc.pb.go file
|
|
LOCK = "all.Lock"
|
|
} else {
|
|
LOCK = pf.Bases.Lockname
|
|
}
|
|
|
|
// append -- no check at all
|
|
fmt.Fprintln(w, "// just a simple Append() with no checking (but still uses the mutex lock)")
|
|
fmt.Fprintln(w, "func (all *"+MSG+") Append(newP *"+BASE+") bool {")
|
|
fmt.Fprintln(w, " "+LOCK+".RLock()")
|
|
fmt.Fprintln(w, " defer "+LOCK+".RUnlock()")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " all."+MSG+" = append(all."+MSG+", newP)")
|
|
fmt.Fprintln(w, " return true")
|
|
fmt.Fprintln(w, "}")
|
|
fmt.Fprintln(w, "")
|
|
|
|
// append for single keys
|
|
for _, KEY := range pf.Base.Unique {
|
|
fmt.Fprintln(w, "// enforces "+BASE+" is unique")
|
|
fmt.Fprintln(w, "func (all *"+MSG+") AppendUnique"+KEY+"(newP *"+BASE+") bool {")
|
|
fmt.Fprintln(w, " "+LOCK+".RLock()")
|
|
fmt.Fprintln(w, " defer "+LOCK+".RUnlock()")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " for _, p := range all."+MSG+" {")
|
|
fmt.Fprintln(w, " if p."+KEY+" == newP."+KEY+" {")
|
|
fmt.Fprintln(w, " return false")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " all."+MSG+" = append(all."+MSG+", newP)")
|
|
fmt.Fprintln(w, " return true")
|
|
fmt.Fprintln(w, "}")
|
|
fmt.Fprintln(w, "")
|
|
}
|
|
|
|
// append check for every key
|
|
if len(pf.Base.Unique) == 0 {
|
|
// there are no keys defined
|
|
return
|
|
}
|
|
fmt.Fprintln(w, "// enforces "+BASE+" is unique")
|
|
fmt.Fprintln(w, "func (all *"+MSG+") AppendUnique(newP *"+BASE+") bool {")
|
|
fmt.Fprintln(w, " "+LOCK+".RLock()")
|
|
fmt.Fprintln(w, " defer "+LOCK+".RUnlock()")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " for _, p := range all."+MSG+" {")
|
|
for _, KEY := range pf.Base.Unique {
|
|
fmt.Fprintln(w, " if p."+KEY+" == newP."+KEY+" {")
|
|
fmt.Fprintln(w, " return false")
|
|
fmt.Fprintln(w, " }")
|
|
}
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " all."+MSG+" = append(all."+MSG+", newP)")
|
|
fmt.Fprintln(w, " return true")
|
|
fmt.Fprintln(w, "}")
|
|
fmt.Fprintln(w, "")
|
|
}
|
|
|
|
func (pf *File) replaceFunc(w io.Writer) {
|
|
var MSG string = pf.Bases.Name
|
|
var BASE string = pf.Base.Name
|
|
var LOCK string = pf.Bases.Lockname
|
|
|
|
for _, KEY := range pf.Base.Unique {
|
|
fmt.Fprintln(w, "// enforces "+KEY+" is unique")
|
|
fmt.Fprintln(w, "func (all *"+MSG+") Replace"+KEY+"(newP *"+BASE+") bool { // todo: make unique name here")
|
|
fmt.Fprintln(w, " "+LOCK+".RLock()")
|
|
fmt.Fprintln(w, " defer "+LOCK+".RUnlock()")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " for _, p := range all."+MSG+" {")
|
|
fmt.Fprintln(w, " if p."+KEY+" == newP."+KEY+" {")
|
|
fmt.Fprintln(w, " return false")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " all."+MSG+" = append(all."+MSG+", newP)")
|
|
fmt.Fprintln(w, " return true")
|
|
fmt.Fprintln(w, "}")
|
|
fmt.Fprintln(w, "")
|
|
}
|
|
}
|
|
|
|
func (pf *File) deleteFunc(w io.Writer) {
|
|
var MSG string = pf.Bases.Name
|
|
var LOCK string = pf.Bases.Lockname
|
|
|
|
for _, KEY := range pf.Base.Unique {
|
|
fmt.Fprintln(w, "func (all *"+MSG+") DeleteBy"+KEY+"(s string) bool {")
|
|
fmt.Fprintln(w, " "+LOCK+".RLock()")
|
|
fmt.Fprintln(w, " defer "+LOCK+".RUnlock()")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " for i, _ := range all."+MSG+" {")
|
|
fmt.Fprintln(w, " if all."+MSG+"[i]."+KEY+" == s {")
|
|
fmt.Fprintln(w, " all."+MSG+"[i] = all."+MSG+"[len(all."+MSG+")-1]")
|
|
fmt.Fprintln(w, " all."+MSG+" = all."+MSG+"[:len(all."+MSG+")-1]")
|
|
fmt.Fprintln(w, " return true")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, " return false")
|
|
fmt.Fprintln(w, "}")
|
|
fmt.Fprintln(w, "")
|
|
}
|
|
}
|
|
|
|
// this tries to return the deleted one but is wrong/gives warning if mutex lock is in struct
|
|
func (pf *File) deleteWithCopyFunc(w io.Writer) {
|
|
var MSG string = pf.Bases.Name
|
|
var BASE string = pf.Base.Name
|
|
var LOCK string = pf.Bases.Lockname
|
|
|
|
for _, KEY := range pf.Base.Unique {
|
|
fmt.Fprintln(w, "func (all *"+MSG+") DeleteBy"+KEY+"(s string) *"+BASE+" {")
|
|
fmt.Fprintln(w, " "+LOCK+".RLock()")
|
|
fmt.Fprintln(w, " defer "+LOCK+".RUnlock()")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " var newr "+BASE)
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " for i, _ := range all."+MSG+" {")
|
|
fmt.Fprintln(w, " if all."+MSG+"[i]."+KEY+" == s {")
|
|
fmt.Fprintln(w, " newr = *all."+MSG+"[i]")
|
|
fmt.Fprintln(w, " all."+MSG+"[i] = all."+MSG+"[len(all."+MSG+")-1]")
|
|
fmt.Fprintln(w, " all."+MSG+" = all."+MSG+"[:len(all."+MSG+")-1]")
|
|
fmt.Fprintln(w, " return &newr")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, " return nil")
|
|
fmt.Fprintln(w, "}")
|
|
fmt.Fprintln(w, "")
|
|
}
|
|
}
|
|
|
|
func (pf *File) findFunc(w io.Writer) {
|
|
var MSG string = pf.Bases.Name
|
|
var BASE string = pf.Base.Name
|
|
var LOCK string = pf.Bases.Lockname
|
|
|
|
for _, KEY := range pf.Base.Unique {
|
|
fmt.Fprintln(w, "// find a dependancy by the go path")
|
|
fmt.Fprintln(w, "func (all *"+MSG+") FindBy"+KEY+"(s string) *"+BASE+" {")
|
|
fmt.Fprintln(w, " if all == nil {")
|
|
fmt.Fprintln(w, " return nil")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " "+LOCK+".RLock()")
|
|
fmt.Fprintln(w, " defer "+LOCK+".RUnlock()")
|
|
fmt.Fprintln(w, "")
|
|
fmt.Fprintln(w, " for i, _ := range all."+MSG+" {")
|
|
fmt.Fprintln(w, " if all."+MSG+"[i]."+KEY+" == s {")
|
|
fmt.Fprintln(w, " return all."+MSG+"[i]")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, " }")
|
|
fmt.Fprintln(w, " return nil")
|
|
fmt.Fprintln(w, "}")
|
|
fmt.Fprintln(w, "")
|
|
}
|
|
}
|