go cond

redrobot發表於2024-06-07

cond:

package main

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

func main() {
    var mu sync.Mutex
    cond := sync.NewCond(&mu)
    var ready bool

    go func() {
        time.Sleep(time.Second) // 模擬一些工作
        mu.Lock()
        fmt.Println("i am lock routine")
        ready = true
        cond.Signal() // 通知一個等待的協程
        mu.Unlock()
    }()

    mu.Lock()
    fmt.Println("i am lock")
    for !ready {
        cond.Wait() // 阻塞直到條件滿足
    }
    fmt.Println("Condition met, continuing...")
    mu.Unlock()
}


output:
i am lock
i am lock routine
Condition met, continuing...

Program exited.

取消cond

package main

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

func main() {
    var mu sync.Mutex
    // cond := sync.NewCond(&mu)
    // var ready bool

    go func() {
        time.Sleep(time.Second) // 模擬一些工作
        mu.Lock()
        fmt.Println("i am lock routine")
        // ready = true
        // cond.Signal() // 通知一個等待的協程
        mu.Unlock()
    }()

    mu.Lock()
    fmt.Println("i am lock")
    time.Sleep(time.Second * 5)
    //for !ready {
    //cond.Wait() // 阻塞直到條件滿足
    //}
    fmt.Println("Condition met, continuing...")
    mu.Unlock()
}

output:
i am lock
Condition met, continuing...
i am lock routine

Program exited.

來自：

深入理解Go語言中的sync.Cond

1. 簡介

本文將介紹 Go 語言中的 sync.Cond併發原語，包括 sync.Cond的基本使用方法、實現原理、使用注意事項以及常見的使用使用場景。能夠更好地理解和應用 Cond 來實現 goroutine 之間的同步。

2. 基本使用

2.1 定義

sync.Cond是Go語言標準庫中的一個型別，代表條件變數。條件變數是用於多個goroutine之間進行同步和互斥的一種機制。sync.Cond可以用於等待和通知goroutine，以便它們可以在特定條件下等待或繼續執行。

2.2 方法說明

sync.Cond的定義如下，提供了Wait ,Singal,Broadcast以及NewCond方法

type Cond struct {
   noCopy noCopy
   // L is held while observing or changing the condition
   L Locker

   notify  notifyList
   checker copyChecker
}

func NewCond(l Locker) *Cond {}
func (c *Cond) Wait() {}
func (c *Cond) Signal() {}
func (c *Cond) Broadcast() {}

NewCond方法：提供建立Cond例項的方法
Wait方法: 使當前執行緒進入阻塞狀態，等待其他協程喚醒
Singal方法: 喚醒一個等待該條件變數的執行緒，如果沒有執行緒在等待，則該方法會立即返回。
Broadcast方法: 喚醒所有等待該條件變數的執行緒，如果沒有執行緒在等待，則該方法會立即返回。

2.3 使用方式

當使用sync.Cond時，通常需要以下幾個步驟：

定義一個互斥鎖，用於保護共享資料；
建立一個sync.Cond物件，關聯這個互斥鎖；
在需要等待條件變數的地方，獲取這個互斥鎖，並使用Wait方法等待條件變數被通知；
在需要通知等待的協程時，使用Signal或Broadcast方法通知等待的協程。
最後，釋放這個互斥鎖。

下面是一個簡單的程式碼的示例，展示了大概的程式碼結構:

var (
    // 1. 定義一個互斥鎖
    mu    sync.Mutex
    cond  *sync.Cond
    count int
)
func init() {
    // 2.將互斥鎖和sync.Cond進行關聯
    cond = sync.NewCond(&mu)
}
go func(){
    // 3. 在需要等待的地方,獲取互斥鎖，呼叫Wait方法等待被通知
    mu.Lock()
    // 這裡會不斷迴圈判斷 是否滿足條件
    for !condition() {
       cond.Wait() // 等待任務
    }
    mu.Unlock()
}

go func(){
     // 執行業務邏輯
     // 4. 滿足條件，此時呼叫Broadcast喚醒處於等待狀態的協程
     cond.Broadcast() 
}

2.4 使用例子

下面透過描述net/http中的 connReader，來展示使用sync.Cond實現阻塞等待通知的機制。這裡我們不需要理解太多，只需要知道connReader下面兩個方法:

func (cr *connReader) Read(p []byte) (n int, err error) {}
func (cr *connReader) abortPendingRead() {}

Read方法則是用於從HTTP連線中讀取資料，不允許併發訪問的。而abortPendingRead則是用於終止正在讀取的連線。

