在 Go 语言中，select 语句用于在多个发送或接收操作中进行选择。当没有 case 准备好时，select 语句的行为取决于是否定义了 default 分支。如果定义了 default 分支，则会执行 default 分支；如果没有定义，则 select 语句会阻塞，直到至少有一个 case 准备好。
default 分支的行为
default 分支在 select 语句中扮演着重要的角色，它允许我们在没有其他 case 可执行时执行一段代码。然而，如果不小心使用，default 分支可能会导致意想不到的结果。
例如，考虑以下代码：
package main
import (
"fmt"
"time"
)
func main() {
tick := time.Tick(100 * time.Millisecond)
boom := time.After(500 * time.Millisecond)
for {
select {
case <-tick:
fmt.Println("tick.")
case <-boom:
fmt.Println("BOOM!")
return
default:
fmt.Println(" .")
time.Sleep(50 * time.Millisecond)
}
}
}这段代码使用 select 语句来监听 tick 和 boom 两个 channel。如果 tick channel 收到数据，则打印 "tick."；如果 boom channel 收到数据，则打印 "BOOM!" 并退出程序；否则，执行 default 分支，打印 " ." 并休眠 50 毫秒。
如果将 default 分支中的代码移除，如下所示：
package main
import (
"fmt"
"time"
)
func main() {
tick := time.Tick(100 * time.Millisecond)
boom := time.After(500 * time.Millisecond)
for {
select {
case <-tick:
fmt.Println("tick.")
case <-boom:
fmt.Println("BOOM!")
return
default:
}
}
}这段代码会陷入无限循环，因为 default 分支总是准备好执行，导致 select 语句永远不会阻塞。由于 Go 协程是非抢占式的，如果没有 I/O 操作，计时器将永远不会触发。
解决方案
有几种方法可以解决这个问题：
- 添加 I/O 操作: 可以在 default 分支中添加 I/O 操作，例如 time.Sleep()，让出 CPU 时间，允许其他 goroutine 运行。
- 使用 runtime.Gosched(): runtime.Gosched() 函数可以显式地让出 CPU 时间，允许其他 goroutine 运行。
- 调整 runtime.GOMAXPROCS(): runtime.GOMAXPROCS() 函数可以设置 Go 程序可以同时使用的 CPU 核心数。如果将其设置为大于 1 的值，则可以允许多个 goroutine 同时运行。
- 移除 default 分支: 这是最推荐的解决方案。如果不需要在没有其他 case 准备好时执行任何操作，则可以简单地移除 default 分支。select 语句会阻塞，直到至少有一个 case 准备好。 如果需要在后台执行一些处理，可以使用 goroutine。
例如，以下代码展示了如何使用 goroutine 来执行后台处理：
package main
import (
"fmt"
"time"
)
func main() {
tick := time.Tick(100 * time.Millisecond)
boom := time.After(500 * time.Millisecond)
// 启动一个 goroutine 来执行后台处理
go func() {
for {
// 执行后台处理
fmt.Println("Background processing...")
time.Sleep(200 * time.Millisecond)
}
}()
for {
select {
case <-tick:
fmt.Println("tick.")
case <-boom:
fmt.Println("BOOM!")
return
}
}
}这段代码启动了一个 goroutine 来执行后台处理，而 select 语句只负责监听 tick 和 boom 两个 channel。这样可以避免 default 分支导致的无限循环和阻塞问题。
总结
select 语句是 Go 语言中一个强大的工具，但如果不小心使用，可能会导致意想不到的结果。在使用 default 分支时，需要特别注意其行为，避免导致无限循环和阻塞。最推荐的做法是移除 default 分支，并使用 goroutine 来执行后台处理。
在实际开发中，应该根据具体的需求选择合适的解决方案。如果需要在没有其他 case 准备好时执行一些操作，可以考虑使用 I/O 操作或 runtime.Gosched() 来让出 CPU 时间。如果不需要执行任何操作，则应该移除 default 分支，并使用 goroutine 来执行后台处理。
