or-done-channel

最后更新于:2022-04-02 06:51:18

有时你会与来自系统不同部分的通道交互。与管道不同的是,当你使用的代码通过done通道取消操作时,你无法对通道的行为方式做出判断。也就是说,你不知道正在执行读取操作的goroutine现在是什么状态。出于这个原因,正如我们在“防止Goroutine泄漏”中所阐述的那样,需要用select语句来封装我们的读取操作和done通道。可以简单的写成这样: ``` for val := range myChan { // 对 val 进行处理 } ``` 展开后可以写成这样: ``` loop: for { select { case <-done: break loop case maybeVal, ok := <-myChan: if ok == false { return // or maybe break from for } // Do something with val } } ``` 这样做可以快速退出嵌套循环。继续使用goroutines编写更清晰的并发代码,而不是过早优化的主题,我们可以用一个goroutine来解决这个问题。 我们封装了细节,以便其他人调用更方便: ``` orDone := func(done, c <-chan interface{}) <-chan interface{} { valStream := make(chan interface{}) go func() { defer close(valStream) for { select { case <-done: return case v, ok := <-c: if ok == false { return } select { case valStream <- v: case <-done: } } } }() return valStream } ``` 这样做允许我们回到简单的循环方式: ``` for val := range orDone(done, myChan) { // Do something with val } ``` 你可能会在代码中发现需要使用一系列select语句的循环代码,但我会鼓励你先尝试提高可读性,并避免过早优化。 * * * * * 学识浅薄,错误在所难免。我是长风,欢迎来Golang中国的群(211938256)就本书提出修改意见。
';