最近尝试用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
已经满了(case
1 阻塞住), 则select
执行default
语句。 这样就可以实现对channel
是否已满的检测, 而不是一直等待。