sing-box/experimental/libbox/iterator.go

package libbox

import "github.com/sagernet/sing/common"

type StringIterator interface {
	Next() string
	HasNext() bool
}

var _ StringIterator = (*iterator[string])(nil)

type iterator[T any] struct {
	values []T
}

func newIterator[T any](values []T) *iterator[T] {
	return &iterator[T]{values}
}

func newPtrIterator[T any](values []T) *iterator[*T] {
	return &iterator[*T]{common.Map(values, func(value T) *T { return &value })}
}

func (i *iterator[T]) Next() T {
	if len(i.values) == 0 {
		return common.DefaultValue[T]()
	}
	nextValue := i.values[0]
	i.values = i.values[1:]
	return nextValue
}

func (i *iterator[T]) HasNext() bool {
	return len(i.values) > 0
}

type abstractIterator[T any] interface {
	Next() T
	HasNext() bool
}

func iteratorToArray[T any](iterator abstractIterator[T]) []T {
	if iterator == nil {
		return nil
	}
	var values []T
	for iterator.HasNext() {
		values = append(values, iterator.Next())
	}
	return values
}