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

Go iter.Seq 怎样给自定义集合提供范围迭代器处理空输入并保留错误边界

来源:17golang原创

时间:2026-11-07 03:01:54 407浏览 收藏

给自定义集合增加返回 iter.Seq[T] 的方法,就能用 for value := range collection.All() 遍历。迭代器内部向 yield 交付元素;收到 false 后立即返回。空集合正常结束,输入是否合法则由构造函数的 error 表达,不能拿 yield 的布尔值冒充错误通道。

官方地址:https://pkg.go.dev/iter

先把小项目的接口约定说清楚

我会把这种改造限定在一个很小的对象上:保存若干标签,允许调用者读完全部标签,也允许找到第一项就退出。这里选用已经在内存中的字符串,避免把文件读取、关闭资源和解析错误混进第一个示例。

项目约定是:构造时修剪标签两侧空白,空白标签返回错误;nil 输入和长度为零的输入都是合法空集合;保留标签顺序与重复项。构造成功后集合只读,每次遍历都从头开始。下面是本次设计示例,不是已有项目的运行记录。

环境使用支持函数范围迭代的 Go 1.23 或更高版本。保存为独立目录中的 main.go,准备对应工具链即可。实现只有标准库依赖,无需部署服务。

实现一个拥有数据的标签集合

Labels 只暴露构造入口与 All 方法。内部 items 是自己构建的切片,调用者修改原输入切片的元素不会替换已经保存的标签。All 使用值接收者,让零值 Labels 也能直接参与遍历;它不修改 items,不维护跨遍历的游标。

package main

import (
    "fmt"
    "iter"
    "strings"
)

type Labels struct {
    items []string
}

// NewLabels 在构造时检查输入,成功对象不再依赖外部切片。
func NewLabels(input []string) (Labels, error) {
    items := make([]string, 0, len(input))
    for i, text := range input {
        label := strings.TrimSpace(text)
        if label == "" {
            // 失败时返回零值对象,不暴露一半构造的集合。
            return Labels{}, fmt.Errorf("第 %d 个标签为空", i+1)
        }
        items = append(items, label)
    }
    return Labels{items: items}, nil
}

// All 按保存顺序迭代;每次调用返回的函数都从头访问。
func (l Labels) All() iter.Seq[string] {
    return func(yield func(string) bool) {
        for _, label := range l.items {
            // false 表示消费方已结束,必须立即返回。
            if !yield(label) {
                return
            }
        }
    }
}

func main() {
    labels, err := NewLabels([]string{" copper ", "linen", "stone"})
    if err != nil {
        fmt.Println("无法创建标签集合:", err)
        return
    }

    // 只读取第一个标签;break 会让迭代器停止交付。
    for label := range labels.All() {
        fmt.Println("首项:", label)
        break
    }
    // 新的一次遍历仍从 copper 开始,早停没有修改集合。
    for label := range labels.All() {
        fmt.Println("标签:", label)
    }

    empty, err := NewLabels(nil)
    if err != nil {
        fmt.Println("无法创建空集合:", err)
        return
    }
    count := 0
    for range empty.All() {
        count++
    }
    fmt.Println("空集合数量:", count)

    // 构造错误与遍历结束使用不同通道表达。
    if _, err := NewLabels([]string{"copper", "   "}); err != nil {
        fmt.Println("输入有误:", err)
    }
}

把循环放在返回的函数内部很关键。外层 All 负责提供迭代器,内层函数负责实际访问集合。若把可变下标放到闭包外层并在每次遍历中递增,同一个迭代器再次使用时就可能接着上次位置读,接口会变成另一种约定。

标签集合与迭代接口的静态组成
结构图:集合保存数据,All 提供访问接口;框与连线描述静态组成,不是执行流程。

在调用端区分构造失败与提前退出

构造错误在进入 range 之前处理,所以调用方不会误把非法输入当成一个合法但没有元素的集合。示例的最后一组输入中,第二项只有空白,NewLabels 返回带索引的错误;构造函数没有把第一项作为部分结果交给调用者。

提前退出属于调用者的正常选择。示例第一次 range 只处理首项就 break;All 内部收到停止反馈后返回,没有继续调用 yield。不能忽略它返回的布尔值继续送入后续元素,也不能在返回函数结束后保存 yield 留到另一个 goroutine 中调用。

iter.Seq 的类型本身没有 error 返回位。对本例这种构造完成后只读的内存集合,错误全部在构造阶段处理就足够。若后续换成可能中途失败的数据源,应单独设计带错误的结果元素、Seq2 或显式错误获取机制,写清何时读取错误与何时停止;不要把底层读取失败悄悄吞掉。

构造错误与消费停止的接口边界
结构图:error 表示构造失败,yield 的布尔值表示消费意愿,两种含义分别归属不同接口。

空输入和重复遍历应该怎样表现

输入或动作本例约定调用者关注点
NewLabels(nil)成功创建空集合range 循环体不进入
Labels 零值同样没有元素All 返回可调用的空迭代器
包含空白标签构造失败先处理 error,再决定后续操作
break 提前退出当前遍历结束集合内容与下次遍历不变
再次完整遍历重新从首项开始可重复并非 Seq 类型自动保证,而是本例实现保证

按代码推导,首项为 copper,完整遍历依次得到 copper、linen、stone,空集合计数为零。错误输入会进入“输入有误”的分支。这些只是示例的预期行为,未展示实测日志。

空集合应返回一个什么也不交付的函数,而不是 nil 函数。本例即使 items 为 nil,也会生成一个正常的函数值,内部 range 自然结束。若让 All 直接返回 nil,消费方尝试执行这个函数时会失败,不能拿“没有元素”解释这种接口错误。

接入现有业务时保留哪些边界

我倾向先把构造阶段的输入规则稳定下来,再让业务调用者替换原来的切片循环。只需要全部遍历和条件 break 的调用者,通常无需知道 items 如何保存。若消费者需要下标和值,可以另设 Seq2 接口,不必为了给 All 增加下标而改变所有调用方。

集成时保留这三个约定:返回函数内部自行建立每次遍历状态;发现停止反馈后不再交付;没有隐藏的失败结果。本文的只读对象还意味着构造完成后不提供原地追加或删除方法。以后若允许修改,应重新定义遍历期间的修改行为与同步方式,不能沿用“只读集合”的并发假设。

如果只是包装普通切片,也可以考虑标准库已有的迭代适配器。自定义 Labels 的价值在于把“标签必须非空、数据归集合拥有、访问顺序固定”这些业务约定放在对象里,而不是仅为 range 语法再包一层。

相关问题

yield 返回 false 是不是一个错误?在这里不是,它只表示消费方不再需要后续元素。业务错误由构造函数返回。

同一个 seq 能遍历两次吗?本例可以,因为每次调用函数都重新进入内部循环。其他迭代器是否可重复,必须看其接口约定。

一定要开启 goroutine 才能实现吗?不需要。这里使用直接函数调用即可,增加后台执行反而会引入回调生命周期问题。

这个小集合适合输入已经完成检查、后续只需要顺序访问的场景;会持续读取外部数据的对象则需要更完整的错误与资源接口。

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