diff options
author | Runxi Yu <me@runxiyu.org> | 2025-04-06 09:26:46 +0800 |
---|---|---|
committer | Runxi Yu <me@runxiyu.org> | 2025-04-06 09:27:53 +0800 |
commit | da1d8f4e7c332c7109427915e6459b10209cedce (patch) | |
tree | 280b921be3b51f93d82d916b4eaa89387b7102cc /forged/internal/misc/iter.go | |
parent | git2c, git2d: Rename cmd1 and cmd2 descriptively (diff) | |
download | forge-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.go | 23 |
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++ + } + } +} |