Golang 学习手册
  • 主页
  • 安装
  • 参考资料
  • Q&A
  • 基础
    • Hello,world!
    • 包
    • 变量
    • 常量
    • 类型
      • 整型
      • 浮点型
      • 复数
      • 布尔型
      • 字符
      • 字符串
      • 数组
      • 切片
      • 字典
      • 指针
      • 方法
      • 结构体
      • 接口
      • 自定义类型
    • 语法
      • 判断
      • 循环
      • 错误处理
    • 协程
      • 锁
      • 等待组
      • 通道
      • 上下文
    • 测试与分析
      • 单元测试
      • 性能测试
      • 性能分析
    • 编译
      • 条件编译
      • 交叉编译
  • 高级
    • 存储
      • 键值存储
        • Etcd
    • HTTP
      • http服务基础
      • Websocket
      • 开源框架
        • Beego
        • Echo
        • Gin
        • Iris
        • Revel
        • 框架对比
      • Kubernetes风格API框架
    • RPC
      • RPCX
      • GRPC
      • grpc-gateway
    • 连接分发器
    • TLS加密
    • 链路追踪
      • skywalking
  • 项目管理
    • 代码规范
    • 包管理
    • 文档
      • GoDoc
      • Markdown
      • Swagger
    • 仓库管理
      • 分支管理
      • 问题管理
      • 里程碑管理
      • 发布管理
    • 持续集成
      • CircleCI
      • TravisCI
由 GitBook 提供支持
在本页

这有帮助吗?

在Git上编辑
  1. 基础
  2. 协程

等待组

通过time.Sleep()这种方式去等待协程执行完毕是不可靠的,我们大多数情况下无法估计协程的运行时长,需要有一个能够准确获取到协程结束运行的方法。

在内置包sync中有一个WaitGroup类型可以应对这种场景,其通过锁机制实现。其原理是在WaitGroup中有一个带锁的计数器,通过其Add()方法可以为计数器累加值;还有一个Done()方法,用于释放释放计数器值(减1);还有一个Wait()方法,该方法会一直阻塞直到计数器归零。

于是就可以这么做,在协程执行前为WaitGroup计数器加1,接着把WaitGroup做为参数传入并运行协程,在协程终止时执行WaitGroup的Done()方法释放计数器。在主方法体中执行WaitGroup的Wait()方法阻塞直到WaitGroup的计数器归零。这样只有当所有协程都执行结束时,计数器才会归零,主方法体也才能解除阻塞状态继续往下运行。

例子:使用WaitGroup等待协程退出

package main

import (
	"fmt"
	"sync"
)

func myFunc(waitgroup *sync.WaitGroup) {
	fmt.Println("Inside my goroutine")
	// 释放一个计数器
	waitgroup.Done()
}

func main() {
	fmt.Println("Hello World")

	// 实例化一个等待组
	var waitgroup sync.WaitGroup

	// 等待组计数器加1
	waitgroup.Add(1)

	// 执行将等待组做为参数传入并执行goroutine
	go myFunc(&waitgroup)

	// 阻塞直到计数器清零
	waitgroup.Wait()

	fmt.Println("Finished Execution")
}

以上代码的执行结果:

Hello World
Inside my goroutine
Finished Execution
上一页锁下一页通道

最后更新于4年前

这有帮助吗?