終止無限循環 goroutine 的安全方法？

問題內容

我有一個充當偵聽器的 goroutine。輸入流傳入某個緩衝的 channel，我希望我的 goroutine 處理傳入該通道的資料。然而，有時 channel 可能會暫時沒有資料輸入。如果 channel 一秒鐘沒有什麼可提供的，我希望我的 goroutine 做一些不同的事情。函數如下所示：

func main () {
    var wg sync.WaitGroup
    arr := make([]*myObject, 0)
    wg.Add(1)
    go listener(c, arr, &wg)
    for {
        // sending stuff to c
    }
}

func listener(c chan *myObject, arr []*myObject, wg *sync.WaitGroup) {
    for {
        select {
        case value := <- c:
            arr = append(arr, value)
        case <- time.After(1 * time.Second):
            fmt.Println(arr)
        }
}

問題是我想看到通過該通道的所有內容都列印出來。如果 main 突然結束，可能 arr 中還剩下一些東西還沒印出來，我就看不到了。所以我需要確保這個goroutine在程式結束之前處理完通道中的所有資料。我認為，這意味著我需要使用 WaitGroup 並使用 Wait() 來確保程式在我的 goroutine 完成需要執行的操作之前不會關閉。 但我不知道在我的 WaitGroup 上呼叫 Done() 的位置。

基本上，我需要一種安全的方法來在程式結束之前「暫停」goroutine，並列印出剩餘的內容。我怎樣才能做到這一點？

更糟的是，對於單元測試之類的事情，我自己將資料發送到通道，在發送一定數量後，我想查看數組。但是，如果我只是在將資料發送到通道的程式碼之後立即檢查數組，則 goroutine 可能還沒有機會處理所有資料。在這種情況下，我想等待 goroutine 處理我發送的所有數據，然後我想暫停它並要求它向我顯示數組。但我如何知道 goroutine 何時完成處理？我可以 sleep 一段時間，給它一個完成的機會，然後停下來看看，但這感覺相當黑客。我認為有一個最佳實踐方法可以解決這個問題，但我還沒有弄清楚。

這是我的一些想法，其中零個有效。

1. 在無限 for 迴圈之外呼叫 Done()。這不起作用，因為據我所知，該程式碼無法存取。

2. 在逾時case中呼叫Done()。這是行不通的，因為超時後可能會有更多數據在路上，我希望我的 goroutine 繼續監聽。

正確答案

修改偵聽器以在通道關閉時返回。返回時呼叫 wg.Done()：

func listener(c chan *myObject, arr []*myObject, wg *sync.WaitGroup) {
    defer wg.Done()
    for {
        select {
        case value, ok := <- c:
            if !ok { return }
            arr = append(arr, value)
        case <- time.After(1 * time.Second):
            fmt.Println(arr)
        }
}

修改 main 以在傳送完成後關閉頻道。在從 main 返回之前等待 goroutine 完成。

var wg sync.WaitGroup
arr := make([]*myObject, 0)
wg.Add(1)
go listener(c, arr, &wg)
for {
    // sending stuff to c
}
close(c)
wg.Wait()

以上是終止無限循環 goroutine 的安全方法？的詳細內容。更多資訊請關注PHP中文網其他相關文章！