Home Backend Development Golang How does golang function wait for goroutine to complete through WaitGroup?

How does golang function wait for goroutine to complete through WaitGroup?

May 01, 2024 am 09:51 AM
golang Synchronization mechanism

WaitGroup is a synchronization mechanism in Go for coordinating goroutine execution. Use this to ensure that the main goroutine waits for a group of goroutines to complete before continuing execution. Specific usage: Create a WaitGroup instance and set an initial counter for it, indicating the number of goroutines to wait for. For each goroutine to wait on, call wg.Add(1) to increment the counter. After the goroutine completes, call wg.Done() to decrement the counter. Finally, the main goroutine calls wg.Wait() to block itself until all goroutines have completed.

How does golang function wait for goroutine to complete through WaitGroup?

How to use WaitGroup to wait for Goroutine to complete in Go

WaitGroup in Go is a kind of synchronization Mechanism for coordinating the execution of multiple goroutines. It allows the main goroutine to wait for a group of goroutines to complete, thereby ensuring that all related tasks have completed before continuing execution.

Usage:

Create a WaitGroup instance and set an initial counter for it, indicating the number of goroutines to wait for. For each goroutine to wait on, call wg.Add(1) to increment the counter, and after the goroutine completes, call wg.Done() to decrement the counter. Finally, the main goroutine calls wg.Wait() to block itself until all goroutines have completed.

Code sample:

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")
}

Practical case:

This code sample simulates a simple Web server that Multiple goroutines will be started to handle requests. To ensure that the goroutine has completed before shutting down the server, we use WaitGroup to wait for all goroutines to complete.

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")
}

The above is the detailed content of How does golang function wait for goroutine to complete through WaitGroup?. For more information, please follow other related articles on the PHP Chinese website!

Statement of this Website
The content of this article is voluntarily contributed by netizens, and the copyright belongs to the original author. This site does not assume corresponding legal responsibility. If you find any content suspected of plagiarism or infringement, please contact admin@php.cn

Hot AI Tools

Undress AI Tool

Undress AI Tool

Undress images for free

Undresser.AI Undress

Undresser.AI Undress

AI-powered app for creating realistic nude photos

AI Clothes Remover

AI Clothes Remover

Online AI tool for removing clothes from photos.

Clothoff.io

Clothoff.io

AI clothes remover

Video Face Swap

Video Face Swap

Swap faces in any video effortlessly with our completely free AI face swap tool!

Hot Tools

Notepad++7.3.1

Notepad++7.3.1

Easy-to-use and free code editor

SublimeText3 Chinese version

SublimeText3 Chinese version

Chinese version, very easy to use

Zend Studio 13.0.1

Zend Studio 13.0.1

Powerful PHP integrated development environment

Dreamweaver CS6

Dreamweaver CS6

Visual web development tools

SublimeText3 Mac version

SublimeText3 Mac version

God-level code editing software (SublimeText3)

What are the alternatives to standard library logging in Golang? What are the alternatives to standard library logging in Golang? Aug 05, 2025 pm 08:36 PM

FornewGo1.21 projects,useslogforofficialstructuredloggingsupport;2.Forhigh-performanceproductionservices,chooseZaporZerologduetotheirspeedandlowallocations;3.ForeaseofuseandrichintegrationslikeSlackorSentryhooks,Logrusisidealdespitelowerperformance;4

What are the best practices for API versioning in a Golang service? What are the best practices for API versioning in a Golang service? Aug 04, 2025 pm 04:50 PM

UseURLpathversioninglike/api/v1forclear,routable,anddeveloper-friendlyversioning.2.Applysemanticversioningwithmajorversions(v1,v2)only,avoidingmicro-versionslikev1.1topreventroutingcomplexity.3.OptionallysupportcontentnegotiationviaAcceptheadersifalr

How to work with NoSQL databases like MongoDB in Golang How to work with NoSQL databases like MongoDB in Golang Aug 03, 2025 pm 03:55 PM

Install MongoDBGo driver and use mongo.Connect() to establish a connection to ensure the connection is successful through Ping; 2. Define a Go structure with bson tag to map MongoDB documents, optionally use primitive.ObjectID as the ID type; 3. Use InsertOne to insert a single document, FindOne query a single document and handle mongo.ErrNoDocuments errors, UpdateOne updates the document, DeleteOne deletes the document, Find cooperates with cursor.All to get multiple documents; 4. Always use a context with timeout to avoid request hangs, and reuse Mon

How do you implement an observer pattern in Golang? How do you implement an observer pattern in Golang? Aug 14, 2025 pm 12:04 PM

In Go, observer mode can be implemented through interfaces and channels, the Observer interface can be defined, the Observer interface includes the Update method, the Subject structure maintains the observer list and message channel, add observers through Attach, Notify sends messages, listengoroutine asynchronous broadcast updates, specific observers such as EmailService and LogService implement the Update method to handle notifications, the main program registers the observer and triggers events, and realizes a loosely coupled event notification mechanism, which is suitable for event-driven systems, logging and message notifications and other scenarios.

How does Golang handle concurrency? How does Golang handle concurrency? Aug 04, 2025 pm 04:13 PM

Gohandlesconcurrencythroughgoroutinesandchannels,makingitsimpleandefficienttowriteconcurrentprograms.1.GoroutinesarelightweightthreadsmanagedbytheGoruntime,startedwiththegokeyword,andcanscaletothousandsormillionsduetosmallinitialstacksize,efficientsc

What is benchmarking in Golang? What is benchmarking in Golang? Aug 13, 2025 am 12:14 AM

Gobenchmarkingmeasurescodeperformancebytimingfunctionexecutionandmemoryusage,usingbuilt-intestingtools;benchmarksarewrittenin_test.gofileswithnamesstartingwithBenchmark,takeatesting.Bparameter,andruntargetcodeinaloopcontrolledbyb.N,whichGoautomatical

How to manage memory allocation in Golang How to manage memory allocation in Golang Aug 11, 2025 pm 12:23 PM

UnderstandGo’smemoryallocationmodelbyusingescapeanalysistominimizeheapallocations;2.Reduceheapallocationswithvaluetypes,pre-allocatedslices,andsync.Poolforbufferreuse;3.Optimizestringandbytehandlingusingstrings.Builderandreusablebyteslicestoavoidunne

What are the best practices for benchmarking code in Golang? What are the best practices for benchmarking code in Golang? Aug 03, 2025 am 06:35 AM

Use the correct benchmark function structure, name starts with BenchmarkXxx, and use b.N to automatically adjust the number of iterations; 2. Avoid unused results, set code mixing timing and memory allocation interference, eliminate deviations through black box variables or b.ResetTimer(); 3. Use b.Run() to test different input scales to analyze performance scalability; 4. Keep the environment stable, turn off background interference, control GOMAXPROCS and use -benchmem to view memory allocation; 5. Use the benchmark tool to run multiple times and analyze the results to ensure the reliability of the comparison. Follow these steps to obtain accurate, repeatable Go benchmark results.

See all articles