From 655b6b211ae6df0186abd740f248939f7ddeaec1 Mon Sep 17 00:00:00 2001
From: Runxi Yu <me@runxiyu.org>
Date: Mon, 31 Mar 2025 16:59:18 +0800
Subject: Add descriptive comments to most Go functions

---
 iter.go | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 iter.go

(limited to 'iter.go')

diff --git a/iter.go b/iter.go
new file mode 100644
index 0000000..d4c7175
--- /dev/null
+++ b/iter.go
@@ -0,0 +1,23 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// SPDX-FileCopyrightText: Copyright (c) 2025 Runxi Yu <https://runxiyu.org>
+
+package main
+
+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