misc: Move IterSeqLimit to misc

1 files changed, 23 insertions, 0 deletions

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 <https://runxiyu.org>
+
+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++
+		}
+	}
+}