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

Go iter.Pull2 怎样让分页生成器在空结果时结束

来源:17golang原创

时间:2026-10-31 15:31:32 448浏览 收藏

我第一次把分页查询改成迭代器时,最容易漏掉的不是泛型,而是“什么时候结束”。如果把每一页结果转成 iter.Seq2,空页应该让生成器返回;如果消费者提前拿到目标记录,也要调用 stop 让生成器完成清理。iter.Pull2 正好把推送式键值迭代器变成逐条拉取的 next,但它不会替你猜测分页边界。

官方文档中的 Pull2 返回 next func() (K, V, bool) 和 stop func()。ok=false 表示没有下一对有效键值;消费尚未结束就退出时,应使用 defer stop()。

我先把分页状态放进 Seq2

分页接口通常有游标和记录列表。下面的示例用页码作为键、记录作为值;真实项目可以把键换成游标字符串。关键点是:空结果直接结束当前生成器,而不是继续请求同一个游标。

package main

import "iter"

type Record struct {
	ID   string
	Title string
}

// PageLoader 代表一次分页请求;空切片表示没有下一页。
type PageLoader func(page int) (items []Record, nextPage int)

// Pages 把分页结果包装成带页码的 Seq2,空页会结束生成器。
func Pages(load PageLoader) iter.Seq2[int, Record] {
	return func(yield func(int, Record) bool) {
		page := 1
		for {
			items, nextPage := load(page)
			if len(items) == 0 {
				return // 空页是结束条件,不能原地重试同一页。
			}
			for _, item := range items {
				if !yield(page, item) {
					return // 下游停止时立即结束上游生成器。
				}
			}
			if nextPage 
分页源、Seq2、yield 与空页结束之间的静态关系说明图
图1:结构说明图,分页源通过 Seq2 和 yield 暴露页码与记录关系;这是静态说明图,不是运行截图。

这里的 yield 返回 false 时,上游必须立即返回。这样消费者就能表达“我不再需要后续页”,而不必把所有结果拉完。

用 Pull2 按需读取键值对

当调用方不适合写 range,例如只想找到第一个匹配记录,可以把 Seq2 转成 next。每次读取都同时接收页码、记录和有效标志。

func FindTitle(seq iter.Seq2[int, Record], want string) (Record, bool) {
	next, stop := iter.Pull2(seq)
	defer stop() // 覆盖命中、错误和提前返回,统一通知生成器收尾。

	for {
		page, item, ok := next()
		if !ok {
			return Record{}, false // 空页或自然结束,不把零值当成真实记录。
		}
		_ = page // 需要记录来源页时可以写入日志或结果结构。
		if item.Title == want {
			return item, true // 提前命中,defer stop 负责结束上游。
		}
	}
}

不要只判断记录是否为零值;有效标志才是序列边界。因为合法数据本身也可能包含空字符串或零值字段。

空页、提前退出和 stop 的责任边界

对我来说,最稳妥的组合是“生成器负责定义结束条件,消费者负责保证 stop”。自然读完最后一项后,next 会返回零值对和 false;消费者提前命中、出错或被上层取消时,defer stop() 仍然会运行。

消费者、Pull2、next、stop、分页生成器和资源清理的静态关系说明图
图2:边界关系说明图,消费者通过 next 读取,通过 stop 把清理责任传回分页生成器;这是静态说明图,不是运行截图。

官方约束还包括:同一个 Pull2 实例不能被多个 goroutine 同时调用 next 或 stop。如果需要并发消费,应在外层设计单一拉取者,再把结果通过受控队列交给其他 goroutine。

我会重点检查的三个坑

  • 空页没有 return:分页游标不变时会重复请求,形成死循环。
  • 提前 return 没有 stop:上游生成器可能无法完成关闭文件、释放响应体或结束后台工作。
  • 把 ok 丢掉:next 返回的零值不代表真实记录,必须先判断 ok。

可以用下面的速查表判断边界:

场景生成器动作消费者动作
当前页为空return收到 ok=false
游标不再前进return避免重复拉取
提前命中响应 stopdefer stop()
并发读取保持单一调用边界不要并发调用 next/stop

相关问题

Pull2 自然读完后还要调用 stop 吗?

可以调用。官方文档说明在已经收到 ok=false 后调用 stop 是有效的,使用 defer stop() 可以统一覆盖所有退出路径。

空页能不能继续请求下一页?

只有接口明确提供独立的下一游标时才可以;如果空页意味着没有下一游标,应直接结束,避免拿着旧游标重复请求。

next 和 stop 能放到不同 goroutine 吗?

不能直接这样做。一个 Pull2 实例的 next 和 stop 不应被多个 goroutine 同时调用,需要由单个协调者串行管理。

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