条件变量的使用(golang)

1、背景

最近在学习go的一个开源协程池,在源码中有用到锁、信号量,锁相对来说用的是比较多的,信号量相对用的较少,之前研究学习过c++的std::condition_variable,其实和golang的大同小异,个人感觉c++的略强大一些,其wait有两个重载,第二个语法糖的实现主要是为了防止虚假唤醒。

2、介绍

条件变量的使用场景是,当满足某个条件时,触发某个动作,类似于排队打饭或者挂号看病,当医生或者打饭员准备好后,就触发看病或者打饭的动作。

条件变量有三个函数:

Go 复制代码
con.Wait()

该函数会将当前协程放入等待队列并阻塞,直到Signal或Broadcast方法将协程从等待队列中移除并唤醒。

Go 复制代码
con.Signal()

唤醒等待队列中的一个子协程,先唤醒最先阻塞的子协程,被唤醒之后,子协程继续执行。

Go 复制代码
con.Broadcast()

Broadcast功能类似Signal,不同的是signal只唤醒一个子协程,而broadcast是唤醒所有阻塞的子协程。

3、使用示例

3.1 signal

我们就以在食堂排队打饭为例子,假如本次排队有10位同学排队,这10位同学分别编号为0-9号(就不具体起名字了),当打饭师傅准备好后,就开始排队打饭,示例代码如下:

Go 复制代码
package main

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

func DaFan(x int, num *int, wg *sync.WaitGroup, cond *sync.Cond) {
	defer wg.Done()
	// 当有一个lock成功后,其他协程阻塞在这里
	cond.L.Lock()
	*num++
	fmt.Println("学号为:", x, "的同学排到了第:", *num, "位")
	cond.Wait()
	fmt.Println("学号为:", x, "的同学打饭了")
	cond.L.Unlock()
}
func main() {
	var wg sync.WaitGroup
	num := 0
	var lock sync.Mutex
	con := sync.NewCond(&lock)
	for i := 0; i < 10; i++ {
		wg.Add(1)
		go DaFan(i, &num, &wg, con)
	}
	// 模拟打饭准备
	time.Sleep(5 * time.Second)
	// 循环通知这10位同学打饭
	for i := 0; i < 10; i++ {
		con.Signal()
		// 打饭耗时1秒
		time.Sleep(1 * time.Second)
	}
	wg.Wait()
}

上面代码运行结果如下:

3.2 broadcast

broadcast为广播,好比将军一声令下,所有士兵齐刷刷的冲向敌阵,broadcast的应用场景如下:

将军下令,要求全部将士在10s内修整完成,15s后发起总攻,假如我们有十个兵团,分别为1-10号:

Go 复制代码
package main

import (
	"fmt"
	"math/rand"
	"sync"
	"time"
)

func GeneralAttack(x int, num *int, wg *sync.WaitGroup, cond *sync.Cond) {
	defer wg.Done()
	// 随机一个10秒内的时间用于修整
	time.Sleep(time.Second * time.Duration(rand.Intn(10)))
	fmt.Println("兵团", x+1, "在", time.Now().Format("2006-01-02 15:04:05"), "准备完毕,等待总攻")
	cond.L.Lock()
	// 准备完毕,等待教练发令
	cond.Wait()
	cond.L.Unlock()
	fmt.Println("兵团", x+1, "开始发起总攻", time.Now().Format("2006-01-02 15:04:05"))
}
func main() {
	var wg sync.WaitGroup
	num := 0
	var lock sync.Mutex
	con := sync.NewCond(&lock)
	fmt.Println("将军下令,开始修整:", time.Now().Format("2006-01-02 15:04:05"), ",15s后开始总攻")
	for i := 0; i < 10; i++ {
		wg.Add(1)
		go GeneralAttack(i, &num, &wg, con)
	}
	// 等待发起总攻
	time.Sleep(15 * time.Second)
	fmt.Println("将军下令,开始总攻:", time.Now().Format("2006-01-02 15:04:05"))
	// 发起总攻
	con.Broadcast()
	wg.Wait()
}

运行记录:

Go 复制代码
liupeng@liupengdeMacBook-Pro test % go run main.go
将军下令,开始修整: 2024-04-06 21:04:32 ,15s后开始总攻
兵团 2 在 2024-04-06 21:04:32 准备完毕,等待总攻
兵团 1 在 2024-04-06 21:04:32 准备完毕,等待总攻
兵团 4 在 2024-04-06 21:04:33 准备完毕,等待总攻
兵团 9 在 2024-04-06 21:04:33 准备完毕,等待总攻
兵团 6 在 2024-04-06 21:04:37 准备完毕,等待总攻
兵团 3 在 2024-04-06 21:04:38 准备完毕,等待总攻
兵团 8 在 2024-04-06 21:04:39 准备完毕,等待总攻
兵团 10 在 2024-04-06 21:04:39 准备完毕,等待总攻
兵团 5 在 2024-04-06 21:04:40 准备完毕,等待总攻
兵团 7 在 2024-04-06 21:04:41 准备完毕,等待总攻
将军下令,开始总攻: 2024-04-06 21:04:47
兵团 6 开始发起总攻 2024-04-06 21:04:47
兵团 7 开始发起总攻 2024-04-06 21:04:47
兵团 2 开始发起总攻 2024-04-06 21:04:47
兵团 3 开始发起总攻 2024-04-06 21:04:47
兵团 8 开始发起总攻 2024-04-06 21:04:47
兵团 10 开始发起总攻 2024-04-06 21:04:47
兵团 5 开始发起总攻 2024-04-06 21:04:47
兵团 1 开始发起总攻 2024-04-06 21:04:47
兵团 9 开始发起总攻 2024-04-06 21:04:47
兵团 4 开始发起总攻 2024-04-06 21:04:47
liupeng@liupengdeMacBook-Pro test % 

如上运行记录,十个兵团在10s内准备完毕,准备完成后等待将军下令发起总攻,每个兵团就是一个协程,并行开始准备,现实情况也是如此,每个兵团准备时间不同,但都在上级要求的时间内准备完成,之后等待将军下令。

使用 条件变量,在调用 Wait() 方法之前,必须要加锁,否则代码会导致程序 panic。原因是调用 Wait 方法,会先把调用者放入等待队列中,然后释放锁。此时如果在未持有锁时调用释放锁的方法,就会导致程序 panic。

相关推荐
小爬虫程序猿1 分钟前
利用Java爬虫获取速卖通(AliExpress)商品详情的详细指南
java·开发语言·爬虫
xlsw_10 分钟前
java全栈day21--Web后端实战之利用Mybaits查询数据
java·开发语言
Murphy202325 分钟前
.net4.0 调用API(form-data)上传文件及传参
开发语言·c#·api·httpwebrequest·form-data·uploadfile·multipart/form-
我曾经是个程序员35 分钟前
C#Directory类文件夹基本操作大全
服务器·开发语言·c#
白云~️37 分钟前
uniappX 移动端单行/多行文字隐藏显示省略号
开发语言·前端·javascript
编码浪子43 分钟前
构建一个rust生产应用读书笔记7-确认邮件2
开发语言·后端·rust
昙鱼1 小时前
springboot创建web项目
java·前端·spring boot·后端·spring·maven
天之涯上上1 小时前
JAVA开发 在 Spring Boot 中集成 Swagger
java·开发语言·spring boot
2402_857583491 小时前
“协同过滤技术实战”:网上书城系统的设计与实现
java·开发语言·vue.js·科技·mfc
白宇横流学长1 小时前
基于SpringBoot的停车场管理系统设计与实现【源码+文档+部署讲解】
java·spring boot·后端