Bagaimanakah fungsi golang menunggu goroutine selesai melalui WaitGroup?

WaitGroup ialah mekanisme penyegerakan dalam Go untuk menyelaraskan pelaksanaan goroutine. Gunakan ini untuk memastikan bahawa goroutine utama menunggu sekumpulan goroutine selesai sebelum meneruskan pelaksanaan. Penggunaan khusus: Buat instance WaitGroup dan tetapkan pembilang awal untuknya, menunjukkan bilangan goroutine yang perlu ditunggu. Untuk setiap goroutine menunggu, hubungi wg.Add(1) untuk menambah kaunter. Selepas goroutine selesai, panggil wg.Done() untuk mengurangkan kaunter. Akhir sekali, goroutine utama memanggil wg.Wait() untuk menyekat dirinya sehingga semua goroutine selesai.

Cara menggunakan WaitGroup untuk menunggu Goroutine selesai dalam Go

WaitGroup dalam Go ialah mekanisme penyegerakan yang digunakan untuk menyelaraskan pelaksanaan berbilang goroutin. Ia membolehkan goroutine utama menunggu sekumpulan goroutine selesai, sekali gus memastikan semua tugasan berkaitan telah selesai sebelum meneruskan pelaksanaan. WaitGroup 是一种同步机制，用于协调多个 goroutine 的执行。它允许主 goroutine 等待一组 goroutine 完成，从而确保在继续执行之前所有相关任务都已完成。

用法：

创建一个 WaitGroup 实例，并为其设置一个初始计数器，表示要等待的 goroutine 数。对于每个要等待的 goroutine，调用 wg.Add(1) 来增加计数器，并在 goroutine 完成后调用 wg.Done() 来减少计数器。最后，主 goroutine 调用 wg.Wait() 来阻塞自身，直到所有 goroutine 都已完成。

代码示例：

package main

import (
    "fmt"
    "sync"
)

var wg sync.WaitGroup

func main() {
    // 创建一个需要等待 3 个 goroutine 完成的 WaitGroup
    wg.Add(3)

    // 创建 3 个 goroutine
    for i := 0; i < 3; i++ {
        go func(i int) {
            fmt.Printf("Goroutine %d started\n", i)
            
            // 模拟一些需要时间的工作
            for j := 0; j < 10000; j++ {}
            
            fmt.Printf("Goroutine %d finished\n", i)
            
            // 通知 WaitGroup 该 goroutine 已完成
            wg.Done()
        }(i)
    }

    // 等待所有 goroutine 完成
    wg.Wait()
    fmt.Println("All goroutines have finished")
}

实战案例：

此代码示例模拟了一个简单的 Web 服务器，该服务器会启动多个 goroutine 来处理请求。为了确保在关闭服务器之前 goroutine 已完成，我们使用 WaitGroup

🎜Penggunaan: 🎜🎜🎜Buat contoh WaitGroup dan tetapkan pembilang awal untuknya, menunjukkan bilangan goroutine yang perlu ditunggu. Untuk setiap goroutine menunggu, panggil wg.Add(1) untuk menaikkan kaunter dan apabila goroutine selesai, panggil wg.Done() untuk mengurangkan kaunter. Akhir sekali, goroutine utama memanggil wg.Wait() untuk menyekat dirinya sehingga semua goroutine selesai. 🎜🎜🎜Sampel kod: 🎜🎜

package main

import (
    "fmt"
    "net/http"
    "sync"
)

var wg sync.WaitGroup

func main() {
    // 创建一个需要等待所有 HTTP 请求完成的 WaitGroup
    wg.Add(10)

    // 监听端口 8080
    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        fmt.Fprintf(w, "Hello, World!")
        
        // 请求处理完毕后，通知 WaitGroup
        wg.Done()
    })
    http.ListenAndServe(":8080", nil)
    
    // 等待所有 HTTP 请求完成
    wg.Wait()
    fmt.Println("All HTTP requests have been processed")
}

🎜🎜Kes praktikal: 🎜🎜🎜Sampel kod ini mensimulasikan pelayan web ringkas yang memulakan berbilang goroutine untuk mengendalikan permintaan. Untuk memastikan bahawa goroutine telah selesai sebelum menutup pelayan, kami menggunakan WaitGroup untuk menunggu semua goroutine selesai. 🎜rreeee

Atas ialah kandungan terperinci Bagaimanakah fungsi golang menunggu goroutine selesai melalui WaitGroup?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!