Go語言條件變量如何使用

在Go語言中，條件變量通常通過sync.Cond類型來實(shí)現(xiàn)。sync.Cond是一個(gè)條件變量，它允許你在多個(gè)goroutine之間同步訪問共享資源。條件變量通常與互斥鎖（sync.Mutex）一起使用，以確保在檢查條件和等待通知時(shí)不會(huì)發(fā)生競態(tài)條件。

下面是一個(gè)簡單的示例，展示了如何使用sync.Cond來實(shí)現(xiàn)一個(gè)生產(chǎn)者-消費(fèi)者模型：

package main

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

type Buffer struct {
	data []int
	mu   sync.Mutex
	cond *sync.Cond
}

func NewBuffer(size int) *Buffer {
	return &Buffer{
		data: make([]int, size),
		cond: sync.NewCond(&sync.Mutex{}),
	}
}

func (b *Buffer) Produce(value int) {
	b.mu.Lock()
	defer b.mu.Unlock()

	for len(b.data) >= cap(b.data) {
		b.cond.Wait() // 等待條件滿足
	}

	b.data = append(b.data, value)
	fmt.Println("Produced:", value)
	b.cond.Signal() // 通知消費(fèi)者可以消費(fèi)數(shù)據(jù)
}

func (b *Buffer) Consume() int {
	b.mu.Lock()
	defer b.mu.Unlock()

	for len(b.data) == 0 {
		b.cond.Wait() // 等待條件滿足
	}

	value := b.data[0]
	b.data = b.data[1:]
	fmt.Println("Consumed:", value)
	b.cond.Signal() // 通知生產(chǎn)者可以生產(chǎn)數(shù)據(jù)

	return value
}

func main() {
	buffer := NewBuffer(3)
	var wg sync.WaitGroup

	wg.Add(2)
	go func() {
		defer wg.Done()
		for i := 0; i < 5; i++ {
			buffer.Produce(i)
			time.Sleep(1 * time.Second)
		}
	}()

	go func() {
		defer wg.Done()
		for i := 0; i < 5; i++ {
			buffer.Consume()
			time.Sleep(1 * time.Second)
		}
	}()

	wg.Wait()
}

在這個(gè)示例中，我們創(chuàng)建了一個(gè)Buffer結(jié)構(gòu)體，它包含一個(gè)整數(shù)切片、一個(gè)互斥鎖和一個(gè)條件變量。Produce方法用于向緩沖區(qū)添加數(shù)據(jù)，而Consume方法用于從緩沖區(qū)中獲取數(shù)據(jù)。在Produce和Consume方法中，我們使用sync.Cond的Wait方法來等待條件滿足，然后使用Signal方法來通知其他goroutine條件已滿足。

注意，在使用條件變量時(shí)，務(wù)必確保在調(diào)用Wait方法之前已經(jīng)鎖定互斥鎖，并在喚醒其他goroutine之后解鎖互斥鎖。這樣可以避免死鎖和競態(tài)條件。