minor
This commit is contained in:
parent
95afc5e0b4
commit
0b4ab2fb9f
|
@ -37,7 +37,7 @@ func header(w io.Writer, pf *File) {
|
|||
fmt.Fprintf(w, "package %s\n", pf.Package)
|
||||
fmt.Fprintln(w, "")
|
||||
fmt.Fprintln(w, "import (")
|
||||
fmt.Fprintln(w, " \"fmt\"")
|
||||
// fmt.Fprintln(w, " \"fmt\"")
|
||||
fmt.Fprintln(w, " \"sort\"")
|
||||
fmt.Fprintln(w, " \"sync\"")
|
||||
fmt.Fprintln(w, ")")
|
||||
|
|
|
@ -1,63 +0,0 @@
|
|||
// DEFINE THE ITERATOR. Only one per Patch message
|
||||
|
||||
// NewPatchsetIterator initializes a new iterator.
|
||||
func NewPatchIterator(things []*Patch) *PatchIterator {
|
||||
return &PatchIterator{things: things}
|
||||
}
|
||||
|
||||
// safely returns a slice of pointers to the Patchset protobufs
|
||||
func (x *Patchset) all() []*Patch {
|
||||
x.Lock.RLock()
|
||||
defer x.Lock.RUnlock()
|
||||
|
||||
// Create a new slice to hold pointers to each Patchset
|
||||
var tmp []*Patch
|
||||
tmp = make([]*Patch, len(x.Patches))
|
||||
for i, p := range x.Patches {
|
||||
tmp[i] = p // Copy pointers for safe iteration
|
||||
}
|
||||
|
||||
return tmp
|
||||
}
|
||||
|
||||
type PatchIterator struct {
|
||||
sync.RWMutex
|
||||
|
||||
things []*Patch
|
||||
index int
|
||||
}
|
||||
|
||||
func (it *PatchIterator) Scan() bool {
|
||||
if it.index >= len(it.things) {
|
||||
return false
|
||||
}
|
||||
it.index++
|
||||
return true
|
||||
}
|
||||
|
||||
// Next() returns the next thing in the array
|
||||
func (it *PatchIterator) Next() *Patch {
|
||||
if it.things[it.index-1] == nil {
|
||||
fmt.Println("Next() error in PatchIterator", it.index)
|
||||
}
|
||||
return it.things[it.index-1]
|
||||
}
|
||||
|
||||
// END DEFINE THE ITERATOR
|
||||
|
||||
// START sort by Filename (this is all you need once the Iterator is defined)
|
||||
type PatchFilename []*Patch
|
||||
|
||||
func (a PatchFilename) Len() int { return len(a) }
|
||||
func (a PatchFilename) Less(i, j int) bool { return a[i].Filename < a[j].Filename }
|
||||
func (a PatchFilename) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
|
||||
|
||||
func (x *Patchset) SortByFilename() *PatchIterator {
|
||||
things := x.all()
|
||||
|
||||
sort.Sort(PatchFilename(things))
|
||||
|
||||
iterator := NewPatchIterator(things)
|
||||
return iterator
|
||||
}
|
||||
// END sort by Filename
|
Loading…
Reference in New Issue