温馨提示×

Go语言条件变量怎样实现等待

小樊
82
2024-11-02 16:42:32
栏目: 编程语言

在Go语言中,我们可以使用sync.Cond来实现条件变量的等待功能。sync.Cond是一个条件变量,它可以帮助我们在满足某个条件时阻塞当前goroutine,直到其他goroutine通知它条件已经满足。

下面是一个简单的示例,展示了如何使用sync.Cond实现等待功能:

package main

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

type Example struct {
	cond *sync.Cond
	data int
}

func NewExample() *Example {
	return &Example{
		cond: sync.NewCond(&sync.Mutex{}),
	}
}

func (e *Example) WaitUntilDataGreaterThan(target int) {
	e.cond.L.Lock() // 加锁
	for e.data <= target {
		e.cond.Wait() // 等待条件满足
	}
	e.cond.L.Unlock() // 解锁
}

func (e *Example) SetData(value int) {
	e.cond.L.Lock() // 加锁
	e.data = value
	e.cond.Signal() // 通知等待的goroutine条件已经满足
	e.cond.L.Unlock() // 解锁
}

func main() {
	example := NewExample()

	go func() {
		time.Sleep(2 * time.Second) // 模拟一些工作
		example.SetData(5)          // 设置数据为5
	}()

	example.WaitUntilDataGreaterThan(4) // 等待数据大于4
	fmt.Println("Data is greater than 4")
}

在这个示例中,我们创建了一个名为Example的结构体,它包含一个sync.Cond类型的字段cond和一个整数字段dataWaitUntilDataGreaterThan方法用于等待数据大于指定值,SetData方法用于设置数据并通知等待的goroutine条件已经满足。

main函数中,我们创建了一个Example实例,并启动一个goroutine来模拟一些工作。在这个goroutine中,我们等待2秒后设置数据为5。在主goroutine中,我们调用WaitUntilDataGreaterThan方法等待数据大于4。当数据被设置为5时,条件满足,WaitUntilDataGreaterThan方法返回,程序输出"Data is greater than 4"。

0