
本文介绍了在 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 startingn", id) time.Sleep(time.Second) // 模拟工作 fmt.Printf("Worker %d donen", 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 程序。


