當所有工作完成後，如何關閉由多個 goroutine 填充的通道？

當所有工作完成後，如何關閉由多個goroutine填滿的通道是一個經常被提及的問題。在Go語言中，關閉一個通道是一種通知接收方沒有更多資料的方式。透過關閉通道，接收方可以及時得知發送方已經完成了所有的發送操作。在多個goroutine填充的通道中，我們可以使用一個計數器來追蹤還有幾個goroutine在向通道發送資料。當計數器減為0時，表示所有的工作都已完成，此時我們可以安全地關閉通道。關閉通道後，接收方可以透過在接收表達式中使用額外的變數來判斷通道是否已關閉。透過這種方式，我們可以保證在所有工作完成後正確關閉由多個goroutine填充的通道，從而避免資源洩漏和死鎖的問題。

問題內容

我試圖遵循「不要透過共享記憶體來通信，而是透過通信來共享記憶體」的Go方式，並使用通道來異步地傳達要完成的任務並發送回處理任務的結果。

為了簡單起見，我已將通道類型更改為 int，而不是它們真正的結構。並用 time.Sleep() 取代了長時間的處理。

如何在發回所有任務結果後關閉 ProducedResults ，以便此程式碼不會卡在最後一個 for ？

quantityOfTasks:= 100
    quantityOfWorkers:= 60
    remainingTasks := make(chan int)
    producedResults := make(chan int)

    // produce tasks
    go func() {
        for i := 0; i < quantityOfTasks; i++ {
            remainingTasks <- 1
        }
        close(remainingTasks)
    }()

    // produce workers
    for i := 0; i < quantityOfWorkers; i++ {
        go func() {
            for taskSize := range remainingTasks {
                // simulate a long task
                time.Sleep(time.Second * time.Duration(taskSize))
                // return the result of the long task
                producedResults <- taskSize
            }
        }()
    }

    // read the results of the tasks and agregate them
    executedTasks := 0
    for resultOfTheTask := range producedResults { //this loop will never finish because producedResults never gets closed
        // consolidate the results of the tasks
        executedTasks += resultOfTheTask
    }

解決方法

您希望在寫入該通道的所有 goroutine 返回後關閉該通道。您可以使用 WaitGroup 來實現：

wg:=sync.WaitGroup{}

for i := 0; i < quantityOfWorkers; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            for taskSize := range remainingTasks {
                //simulate a long task
                time.Sleep(time.Second * time.Duration(taskSize))
                //return the result of the long task
                producedResults <- taskSize
            }
        }()
}

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

以上是當所有工作完成後，如何關閉由多個 goroutine 填充的通道？的詳細內容。更多資訊請關注PHP中文網其他相關文章！