温馨提示×

Go语言条件变量如何管理状态

小樊
82
2024-11-02 16:48:30
栏目: 编程语言

在Go语言中,我们可以使用sync.Cond来实现条件变量的管理。sync.Cond是一个基于sync.Mutex的条件变量,它允许在满足特定条件时通知等待的goroutine。下面是一个简单的示例,展示了如何使用sync.Cond来管理状态:

package main

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

type State struct {
	mu        sync.Mutex
	cond      *sync.Cond
	condition bool
}

func NewState() *State {
	s := &State{
		cond: sync.NewCond(&s.mu),
	}
	return s
}

func (s *State) SetCondition(condition bool) {
	s.mu.Lock()
	defer s.mu.Unlock()
	s.condition = condition
	s.cond.Broadcast() // 通知所有等待的goroutine
}

func (s *State) WaitForCondition() {
	s.mu.Lock()
	defer s.mu.Unlock()

	for !s.condition {
		s.cond.Wait() // 等待条件满足
	}
}

func main() {
	state := NewState()

	go func() {
		state.WaitForCondition()
		fmt.Println("Condition met!")
	}()

	time.Sleep(2 * time.Second)
	state.SetCondition(true)
}

在这个示例中,我们创建了一个名为State的结构体,它包含一个互斥锁mu和一个条件变量cond。我们还定义了一个SetCondition方法来设置条件并通知所有等待的goroutine,以及一个WaitForCondition方法来等待条件满足。

main函数中,我们创建了一个State实例,并启动一个goroutine来等待条件满足。然后,我们在主线程中等待2秒后设置条件为true,这将通知等待的goroutine继续执行。

0