迭代时替换通道

一分耕耘，一分收获！既然都打开这篇《迭代时替换通道》，就坚持看下去，学下去吧！本文主要会给大家讲到等等知识点，如果大家对本文有好的建议或者看到有不足之处，非常欢迎大家积极提出！在后续文章我会继续更新Golang相关的内容，希望对大家都有所帮助！

我想在某种情况下用新的通道替换通道，例如：

package main

import (
    "log"
    "time"
)

func subMsg(s string) chan string {
    ch := make(chan string)
    go func() {
        ticker := time.NewTicker(time.Second * 2)
        for range ticker.C {
            ch <- s
        }
    }()

    return ch
}

func main() {
    chStr := subMsg("hello")

    go func() {
        i := 0
        for s := range chStr {
            log.Print(s)
            i++
            if i > 5 {
                log.Print("new topic")
                i = 0

                chStr = subMsg("world")
            }
        }
    }()

    select {}
}

我希望这个代码片段输出 5 个“hello”，然后是“world”，但它并没有这样工作。我不太清楚重新分配频道时发生了什么。有什么建议吗？

解决方案

您正在使用 作为 range，并且根据 Spec: For statements，范围表达式仅计算一次。

范围表达式 x 在开始循环之前计算一次，但有一个例外：如果最多存在一个迭代变量且 len(x) 为 constant，则不计算范围表达式。

稍后 for 循环不会检查 chstr 变量，因此更改其值没有任何效果。

如果您想切换到另一个频道，则不能使用 作为 range。

只需使用“正常”循环并从其中的通道接收即可。使用特殊形式 x, ok := <-ch 这样你就知道通道何时关闭，这样你就可以从循环中中断（模仿 for range 工作）：

for {
    s, ok := <-chStr
    if !ok {
        break
    }
    log.Print(s)
    i++
    if i > 5 {
        log.Print("new topic")
        i = 0

        chStr = subMsg("world")
    }
}

拨打Go Playground试试。

到这里，我们也就讲完了《迭代时替换通道》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！