aboutsummaryrefslogtreecommitdiff
path: root/forged/internal/misc/iter.go
diff options
context:
space:
mode:
authorRunxi Yu <me@runxiyu.org>2025-04-06 09:26:46 +0800
committerRunxi Yu <me@runxiyu.org>2025-04-06 09:27:53 +0800
commitda1d8f4e7c332c7109427915e6459b10209cedce (patch)
tree280b921be3b51f93d82d916b4eaa89387b7102cc /forged/internal/misc/iter.go
parentgit2c, git2d: Rename cmd1 and cmd2 descriptively (diff)
downloadforge-38ae46993306df07a82941afb85dd083d6388821.tar.gz
forge-38ae46993306df07a82941afb85dd083d6388821.tar.zst
forge-38ae46993306df07a82941afb85dd083d6388821.zip
Move the Go stuff to ./forged/v0.1.32
Diffstat (limited to 'forged/internal/misc/iter.go')
-rw-r--r--forged/internal/misc/iter.go23
1 files changed, 23 insertions, 0 deletions
diff --git a/forged/internal/misc/iter.go b/forged/internal/misc/iter.go
new file mode 100644
index 0000000..61a96f4
--- /dev/null
+++ b/forged/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++
+ }
+ }
+}