diff options
Diffstat (limited to 'sortFunc.go')
| -rw-r--r-- | sortFunc.go | 293 |
1 files changed, 293 insertions, 0 deletions
diff --git a/sortFunc.go b/sortFunc.go new file mode 100644 index 0000000..2bcff18 --- /dev/null +++ b/sortFunc.go @@ -0,0 +1,293 @@ +package main + +import ( + "fmt" + "io" +) + +func (pf *File) syncLock(w io.Writer) { + var LOCK string = pf.Bases.Lockname + + fmt.Fprintln(w, "// bad global lock until modifying the .pb.go file is tested") + fmt.Fprintln(w, "// sync.RWMutex or sync.Mutex?") + fmt.Fprintln(w, "var "+LOCK+" sync.RWMutex") + fmt.Fprintln(w, "") +} + +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) iterAll(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 (all *"+BASES+") All() *"+BASE+"Iterator {") + fmt.Fprintln(w, " "+BASE+"Pointers := all.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 (all *"+BASES+") Len() int {") + fmt.Fprintln(w, " "+LOCK+".RLock()") + fmt.Fprintln(w, " defer "+LOCK+".RUnlock()") + fmt.Fprintln(w, "") + fmt.Fprintln(w, " return len(all."+BASES+")") + fmt.Fprintln(w, "}") + fmt.Fprintln(w, "") +} + +func (pf *File) iterSortBy(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 (all *"+BASES+") SortBy"+SORT+"() *"+BASE+"Iterator {") + fmt.Fprintln(w, " things := all.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 (all *"+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(all."+BASES+"))") + fmt.Fprintln(w, " for i, p := range all."+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) appendUnique(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) iterReplace(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) iterDelete(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) iterDeleteWithCopy(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) iterFind(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, "") + } +} |