Golang 中可使用 goroutine 实现并发,创建 goroutine 时使用 go 关键字,管理 goroutine 时使用 WaitGroup,并行执行 goroutine 时使用 Mutex。在实战案例中,可以通过并发获取网页内容来体验并发的优势,通过 goroutine 并行获取多个网页内容,并通过 WaitGroup 等待所有 goroutine 完成,最后通过 channel 获取所有并行获取的数据。
在 Golang 中,goroutine 是轻量级的协程,可以并发执行。这篇文章将介绍如何创建、管理、并行执行多个 goroutine。
使用 go
关键字创建 Goroutine:
go func() { // Goroutine 体 }
WaitGroup
用于等待一组 goroutine 完成执行:
import "sync" var wg sync.WaitGroup
在 goroutine 中,调用 wg.Done()
来通知 WaitGroup 完成:
wg.Done()
在主函数中,调用 wg.Wait()
来等待所有 goroutine 完成:
wg.Wait()
使用 sync.Mutex
来确保并行执行时数据的并发访问安全:
import "sync" var mu sync.Mutex
在并行执行代码块时,使用 mu.Lock()
和 mu.Unlock()
来控制对共享数据的访问:
mu.Lock() // 并行执行代码 mu.Unlock()
package main import ( "fmt" "io" "net/http" "time" ) func fetchURL(url string, wg *sync.WaitGroup, results chan<- string) { resp, err := http.Get(url) if err != nil { fmt.Println("Error fetching URL:", err) return } defer resp.Body.Close() body, err := io.ReadAll(resp.Body) if err != nil { fmt.Println("Error reading body:", err) return } results <- string(body) wg.Done() } func main() { urls := []string{"https://example.com", "https://google.com", "https://amazon.com"} results := make(chan string) var wg sync.WaitGroup wg.Add(len(urls)) for _, url := range urls { go fetchURL(url, &wg, results) } wg.Wait() close(results) // 获取所有并行获取的数据 for result := range results { fmt.Println(result) } }