node/utils/iterator/slice.go

// Copyright (C) 2019-2025, Lux Industries Inc. All rights reserved.
// See the file LICENSE for licensing terms.

package iterator

var _ Iterator[any] = (*slice[any])(nil)

// ToSlice returns a slice that contains all of the elements from [it] in order.
// [it] will be released before returning.
func ToSlice[T any](it Iterator[T]) []T {
	defer it.Release()

	var elements []T
	for it.Next() {
		elements = append(elements, it.Value())
	}
	return elements
}

type slice[T any] struct {
	index    int
	elements []T
}

// FromSlice returns an iterator that contains [elements] in order. Doesn't sort
// by anything.
func FromSlice[T any](elements ...T) Iterator[T] {
	return &slice[T]{
		index:    -1,
		elements: elements,
	}
}

func (i *slice[_]) Next() bool {
	i.index++
	return i.index < len(i.elements)
}

func (i *slice[T]) Value() T {
	return i.elements[i.index]
}

func (*slice[_]) Release() {}