本文介绍了在 Go 语言中阻塞主 Goroutine 的几种方法，重点讲解了使用 sync.WaitGroup 和 select {} 的实现原理和适用场景。前者适用于需要等待子 Goroutine 完成任务的情况，后者则适用于无限期阻塞，让其他 Goroutine 在后台运行。选择合适的方法可以有效地控制 Goroutine 的执行流程。
在 Go 语言中，有时我们需要阻塞主 Goroutine，让其他的 Goroutine 在后台执行。这种情况通常发生在主 Goroutine 负责启动后台服务，然后需要无限期等待，防止程序退出。本文将介绍几种实现 Goroutine 阻塞的方法。
使用 sync.WaitGroup
sync.WaitGroup 是 Go 语言中用于等待一组 Goroutine 完成的常用工具。 它可以用来确保所有子 Goroutine 都执行完毕后再继续执行主 Goroutine。虽然它主要用于等待 Goroutine 完成，但也可以通过一些技巧来实现阻塞。
以下是使用 sync.WaitGroup 的基本步骤：
- 创建一个 sync.WaitGroup 实例。
- 在启动每个 Goroutine 之前，调用 Add(1) 来增加计数器的值。
- 在每个 Goroutine 完成时，调用 Done() 来减少计数器的值。
- 在主 Goroutine 中，调用 Wait() 来阻塞，直到计数器的值为零。
例如：
package main
import (
"fmt"
"sync"
"time"
)
func worker(id int, wg *sync.WaitGroup) {
defer wg.Done() // Goroutine 完成时减少计数器
fmt.Printf("Worker %d starting\n", id)
time.Sleep(time.Second) // 模拟工作
fmt.Printf("Worker %d done\n", id)
}
func main() {
var wg sync.WaitGroup
for i := 1; i <= 3; i++ {
wg.Add(1) // 启动 Goroutine 前增加计数器
go worker(i, &wg)
}
wg.Wait() // 阻塞直到所有 Goroutine 完成
fmt.Println("All workers done")
}注意事项：
- 确保每个 Goroutine 在退出前都调用 Done()，否则 Wait() 将会一直阻塞。
- sync.WaitGroup 适用于需要等待子 Goroutine 完成任务的情况。
使用 select {}
如果不需要等待任何 Goroutine 完成，只是想让主 Goroutine 无限期阻塞，可以使用 select {} 语句。
select 语句用于在多个 channel 操作中进行选择。当 select 语句没有任何 case 时，它会无限期阻塞。
package main
import (
"fmt"
"time"
)
func backgroundTask() {
for {
fmt.Println("Background task running...")
time.Sleep(time.Second)
}
}
func main() {
go backgroundTask()
fmt.Println("Main goroutine blocking...")
select {} // 无限期阻塞
}原理：
根据 Go 语言规范，如果 select 语句没有任何 case 并且没有 default case，那么它会无限期阻塞，直到有 case 可以执行。由于这里没有任何 case，所以 select {} 会一直阻塞，从而阻止主 Goroutine 退出。
注意事项：
- select {} 适用于不需要等待任何 Goroutine，只需要无限期阻塞的情况。
- 使用 select {} 可以有效地将控制权交给其他的 Goroutine。
使用无缓冲 Channel
使用无缓冲 channel 也可以实现阻塞的效果。主 Goroutine 尝试从一个永远不会被写入的 channel 中读取数据，就会被永久阻塞。
package main
import (
"fmt"
"time"
)
func backgroundTask() {
for {
fmt.Println("Background task running...")
time.Sleep(time.Second)
}
}
func main() {
go backgroundTask()
fmt.Println("Main goroutine blocking...")
var ch chan int = make(chan int)
<-ch // 阻塞等待从 channel 中读取数据
}注意事项：
- 这种方法与 select {} 类似，都会导致永久阻塞。
- 需要注意 channel 的类型，确保不会与其他地方的代码冲突。
总结
本文介绍了三种在 Go 语言中阻塞 Goroutine 的方法：sync.WaitGroup、select {} 和无缓冲 Channel。选择哪种方法取决于具体的需求。
- 如果需要等待一组 Goroutine 完成，应该使用 sync.WaitGroup。
- 如果只需要无限期阻塞主 Goroutine，可以使用 select {} 或者无缓冲 Channel。
理解这些方法可以帮助你更好地控制 Goroutine 的执行流程，编写更高效、更可靠的 Go 程序。
