使用 WaitGroup 和緩衝區有限的通道時，如何避免 Go 中的死鎖？

Go 中WaitGroup 和Channel 的死鎖

在Go 中，當兩個或多個Goroutine 無限期地等待另一個Goroutine完成時，就會發生死鎖。在此範例中，我們將探討由於通道緩衝區不足和使用 WaitGroup 的 goroutine 同步不當而導致的死鎖問題。

代碼

<code class="go">package main

import "fmt"
import "sync"

func main() {
    ch := make(chan []int, 4)
    var m []int

    var wg sync.WaitGroup
    for i := 0; i < 5; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            ch <- m
            return
        }()
    }
    wg.Wait()

    for c := range ch {
        fmt.Printf("c is %v", c)
    }
}</code>

問題

代碼嘗試透過大小為 4 的緩衝通道傳送 5 個值。但是，當通道已滿時，它會阻塞，直到接收器可用為止。由於所有正在發送的 Goroutine 都已完成，因此沒有一個可以從通道接收。

此外，在通道上運行的 Goroutine（對於 c := range ch）也會無限期地保持阻塞狀態，因為它需要更多值即使沒有發送更多內容，也能到達通道。這會造成發送方和接收方互相等待的死鎖。

解決方案1：增加通道緩衝區並關閉它

避免死鎖的一種解決方案是將通道緩衝區大小增加到大於或等於發送goroutine數量的值。另外，所有發送完成後，通道應該關閉，表示不會再收到任何值。

<code class="go">ch := make(chan []int, 5)
...
wg.Wait()
close(ch)</code>

解決方案2：在接收Goroutine 中執行Done()

另一個解決方案是在接收goroutine 中而不是在main 函數中執行Done() 。透過這樣做，WaitGroup 不會遞減，直到 goroutine 接收並消耗該值：

<code class="go">func main() {
    ch := make(chan []int, 4)
    var m []int

    var wg sync.WaitGroup
    for i := 0; i < 5; i++ {
        wg.Add(1)
        go func() {
            ch <- m
            return
        }()
    }
    go func() {
        for c := range ch {
            fmt.Printf("c is %v\n", c)
            wg.Done()
        }
    }()
    wg.Wait()
}</code>

以上是使用 WaitGroup 和緩衝區有限的通道時，如何避免 Go 中的死鎖？的詳細內容。更多資訊請關注PHP中文網其他相關文章！