一次CPU跑满的程序调试经历(Golang)

最近尝试用Go写程序。写了一个tcp网络服务，跑起来之后会把所有的CPU都跑到了100%，略吓人。
经过简单分析发现了问题所在

package main

import (
    "runtime"
    "time"
)

func main() {
    quit := make(chan bool)
    for i := 0; i != runtime.NumCPU(); i++ {
        go func() {
            for {
                select {
                case <-quit:
                    break
                default:
                }
            }
        }()
    }

    time.Sleep(time.Second * 15)
    for i := 0; i != runtime.NumCPU(); i++ {
        quit <- true
    }
}

上面这段代码会把所有CPU都跑满，原因就就在select的用法上。

一般来说，我们都用select监听各个case的IO事件，每个case都是阻塞的。上面的例子中，我们希望select在获取到quit通道里面的数据时退出循环。而实际上每一次循环select会执行满足条件的case，当quit通道中没有数据时，即不满足该case的条件，就会直接跳到default，然后退出select；而外面又是一个死循环，所以直到quit通道里读到数据，否则会一直在一个死循环中运行，即使放到一个goroutine里运行，也是会占满所有的CPU。

解决方法也很简单，把default去掉就可以了，这样select就会一直阻塞在quit通道的IO上， 把这个循环放到goroutine中，就能够随时响应通道中的信息。

如果select没有放在循环中，还是可以加default的，比如用来判断一个channel是否已满：

ch := make (chan int, 1)
ch <- 1
select {
case ch <- 2:
default:
    fmt.Println("channel is full !")
}

因为ch插入 1 的时候已经满了， 当ch要插入 2 的时候，发现ch已经满了（case1 阻塞住）， 则select执行default语句。 这样就可以实现对channel是否已满的检测， 而不是一直等待。

参考：golang的select典型用法

有疑问加站长微信联系（非本文作者）