golang 常见坑(1)-select

引

这个系列会介绍 golang 常见的坑。当然很多坑是由于对 golang 理解不到位引起的。

猜测下如下代码的输出

这是一段很简单的代码，生产者 go 程打印数字，结束之后发送 cancel 信号。 是不是认为会打印 0-999。如果是这样想的可以继续往下看。

package main

import (
        "context"
        "fmt"
)

func main() {
        ctx, cancel := context.WithCancel(context.Background())

        data := make(chan int, 1000)

        go func() {
                for i := 0; i < 1000; i++ {
                        data <- i
                }
                cancel()
        }()

        for {
                select {
                case <-ctx.Done():
                        return
                case v := <-data:
                        fmt.Printf("%d\n", v)
                }
        }
}

问题分析

你以为会打印 0-999 ？其实不是。。运行下代码你会发现。输出是随机的。what? 这其实和 select 的机制有关系。当 case 条件有多个为真，就想象成随机函数从 case 里面选择一个执行 。上面的代码是两个条件都满足，调用 cancel 函数，有些数据还缓存在 data chan 里面，ctx.Done()条件也为真。选择到 ctx.Done()的时候，这里很可能 case v:=<-data 都没打印全。

解决问题

刚刚聊了 case 的内部逻辑。再聊下如何解决这个问题。data 每个发送的数据都确保消费掉，最后再调用 cancel 函数就可解决这个问题。做法把带缓冲的 chan 修改为不带缓冲。

// data := make(chan int, 1000)
data := make(chan int)

最佳实践

如果不是必须的理由要用带缓冲的 chan。推荐使用无缓冲的 chan。至于担心的性能问题，他们性能差距不大。后面会补上 benchmark。

我的 github

https://github.com/guonaihong/gout