Golang 函数：使用 done 通道进行上下文取消

php中文网 2024-10-15 11:55:14

使用 done 通道可实现 go 函数中的上下文取消：创建一个无缓冲 done 通道，并由调用方传递给函数。函数中启动一个 goroutine 监听 done 通道。当需要取消函数时，调用方向 done 通道发送一个值，触发函数 goroutine 退出。

Go 语言函数：使用 done 通道进行上下文取消

done 通道是一种用于在 Go 语言函数中取消上下文的机制。当需要在特定条件下终止函数执行时，它非常有用，例如当接收到信号或超时时。

概念

立即学习“go语言免费学习笔记（深入）”；

done 通道是一个无缓冲通道，通常由调用方创建并传递给函数。函数启动一个 goroutine 来监听 done 通道。当调用方决定取消函数时，它会向 done 通道发送一个值，从而导致 goroutine 退出。

代码示例

考虑以下函数，该函数无限循环地打印数字：

func printNumbers(done <-chan struct{}) {
    for {
        select {
        case <-done:
            return
        default:
            fmt.Println("Printing number...")
            time.Sleep(1 * time.Second)
        }
    }
}

我们可以使用 done 通道来取消此函数：

func main() {
    // 创建一个 done 通道
    done := make(chan struct{})

    // 启动打印数字的 goroutine
    go printNumbers(done)

    // 等待 5 秒钟
    time.Sleep(5 * time.Second)

    // 取消函数
    close(done)
}

当 main 函数关闭 done 通道时，printNumbers 函数中的 goroutine 会收到信号并退出循环。

实战案例

done 通道在需要以可取消的方式执行任务的场景中特别有用。例如：