如何在不阻塞的情况下判断 Goroutine 是否完成？

在 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 占用过高
        }
    }
}

代码解释：

InsCode

InsCode 是CSDN旗下的一个无需安装的编程、协作和分享社区

下载

1. worker 函数模拟一个 Goroutine 的工作，完成后会关闭 channel ch。
2. 在 main 函数中，使用 select 语句来非阻塞地从 channel 接收数据。
3. case val, ok :=
4. 如果 ok 为 true，表示 channel 尚未关闭，Goroutine 仍在运行，接收到数据 val。
5. 如果 ok 为 false，表示 channel 已经关闭，Goroutine 已经完成。
6. default: 分支在 channel 中没有数据可读时执行，避免阻塞。在这里可以执行其他操作，或者简单地等待一段时间后再次检查。

注意事项：

• 务必在 Goroutine 完成后关闭 channel，否则 "comma, ok" 模式无法正常工作。
• 使用 select 语句可以避免阻塞，并且可以在 channel 没有数据时执行其他操作。
• time.Sleep 可以降低 CPU 占用率，避免过度消耗资源。

总结：

通过 "comma, ok" 模式和 select 语句，可以在 Go 语言中实现非阻塞地检测 Goroutine 是否完成。这种方法简单有效，避免了阻塞，并且可以灵活地处理 channel 中没有数据的情况。 这种方法适用于需要异步执行任务，并在不阻塞主线程的情况下检测任务完成状态的场景。