從abortPendingRead方法的語意來看，是需要成功終止其他協程進行資料的讀取之後，才能正常返回，也就是此時沒有協程再繼續讀取資料了，才可以返回。

那abortPendingRead如何得知是否還有協程在讀取資料呢，其實是可以透過定時輪訓connReader的狀態，從而判斷當前Read方法是否仍在讀取資料。但是定時輪訓效率太低，可能會造成cpu的大量空轉。更好的方式，應該是讓協程進入阻塞狀態，然後等條件滿足了，其他協程再來喚醒當前協程，然後再繼續執行下去。

這個其實就是sync.Cond設計的用途，當不滿足執行條件時，先進入阻塞狀態，等待條件滿足時，再由其他協程來喚醒，然後再繼續執行下去，能夠提高程式的執行效率。其中Wait方法便是讓協程進入阻塞狀態，而Singal和Boardcast便是喚醒處於阻塞狀態的協程，告知其條件滿足了，可以繼續向下執行了。

回到我們connReader的例子，我們使用sync.Cond實現阻塞等待通知的效果。

type connReader struct {
    // 是否正在讀取資料
    inRead bool
    mu      sync.Mutex // guards following
    cond    *sync.Cond
}

func (cr *connReader) abortPendingRead() {
    if !cr.inRead{
        return
    }
    //1. 透過一定手段,讓Read方法中斷
    cr.mu.Lock()
    // 判斷Read方法是否仍然在讀取資料
    for cr.inRead {
        //2. 此時Read方法仍然在讀取資料, 不滿足條件，等待通知
        cr.cond.Wait()
    }
    cr.mu.Unlock()
}

func (cr *connReader) Read(p []byte) (n int, err error) {
     cr.mu.Lock()
     cr.inRead = true
    // 1. 讀取資料
    // 2. abortPendingRead透過某種手段,讓Read方法中斷
    
    cr.inRead = false
    cr.mu.Unlock()
    // 3. 現在已經滿足abortPendingRead繼續執行下去的條件了，可以喚醒abortPendingRead協程了
    cond.Boardcast()
}

這裡abortPendingRead方法首先判斷是否還在讀取資料，是的話，呼叫Wait方法進入阻塞狀態，等待條件滿足後繼續執行。

對於Read方法，因為其不執行併發訪問，當其將退出時，說明此時已經沒有協程在讀取資料了，滿足abortPendingRead繼續執行下去的條件了，此時可以呼叫Boardcast來喚醒等待條件滿足的協程。之後呼叫abortPendingRead方法的協程此時能夠接收到通知，便能夠順利被喚醒，從而正確返回。

這裡便展示了一個簡單的，使用sync.Cond實現阻塞等待通知的例子。

3. 原理

3.1 基本原理

在Sync.Cond存在一個通知佇列，儲存了所有處於等待狀態的協程。通知佇列定義如下:

type notifyList struct {
   wait   uint32
   notify uint32
   lock   uintptr // key field of the mutex
   head   unsafe.Pointer
   tail   unsafe.Pointer
}

當呼叫Wait方法時，此時Wait方法會釋放所持有的鎖，然後將自己放到notifyList等待佇列中等待。此時會將當前協程加入到等待佇列的尾部，然後進入阻塞狀態。

當呼叫Signal 時，此時會喚醒等待佇列中的第一個協程，其他繼續等待。如果此時沒有處於等待狀態的協程，呼叫Signal不會有其他作用，直接返回。當呼叫BoradCast方法時，則會喚醒notfiyList中所有處於等待狀態的協程。

sync.Cond的程式碼實現比較簡單，協程的喚醒和阻塞已經由執行時包實現了，sync.Cond的實現直接呼叫了執行時包提供的API。

3.2 實現

3.2.1 Wait方法實現

Wait方法首先呼叫runtime_notifyListAd方法，將自己加入到等待佇列中，然後釋放鎖，等待其他協程的喚醒。

func (c *Cond) Wait() {
   // 將自己放到等待佇列中
   t := runtime_notifyListAdd(&c.notify)
   // 釋放鎖
   c.L.Unlock()
   // 等待喚醒
   runtime_notifyListWait(&c.notify, t)
   // 重新獲取鎖
   c.L.Lock()
}

3.2.2 Singal方法實現

Singal方法呼叫runtime_notifyListNotifyOne喚醒等待佇列中的一個協程。

func (c *Cond) Signal() {
   // 喚醒等待佇列中的一個協程
   runtime_notifyListNotifyOne(&c.notify)
}

