package main /* 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, parent, child *MsgName) { var BASES string = parent.Name var BASE string = child.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, "}") } */