如何在 Go 中阻塞程序/Goroutine？

本文介绍了在 Go 语言中阻塞主 Goroutine 的几种方法，重点讲解了使用 sync.WaitGroup 和 select {} 的实现原理和适用场景。前者适用于需要等待子 Goroutine 完成任务的情况，后者则适用于无限期阻塞，让其他 Goroutine 在后台运行。选择合适的方法可以有效地控制 Goroutine 的执行流程。

在 Go 语言中，有时我们需要阻塞主 Goroutine，让其他的 Goroutine 在后台执行。这种情况通常发生在主 Goroutine 负责启动后台服务，然后需要无限期等待，防止程序退出。本文将介绍几种实现 Goroutine 阻塞的方法。

使用 sync.WaitGroup

sync.WaitGroup 是 Go 语言中用于等待一组 Goroutine 完成的常用工具。 它可以用来确保所有子 Goroutine 都执行完毕后再继续执行主 Goroutine。虽然它主要用于等待 Goroutine 完成，但也可以通过一些技巧来实现阻塞。

以下是使用 sync.WaitGroup 的基本步骤：

1. 创建一个 sync.WaitGroup 实例。
2. 在启动每个 Goroutine 之前，调用 Add(1) 来增加计数器的值。
3. 在每个 Goroutine 完成时，调用 Done() 来减少计数器的值。
4. 在主 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 时，它会无限期阻塞。

如知AI笔记

如知笔记——支持markdown的在线笔记，支持ai智能写作、AI搜索，支持DeepseekR1满血大模型

27

查看详情

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 程序。

以上就是如何在 Go 中阻塞程序/Goroutine？的详细内容，更多请关注php中文网其它相关文章！