Go: Limit the size of an iterator; Limit(n) or Take(n)

limit_or_take.go

func Limit[E any](seq iter.Seq[E], limit int) iter.Seq[E] {
	return func(yield func(E) bool) {
		var count int
		for e := range seq {
			if count >= limit {
				return
			}
			if !yield(e) {
				return
			}
			count++
		}
	}
}