GO语言实战之并发和 goroutine

package main
 
import (
    "fmt"
    "sync"
    "time"
)
 
func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done() // 确保 wg.Done() 会在函数退出时被调用
    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 <= 5; i++ {
        wg.Add(1) // 为每个 worker 增加一个计数
        go worker(i, &wg) // 启动一个 goroutine 执行 worker 函数
    }
    wg.Wait() // 等待所有 goroutines 完成
}

这段代码使用了sync.WaitGroup来同步并发执行的 goroutines。主函数中，我们启动了5个并发的 goroutines，每个 goroutine 都执行 worker 函数。使用 wg.Add(1) 在 goroutine 启动前增加计数，在 worker 函数结束前调用 wg.Done()，以此来保证主函数中的 wg.Wait() 能够正确等待所有 worker 完成。这是 Go 语言中并发编程的一个基本示例。