Golang并发编程-Goroutine
package main
import (
"fmt"
"sync"
"time"
)
func worker(id int, wg *sync.WaitGroup) {
fmt.Printf("Worker %d starting\n", id)
time.Sleep(2 * time.Second)
fmt.Printf("Worker %d done\n", id)
wg.Done()
}
func main() {
var wg sync.WaitGroup
for i := 1; i <= 5; i++ {
wg.Add(1)
go worker(i, &wg) // 启动goroutine
}
wg.Wait() // 等待所有goroutine完成
}
这段代码使用了sync.WaitGroup
来同步主goroutine和工作goroutine。主goroutine启动了5个工作goroutine,并使用wg.Add(1)
来跟踪goroutine的数量。每个工作goroutine完成后通过调用wg.Done()
来通知wg.Wait()
它已经完成。这确保了主goroutine会等待所有工作goroutine完成后再继续执行。
评论已关闭