mirror of
https://github.com/luxfi/container.git
synced 2026-09-24 00:24:34 +00:00
zeekayandHanzo Dev
adf4508709
Modules declare go 1.27.1 and the build images follow. go fix applied the modernizers: any, WaitGroup.Go, range over int, and a promoted-field key in a struct literal. omitzero is not applied, since it changes what the JSON says. Co-authored-by: Hanzo Dev <[email protected]>
62 lines
1.1 KiB
Go
62 lines
1.1 KiB
Go
// Copyright (C) 2019-2025, Lux Industries, Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package iterator
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/google/btree"
|
|
)
|
|
|
|
var _ Iterator[any] = (*tree[any])(nil)
|
|
|
|
type tree[T any] struct {
|
|
current T
|
|
next chan T
|
|
releaseOnce sync.Once
|
|
release chan struct{}
|
|
wg sync.WaitGroup
|
|
}
|
|
|
|
// FromTree returns a new iterator of the stakers in [tree] in ascending order.
|
|
// Note that it isn't safe to modify [tree] while iterating over it.
|
|
func FromTree[T any](btree *btree.BTreeG[T]) Iterator[T] {
|
|
if btree == nil {
|
|
return Empty[T]{}
|
|
}
|
|
it := &tree[T]{
|
|
next: make(chan T),
|
|
release: make(chan struct{}),
|
|
}
|
|
it.wg.Go(func() {
|
|
btree.Ascend(func(i T) bool {
|
|
select {
|
|
case it.next <- i:
|
|
return true
|
|
case <-it.release:
|
|
return false
|
|
}
|
|
})
|
|
close(it.next)
|
|
})
|
|
return it
|
|
}
|
|
|
|
func (i *tree[_]) Next() bool {
|
|
next, ok := <-i.next
|
|
i.current = next
|
|
return ok
|
|
}
|
|
|
|
func (i *tree[T]) Value() T {
|
|
return i.current
|
|
}
|
|
|
|
func (i *tree[_]) Release() {
|
|
i.releaseOnce.Do(func() {
|
|
close(i.release)
|
|
})
|
|
i.wg.Wait()
|
|
}
|