3.2.3 Broadcast方法實現

Broadcast方法呼叫runtime_notifyListNotifyAll喚醒所有處於等待狀態的協程。

func (c *Cond) Broadcast() {
   // 喚醒等待佇列中所有的協程
   runtime_notifyListNotifyAll(&c.notify)
}

4.使用注意事項

4.1 呼叫Wait方法前未加鎖

4.1.1 問題

如果在呼叫Wait方法前未加鎖，此時會直接panic，下面是一個簡單例子的說明:

package main

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

var (
   count int
   cond  *sync.Cond
   lk    sync.Mutex
)

func main() {
    cond = sync.NewCond(&lk)
    wg := sync.WaitGroup{}
    wg.Add(2)
    go func() {
       defer wg.Done()
       for {
          time.Sleep(time.Second)
          count++
          cond.Broadcast()
       }
    }()
    
    go func() {
       defer wg.Done()
       for {
          time.Sleep(time.Millisecond * 500)          
          //cond.L.Lock() 
          for count%10 != 0 {
               cond.Wait()
          }
          t.Logf("count = %d", count)
          //cond.L.Unlock()  
       }
    }()
    wg.Wait()
}

上面程式碼中，協程一每隔1s，將count欄位的值自增1，然後喚醒所有處於等待狀態的協程。協程二執行的條件為count的值為10的倍數，此時滿足執行條件，喚醒後將會繼續往下執行。

但是這裡在呼叫sync.Wait方法前，沒有先獲取鎖，下面是其執行結果，會丟擲 fatal error: sync: unlock of unlocked mutex 錯誤，結果如下:

count = 0
fatal error: sync: unlock of unlocked mutex

因此，在呼叫Wait方法前，需要先獲取到與sync.Cond關聯的鎖，否則會直接丟擲異常。

4.1.2 為什麼呼叫Wait方法前需要先獲取該鎖

強制呼叫Wait方法前需要先獲取該鎖。這裡的原因在於呼叫Wait方法如果不加鎖，有可能會出現競態條件。

這裡假設多個協程都處於等待狀態，然後一個協程呼叫了Broadcast喚醒了其中一個或多個協程，此時這些協程都會被喚醒。

如下，假設呼叫Wait方法前沒有加鎖的話，那麼所有協程都會去呼叫condition方法去判斷是否滿足條件，然後都透過驗證，執行後續操作。

for !condition() {
    c.Wait()
}
c.L.Lock()
// 滿足條件情況下,執行的邏輯
c.L.Unlock()

此時會出現的情況為，本來是需要在滿足condition方法的前提下，才能執行的操作。現在有可能的效果，為前面一部分協程執行時，還是滿足condition條件的；但是後面的協程，儘管不滿足condition條件，還是執行了後續操作，可能導致程式出錯。

正常的用法應該是，在呼叫Wait方法前便加鎖，只會有一個協程判斷是否滿足condition條件，然後執行後續操作。這樣子就不會出現即使不滿足條件，也會執行後續操作的情況出現。

c.L.Lock()
for !condition() {
    c.Wait()
}
// 滿足條件情況下,執行的邏輯
c.L.Unlock()

4.2 Wait方法接收到通知後，未重新檢查條件變數

呼叫sync.Wait方法，協程進入阻塞狀態後被喚醒，沒有重新檢查條件變數，此時有可能仍然處於不滿足條件變數的場景下。然後直接執行後續操作，有可能會導致程式出錯。下面舉一個簡單的例子:

package main

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

var (
   count int
   cond  *sync.Cond
   lk    sync.Mutex
)

func main() {
    cond = sync.NewCond(&lk)
    wg := sync.WaitGroup{}
    wg.Add(3)
    go func() {
       defer wg.Done()
       for {
          time.Sleep(time.Second)
          cond.L.Lock()
          // 將flag 設定為true
          flag = true
          // 喚醒所有處於等待狀態的協程
          cond.Broadcast()
          cond.L.Unlock()
       }
    }()
    
    for i := 0; i < 2; i++ {
       go func(i int) {
          defer wg.Done()
          for {
             time.Sleep(time.Millisecond * 500)
             cond.L.Lock()
             // 不滿足條件，此時進入等待狀態
             if !flag {
                cond.Wait()
             }
             // 被喚醒後，此時可能仍然不滿足條件
             fmt.Printf("協程 %d flag = %t", i, flag)
             flag = false
             cond.L.Unlock()
          }
       }(i)
    }
    wg.Wait()
}

