在 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
