當生產者壽命較短時，如何避免 Go Goroutine 死鎖？-Golang-PHP中文網

當生產者壽命較短時，如何避免 Go Goroutine 死鎖？

Linda Hamilton

發布： 2024-10-25 07:09:02

原創

896 人瀏覽過

How to Avoid Deadlock in Go Goroutines When Producers Are Short-Lived?

解決Go Goroutine 中的死鎖

在並發程式設計中，當多個Goroutine 無限期地等待彼此完成操作時，就會發生死鎖，從而有效地停止進程程序。本文解決了Go 並發中遇到的特定死鎖，如原始問題中所述：

<code class="go">package main

import (
    "fmt"
    "time"
)

func producer(ch chan int, d time.Duration, num int) {
    for i := 0; i < num; i++ {
        ch <- i
        time.Sleep(d)
    }
}

func main() {
    ch := make(chan int)
    go producer(ch, 100*time.Millisecond, 2)
    go producer(ch, 200*time.Millisecond, 5)
    for {
        fmt.Println(<-ch)
    }
    close(ch)
}</code>

登入後複製

此程式碼由於以下因素觸發死鎖錯誤：

生產者向通道發送值的過程是短暫的，最終會停止產生資料。
主函數中的無限 for 迴圈不斷從通道接收值，沒有終止條件。
無限循環後通道關閉，導致沒有更多的值可用於接收。

解決方案：協調終止

為了避免死鎖，生產者必須協調以發出完成信號，並且通道必須由最後一個生產者關閉。這是使用sync.WaitGroup進行協調的有效解決方案：

<code class="go">func producer(ch chan int, d time.Duration, num int, wg *sync.WaitGroup) {
    defer wg.Done()
    for i := 0; i < num; i++ {
        ch <- i
        time.Sleep(d)
    }
}

func main() {
    wg := &sync.WaitGroup{}
    ch := make(chan int)

    wg.Add(1)
    go producer(ch, 100*time.Millisecond, 2, wg)
    wg.Add(1)
    go producer(ch, 200*time.Millisecond, 5, wg)

    go func() {
        wg.Wait()
        close(ch)
    }()

    for v := range ch {
        fmt.Println(v)
    }
}</code>

登入後複製

在此解決方案中：