Go语言用Channel实现Python风格生成器的提前读取问题求助

问题根源 #

结果差异并非Channel调度问题，而是对Go切片append操作的误解：

• 当r := make([]int, 0)时，切片初始长度为0，append会依次添加元素，最终r元素为[0,1,2,3,4]，打印正确。
• 当r := make([]int, 5)时，切片初始长度为5，元素默认值为0，append是在切片末尾追加元素，最终r内容为[0,0,0,0,0,0,1,2,3,4]。后续打印r[i]（i=0~4）取的是初始默认值，因此出现false结果。

修正切片使用 #

若要将迭代结果填充到已初始化长度的切片中，需用索引赋值替代append：

func main() {
    c := NewContainerIterable[int]()
    for i := 0; i < 5; i++ {
        c.Add(i)
    }

    r := make([]int, 5)
    idx := 0
    for v := range c.Iterate() {
        r[idx] = v
        idx++
    }

    for i := 0; i < 5; i++ {
        fmt.Println(i, r[i], i == r[i])
    }
}

迭代器并发安全优化 #

原Iterate()直接遍历原切片，若迭代中调用Add()修改容器会引发并发安全问题。可通过复制切片快照解决：

func (c *ContainerIterable[T]) Iterate() chan T {
    ch := make(chan T)
    // 复制切片快照，隔离迭代与修改操作
    snapshot := make([]T, len(c.content))
    copy(snapshot, c.content)
    go func() {
        for _, v := range snapshot {
            ch <- v
        }
        close(ch)
    }()
    return ch
}

此修改后，迭代结果始终基于调用Iterate()时的容器状态，不受后续修改影响。

内容的提问来源于stack exchange，提问作者vtm11