温馨提示×

Go语言条件变量如何处理通知

小樊
81
2024-11-02 16:43:31
栏目: 编程语言

在Go语言中,条件变量通常使用sync.Cond类型来实现。sync.Cond是一个条件变量,它允许一组goroutine等待某个条件成立。当条件发生变化时,可以使用Signal()Broadcast()方法来通知等待的goroutine。

下面是一个简单的示例,展示了如何使用sync.Cond处理通知:

package main

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

func main() {
	var mu sync.Mutex
	cond := sync.NewCond(&mu)
	data := 0

	// 生产者goroutine
	go func() {
		for i := 0; i < 10; i++ {
			mu.Lock()
			for data <= i {
				cond.Wait() // 等待条件成立
			}
			fmt.Printf("Producer: Data %d is ready\n", i)
			data = i + 1
			cond.Signal() // 通知消费者goroutine
			mu.Unlock()
			time.Sleep(1 * time.Second)
		}
	}()

	// 消费者goroutine
	go func() {
		for i := 0; i < 10; i++ {
			mu.Lock()
			for data > i {
				cond.Wait() // 等待条件成立
			}
			fmt.Printf("Consumer: Data %d is consumed\n", i)
			mu.Unlock()
			time.Sleep(1 * time.Second)
		}
	}()

	time.Sleep(10 * time.Second)
}

在这个示例中,我们创建了一个sync.Cond类型的条件变量cond,以及一个互斥锁mu。生产者goroutine在数据大于等于某个值时,通过cond.Signal()通知消费者goroutine。消费者goroutine在数据小于某个值时,通过cond.Wait()等待条件成立。当条件成立时,消费者goroutine会被唤醒并处理数据。

注意:在使用cond.Wait()时,需要确保已经持有互斥锁mu,否则会导致运行时错误。在唤醒等待的goroutine之前,也需要释放互斥锁,以便其他goroutine可以获取锁并执行相应的操作。

0