登录
推荐 文章 Go 技术 课程 下载 专题 AI
首页 >  Golang >  Go教程

Go iter.Seq2 怎样遍历键值对迭代器

来源:17golang原创

时间:2026-10-14 19:45:44 258浏览 收藏

用 Go 遍历一组键值对时,iter.Seq2[K, V] 可以把“产生数据”和“消费数据”分开:生产者调用 yield(key, value),消费者直接写 for key, value := range seq。真正容易出错的地方不在两个变量,而在提前结束时的布尔返回。

只要把 false 当成“消费者不再接收”,生产者在每次调用 yield 后检查返回值并立即 return,就能安全遍历键值对;消费者里的 break 会触发这条停止信号。
要点速览
  • iter.Seq2[K, V] 的本质是接收 func(K, V) bool 的函数。
  • 生产者必须检查 yield 的返回值,不能在收到 false 后继续推送。
  • for key, value := range seq 适合顺序消费;需要逐个拉取时再考虑 iter.Pull2。

先看清 iter.Seq2 的两个值和一个布尔返回

Go 1.23 开始,for range 可以遍历符合约定的函数类型。标准库 iter 将双值形式命名为:

type Seq2[K, V any] func(yield func(K, V) bool)

这里的 K 和 V 是每一项的键和值,yield 是由消费者提供、由生产者反复调用的回调。回调返回 true 表示继续,返回 false 表示停止。它不是第三个业务字段,而是迭代控制信号。

位置含义常见写法
生产者发送一对键值yield(key, value)
消费者接收一对键值for key, value := range seq
返回值是否继续发送if !yield(...) { return }

用 yield 发送键值对并支持提前结束

下面的例子把切片中的记录包装成 iter.Seq2。注意每次调用 yield 后都立即判断结果;这既是正确性要求,也能避免消费者已经退出后仍访问外部资源。

package main

import (
	"fmt"
	"iter"
)

type entry struct {
	key   string
	value int
}

func entries() iter.Seq2[string, int] {
	items := []entry{{key: "go", value: 23}, {key: "iter", value: 2}, {key: "stop", value: 0}}
	return func(yield func(string, int) bool) {
		for _, item := range items {
			// false 代表消费者停止,生产者必须立刻返回。
			if !yield(item.key, item.value) {
				return
			}
		}
	}
}

func main() {
	for key, value := range entries() {
		// range 会把 yield 传来的两个值绑定到这两个变量。
		fmt.Println(key, value)
	}
}
Go iter.Seq2 生产者、yield 双值回调与键值记录之间的静态关系说明图
图1:结构说明图,展示 iter.Seq2、yield、键和值之间的静态关系,不是运行截图。

这个写法的价值是统一接口:数据可以来自切片、树或分页读取器,只要对外返回同一种 Seq2,调用方就能保持相同的遍历语法。

range 遍历时 break 会怎样传回生产者

当消费者只需要前两项时,可以直接使用 break。编译器会把循环体结束时的继续动作映射为回调返回 true,把 break 映射为返回 false。因此生产者的检查不能省略。

func firstPositive(seq iter.Seq2[string, int]) (string, int, bool) {
	var foundKey string
	var foundValue int
	found := false
	for key, value := range seq {
		// 找到第一个正数后 break,让上游停止继续产生数据。
		if value > 0 {
			foundKey, foundValue, found = key, value, true
			break
		}
	}
	return foundKey, foundValue, found
}

如果自定义迭代器在 yield 返回 false 后仍继续调用它,标准迭代器语义就被破坏了。对于需要手动逐项获取的场景,iter.Pull2 会返回 next 和 stop;未消费到结尾就停止时,应通过 defer stop() 释放生产者。

Go iter.Seq2 消费者 break、yield false 与生产者 return 的停止边界说明图
图2:边界说明图,展示消费者 break 与生产者 return 的停止关系,不是运行截图。

常见问题

iter.Seq2 能不能只接收值不接收键?

可以写成 for _, value := range seq,用空白标识符忽略键;如果接口本来只有一个值,更适合返回 iter.Seq[V]。

为什么 yield 返回 false 后不能继续循环?

因为 false 表示消费者已经结束接收。继续调用可能触发迭代器内部的非法状态,也会让文件、锁或其他资源的关闭边界变得不明确。

什么时候不用 Seq2?

如果调用方需要随机访问、修改元素或严格控制每次拉取,切片、位置对象或 iter.Pull2 可能更直观;不要为了使用 range 而强行包装简单数据。

声明:本文转载于:17golang原创 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>