diff options
Diffstat (limited to 'forgeConfig.sort.go')
| -rw-r--r-- | forgeConfig.sort.go | 151 |
1 files changed, 151 insertions, 0 deletions
diff --git a/forgeConfig.sort.go b/forgeConfig.sort.go new file mode 100644 index 0000000..4b0ff69 --- /dev/null +++ b/forgeConfig.sort.go @@ -0,0 +1,151 @@ +package forgepb + +// TODO: autogen this? (probably not feasible. need go-arglike tricks in proto) + +import ( + "fmt" + "os" + "sort" + sync "sync" + "time" +) + +// bad global lock until I figure out some other plan +var forgeConfigsLock sync.RWMutex + +type ForgeConfigIterator struct { + sync.RWMutex + + packs []*ForgeConfig + index int +} + +// NewForgeConfigIterator initializes a new iterator. +func NewForgeConfigIterator(packs []*ForgeConfig) *ForgeConfigIterator { + return &ForgeConfigIterator{packs: packs} +} + +// Scan moves to the next element and returns false if there are no more packs. +func (it *ForgeConfigIterator) Scan() bool { + if it.index >= len(it.packs) { + return false + } + it.index++ + return true +} + +// ForgeConfig returns the current forgeConfig. +func (it *ForgeConfigIterator) Next() *ForgeConfig { + if it.packs[it.index-1] == nil { + for i, d := range it.packs { + fmt.Println("i =", i, d) + } + fmt.Println("len =", len(it.packs)) + fmt.Println("forgeConfig == nil", it.index, it.index-1) + os.Exit(-1) + } + return it.packs[it.index-1] +} + +// Use Scan() in a loop, similar to a while loop +// +// for iterator.Scan() { +// d := iterator.ForgeConfig() +// fmt.Println("ForgeConfig UUID:", d.Uuid) +// } + +func (r *ForgeConfigs) All() *ForgeConfigIterator { + forgeConfigPointers := r.selectAllForgeConfigs() + + iterator := NewForgeConfigIterator(forgeConfigPointers) + return iterator +} + +func (r *ForgeConfigs) SortByPath() *ForgeConfigIterator { + packs := r.selectAllForgeConfigs() + + sort.Sort(ByForgeConfigPath(packs)) + + iterator := NewForgeConfigIterator(packs) + return iterator +} + +// enforces no duplicate forgeConfig paths +func (r *ForgeConfigs) Append(newP *ForgeConfig) bool { + forgeConfigsLock.Lock() + defer forgeConfigsLock.Unlock() + + for _, p := range r.ForgeConfigs { + if p.GoPath == newP.GoPath { + return false + } + } + + r.ForgeConfigs = append(r.ForgeConfigs, newP) + return true +} + +// returns time.Duration since last Update() +func (r *ForgeConfig) Age(newP *ForgeConfig) time.Duration { + t := time.Since(r.Verstamp.AsTime()) + return t +} + +// find a forgeConfig by path +func (r *ForgeConfigs) FindByPath(gopath string) *ForgeConfig { + forgeConfigsLock.RLock() + defer forgeConfigsLock.RUnlock() + + for _, p := range r.ForgeConfigs { + if p.GoPath == gopath { + return p + } + } + + return nil +} + +func (r *ForgeConfigs) Len() int { + forgeConfigsLock.RLock() + defer forgeConfigsLock.RUnlock() + + return len(r.ForgeConfigs) +} + +type ByForgeConfigPath []*ForgeConfig + +func (a ByForgeConfigPath) Len() int { return len(a) } +func (a ByForgeConfigPath) Less(i, j int) bool { return a[i].GoPath < a[j].GoPath } +func (a ByForgeConfigPath) Swap(i, j int) { a[i], a[j] = a[j], a[i] } + +func (all *ForgeConfigs) DeleteByPath(gopath string) *ForgeConfig { + forgeConfigsLock.Lock() + defer forgeConfigsLock.Unlock() + + var newr ForgeConfig + + for i, _ := range all.ForgeConfigs { + if all.ForgeConfigs[i].GoPath == gopath { + newr = *all.ForgeConfigs[i] + all.ForgeConfigs[i] = all.ForgeConfigs[len(all.ForgeConfigs)-1] + all.ForgeConfigs = all.ForgeConfigs[:len(all.ForgeConfigs)-1] + return &newr + } + } + return nil +} + +// safely returns a slice of pointers to the ForgeConfig protobufs +func (r *ForgeConfigs) selectAllForgeConfigs() []*ForgeConfig { + forgeConfigsLock.RLock() + defer forgeConfigsLock.RUnlock() + + // Create a new slice to hold pointers to each ForgeConfig + var allPacks []*ForgeConfig + allPacks = make([]*ForgeConfig, len(r.ForgeConfigs)) + for i, p := range r.ForgeConfigs { + allPacks[i] = p // Copy pointers for safe iteration + } + + return allPacks +} |
