From 38b24afcb653da88765c99049f850c3ec7e234db Mon Sep 17 00:00:00 2001 From: Runxi Yu Date: Sat, 5 Apr 2025 23:22:51 +0800 Subject: misc: Move IterSeqLimit to misc --- internal/misc/iter.go | 23 +++++++++++++++++++++++ iter.go | 23 ----------------------- 2 files changed, 23 insertions(+), 23 deletions(-) create mode 100644 internal/misc/iter.go delete mode 100644 iter.go diff --git a/internal/misc/iter.go b/internal/misc/iter.go new file mode 100644 index 0000000..61a96f4 --- /dev/null +++ b/internal/misc/iter.go @@ -0,0 +1,23 @@ +// SPDX-License-Identifier: AGPL-3.0-only +// SPDX-FileCopyrightText: Copyright (c) 2025 Runxi Yu + +package misc + +import "iter" + +// iterSeqLimit returns an iterator equivalent to the supplied one, but stops +// after n iterations. +func IterSeqLimit[T any](s iter.Seq[T], n uint) iter.Seq[T] { + return func(yield func(T) bool) { + var iterations uint + for v := range s { + if iterations > n-1 { + return + } + if !yield(v) { + return + } + iterations++ + } + } +} diff --git a/iter.go b/iter.go deleted file mode 100644 index e237118..0000000 --- a/iter.go +++ /dev/null @@ -1,23 +0,0 @@ -// SPDX-License-Identifier: AGPL-3.0-only -// SPDX-FileCopyrightText: Copyright (c) 2025 Runxi Yu - -package forge - -import "iter" - -// iterSeqLimit returns an iterator equivalent to the supplied one, but stops -// after n iterations. -func iterSeqLimit[T any](s iter.Seq[T], n uint) iter.Seq[T] { - return func(yield func(T) bool) { - var iterations uint - for v := range s { - if iterations > n-1 { - return - } - if !yield(v) { - return - } - iterations++ - } - } -} -- cgit v1.2.3