在 Go 語言中，判斷一個 Goroutine 是否完成，通常需要通過 channel 進行通信。然而，直接從 channel 接收數據 (
“Comma, Ok” 模式
Go 語言的 “comma, ok” 模式允許在從 channel 接收數據時，同時獲取一個布爾值，指示 channel 是否已關閉。如果 channel 已經關閉，該布爾值為 false；否則，為 true。
以下是一個示例：
package main import ( "fmt" "time" ) func worker(ch chan int) { defer close(ch) // Goroutine 完成后關閉 channel time.Sleep(2 * time.Second) ch <- 123 // 向 channel 發送數據 } func main() { ch := make(chan int) go worker(ch) // 非阻塞地檢查 Goroutine 是否完成 for { select { case val, ok := <-ch: if ok { fmt.Println("Received:", val) fmt.Println("Goroutine is still running.") } else { fmt.Println("Channel is closed.") fmt.Println("Goroutine has finished.") return } default: fmt.Println("No data available yet, checking again...") time.Sleep(500 * time.Millisecond) // 避免 CPU 占用過高 } } }
代碼解釋：
- worker 函數模擬一個 Goroutine 的工作，完成后會關閉 channel ch。
- 在 main 函數中，使用 select 語句來非阻塞地從 channel 接收數據。
- case val, ok :=
- 如果 ok 為 true，表示 channel 尚未關閉，Goroutine 仍在運行，接收到數據 val。
- 如果 ok 為 false，表示 channel 已經關閉，Goroutine 已經完成。
- default: 分支在 channel 中沒有數據可讀時執行，避免阻塞。在這里可以執行其他操作，或者簡單地等待一段時間后再次檢查。
注意事項：
- 務必在 Goroutine 完成后關閉 channel，否則 “comma, ok” 模式無法正常工作。
- 使用 select 語句可以避免阻塞，并且可以在 channel 沒有數據時執行其他操作。
- time.Sleep 可以降低 CPU 占用率，避免過度消耗資源。
總結：
通過 “comma, ok” 模式和 select 語句，可以在 Go 語言中實現非阻塞地檢測 Goroutine 是否完成。這種方法簡單有效，避免了阻塞，并且可以靈活地處理 channel 中沒有數據的情況。 這種方法適用于需要異步執行任務，并在不阻塞主線程的情況下檢測任務完成狀態的場景。
? 版權聲明
文章版權歸作者所有，未經允許請勿轉載。
THE END