在這個例子，我們啟動了一個協程，定時將flag設定為true，相當於每隔一段時間，便滿足執行條件，然後喚醒所有處於等待狀態的協程。

然後又啟動了兩個協程，在滿足條件的前提下，開始執行後續操作，但是這裡協程被喚醒後，沒有重新檢查條件變數，具體看第39行。這裡會出現的場景是，第一個協程被喚醒後，此時執行後續操作，然後將flag重新設定為false，此時已經不滿足條件了。之後第二個協程喚醒後，獲取到鎖，沒有重新檢查此時是否滿足執行條件，直接向下執行，這個就和我們預期不符，可能會導致程式出錯，程式碼執行效果如下:

協程 1 flag = true
協程 0 flag = false
協程 1 flag = true
協程 0 flag = false

可以看到，此時協程0執行時，flag的值均為false,說明此時其實並不符合執行條件，可能會導致程式出錯。因此正確用法應該像下面這樣子，被喚醒後，需要重新檢查條件變數，滿足條件之後才能繼續向下執行。

c.L.Lock()
// 喚醒後,重新檢查條件變數是否滿足條件
for !condition() {
    c.Wait()
}
// 滿足條件情況下,執行的邏輯
c.L.Unlock()

5.總結

本文介紹了 Go 語言中的 sync.Cond 併發原語，它是用於實現 goroutine 之間的同步的重要工具。我們首先學習了 sync.Cond 的基本使用方法，包括建立和使用條件變數、使用Wait和Signal/Broadcast方法等。

在接下來的部分中，我們介紹了 sync.Cond 的實現原理，主要是對等待佇列的使用，從而sync.Cond有更好的理解，能夠更好得使用它。同時，我們也講述了使用sync.Cond的注意事項，如呼叫Wait方法前需要加鎖等。

基於以上內容，本文完成了對 sync.Cond 的介紹，希望能夠幫助大家更好地理解和使用Go語言中的併發原語。

來自：

https://www.cnblogs.com/chenjiazhan/p/17238021.html

Go語言—sync.Cond原始碼分析
2019-12-30
Go原始碼
這一次，徹底搞懂 Go Cond
2021-04-23
Go
一文讀懂 Go sync.Cond 設計
2021-10-28
Go
Go中協程間通訊的方式Sync.Cond
2021-06-15
Go
執行緒中的條件變數pthread_cond_wait、pthread_cond_signal
2011-02-11
執行緒變數threadAI
pthread_cond_wait()函式的理解
2011-02-11
threadAI函式
隨筆：MYSQL_BIN_LOG::COND_done 是什麼
2019-08-16
MySql
秒懂 Golang 中的條件變數（sync.Cond）
2022-12-14
Golang變數
【C/C++】pthread_cond_wait與mutex結合的原因
2019-08-10
C++threadAIMutex
統計Go, Go, Go
2020-01-16
Go
Go 學習、Go 進階、Go 實用工具類
2019-12-11
Go
【Go】go get 自動代理
2019-03-22
Go
Go_go語言初探
2020-12-14
Go
Go框架解析：go-zero
2022-03-30
Go框架
go學習--->安裝go
2022-02-22
Go
Ready！GO PM GO 1.3
2013-07-16
Go
Ready！GO PM GO 1.2
2013-06-28
Go
Ready！GO PM GO 1.1
2013-06-18
Go
Ready！GO PM GO 1.0
2013-06-15
Go
如何用Go構建Go
2013-06-08
Go
Go for/if
2022-01-13
Go
【GO】
2014-03-17
Go
go
2012-09-07
Go
《Go輕鬆學》、《Go示例學》和《Go入門指南》
2016-10-12
Go
Go 模組--開始使用 Go Modules
2020-01-05
Go
Go學習總結-go module
2019-05-05
Go
學習記錄—go—go test
2019-01-23
Go
go reflect包中abi.go
2022-02-17
Go
【GO】《GO HANDBOOK》讀書筆記
2024-02-20
Go筆記
go安裝playwright-go
2024-07-31
Go
Go-ethereum 原始碼解析之 go-ethereum/ethdb/database.go
2018-10-25
Go原始碼Database
Go 每日一庫之 go-flags
2020-01-13
Go
Go 切片 slice - Go 學習記錄
2019-03-04
Go
【GO】MAC安裝和測試Go
2019-02-16
GoMac
重新理解 Go 培訓和 Go 人才
2021-04-15
Go
Go 快速指南：go1.18 特性
2021-12-14
Go
六、GO 程式設計模式：GO GENERATION
2022-02-08
Go程式設計設計模式
go modules
2020-06-11
Go