0

0

如何用Golang实现高效并发控制 详解sync.WaitGroup使用场景

P粉602998670

P粉602998670

发布时间:2025-07-16 10:04:02

|

766人浏览过

|

来源于php中文网

原创

使用sync.waitgroup实现并发控制的关键在于正确匹配add()和done()调用。1. sync.waitgroup通过add(delta int)增加计数器,启动goroutine前调用确保计数准确;2. done()用于减少计数器,通常配合defer确保goroutine退出时执行;3. wait()阻塞主协程直到所有任务完成。常见错误包括add()与done()次数不匹配或在done()后再次调用add(),需仔细检查代码逻辑避免panic。结合context.context可实现更高级的并发控制,如超时取消机制,提升程序灵活性与健壮性。其他并发工具如sync.mutex、chan等也应根据场景灵活选用。

如何用Golang实现高效并发控制 详解sync.WaitGroup使用场景

Golang实现高效并发控制,核心在于合理利用sync.WaitGroup。它能让你优雅地等待一组goroutine完成,避免程序提前退出或资源泄漏。掌握sync.WaitGroup的用法,是写出健壮并发程序的关键一步。

如何用Golang实现高效并发控制 详解sync.WaitGroup使用场景

解决方案:

sync.WaitGroup主要用于等待一组goroutine执行完毕。它内部维护一个计数器,初始值为0。通过Add(delta int)增加计数器,Done()减少计数器,Wait()阻塞直到计数器归零。

立即学习go语言免费学习笔记(深入)”;

如何用Golang实现高效并发控制 详解sync.WaitGroup使用场景

下面是一个简单的例子:

package main

import (
    "fmt"
    "sync"
    "time"
)

func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done() // 确保goroutine退出时计数器减一
    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) // 启动一个goroutine,计数器加一
        go worker(i, &wg)
    }

    wg.Wait() // 阻塞,直到所有goroutine执行完毕
    fmt.Println("All workers done")
}

这个例子中,我们启动了5个worker goroutine,每个goroutine模拟执行一个耗时操作。wg.Add(1)确保每个goroutine启动时计数器加一,wg.Done()确保goroutine退出时计数器减一。wg.Wait()则会阻塞main goroutine,直到所有worker goroutine执行完毕。

如何用Golang实现高效并发控制 详解sync.WaitGroup使用场景

如何避免sync.WaitGroup使用中的常见错误?

最常见的错误之一是Add()Done()的调用不匹配。如果Add()的次数少于实际启动的goroutine数量,Wait()可能会提前返回,导致程序逻辑错误。反之,如果Add()的次数多于实际启动的goroutine数量,可能会导致panic。

Cogram
Cogram

使用AI帮你做会议笔记,跟踪行动项目

下载

另一个常见错误是在Done()之后再次调用Add()。这会导致计数器变为负数,并在Wait()时引发panic。

解决这些问题的关键在于仔细检查代码,确保Add()Done()的调用次数匹配,并且不会在Done()之后再次调用Add()

更高级的并发控制技巧:结合sync.WaitGroupcontext.Context

在复杂的并发场景中,我们可能需要提前取消正在执行的goroutine。这时,可以结合sync.WaitGroupcontext.Context来实现更精细的控制。

package main

import (
    "context"
    "fmt"
    "sync"
    "time"
)

func workerWithContext(ctx context.Context, id int, wg *sync.WaitGroup) {
    defer wg.Done()

    fmt.Printf("Worker %d starting\n", id)
    defer fmt.Printf("Worker %d done\n", id)

    for {
        select {
        case <-ctx.Done():
            fmt.Printf("Worker %d cancelled\n", id)
            return
        case <-time.After(time.Second): // 模拟耗时操作
            fmt.Printf("Worker %d working\n", id)
        }
    }
}

func main() {
    var wg sync.WaitGroup
    ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
    defer cancel()

    for i := 1; i <= 5; i++ {
        wg.Add(1)
        go workerWithContext(ctx, i, &wg)
    }

    wg.Wait()
    fmt.Println("All workers done")
}

在这个例子中,我们使用context.WithTimeout创建了一个带有超时时间的context。每个worker goroutine都会监听context的Done channel,一旦context被取消,goroutine就会退出。这样,我们就可以在一定时间内等待goroutine执行完毕,或者在超时后强制取消它们。

除了sync.WaitGroup,还有哪些Golang并发控制工具值得学习?

Golang提供了丰富的并发控制工具,除了sync.WaitGroup,还有sync.Mutexsync.RWMutexsync.Condsync.Oncechan等。

  • sync.Mutex用于保护共享资源,防止并发访问导致的数据竞争。
  • sync.RWMutex是读写锁,允许多个goroutine同时读取共享资源,但只允许一个goroutine写入共享资源。
  • sync.Cond用于goroutine之间的条件同步,允许goroutine等待某个条件成立后再继续执行。
  • sync.Once用于确保某个函数只执行一次,常用于单例模式的实现。
  • chan是goroutine之间通信的管道,可以用于传递数据和同步信号。

选择合适的并发控制工具,取决于具体的应用场景。理解这些工具的原理和用法,可以帮助我们写出更高效、更健壮的并发程序。

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

178

2024.02.23

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

226

2024.02.23

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

337

2024.02.23

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

208

2024.03.05

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

388

2024.05.21

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

194

2025.06.09

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

189

2025.06.10

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

192

2025.06.17

php与html混编教程大全
php与html混编教程大全

本专题整合了php和html混编相关教程,阅读专题下面的文章了解更多详细内容。

0

2026.01.13

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
golang socket 编程
golang socket 编程

共2课时 | 0.1万人学习

nginx浅谈
nginx浅谈

共15课时 | 0.8万人学习

golang和swoole核心底层分析
golang和swoole核心底层分析

共3课时 | 0.1万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号