Golang 中如何解决 context.Done() 在协程阻塞时无法执行的问题？-Golang-PHP中文网

Golang 中如何解决 context.Done() 在协程阻塞时无法执行的问题？

花韻仙語

发布： 2024-11-13 16:30:01

原创

396人浏览过

golang 中如何解决 context.done() 在协程阻塞时无法执行的问题？

处理 golang context 取消时阻塞的 <- ctx.done()

在 golang 中，使用 context.context 管理协程的取消非常普遍。但是，在某些情况下，即使调用了 cancel 方法，<- ctx.done() 也可能不会执行。

问题描述

考虑以下代码：

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

func gen(ctx context.context) <-chan interface{} {
    ch := make(chan interface{})
    go func() {
        n := 0
        for {
            select {
            case <-ctx.done():
                // 期望打印 "done" 输出
            default:
                n += 1
                ch <- n
            }
        }
    }()
    return ch
}

登录后复制

在 main 函数中，我们希望打印 "done" 输出：

ctx, cancel := context.withcancel(context.background())
for n := range gen(ctx) {
    fmt.println(n)
    if n == 5 {
        cancel()
        break
    }
}

登录后复制

然而，实际运行结果可能未打印 "done" 输出。

无涯·问知

无涯·问知，是一款基于星环大模型底座，结合个人知识库、企业知识库、法律法规、财经等多种知识源的企业级垂直领域问答产品

查看详情

原因和解决方法

当调用 cancel() 时，如果协程中 <- ctx.done() 在执行其他操作之前被阻塞，就会出现这种问题。例如，如果协程正在等待从一个管道中读取数据，则在此读取操作完成之前，<- ctx.done() 将不会执行。

要解决这个问题，考虑在协程中关闭管道或其他阻塞操作。例如：

func gen(ctx context.Context) <-chan interface{} {
    ch := make(chan interface{})
    go func() {
        n := 0
        for {
            var closeCh bool // 关闭标记

            select {
            case <-ctx.Done():
                closeCh = true // 设置关闭标记
                fmt.Println("done")
                close(ch) // 关闭管道
                return
            default:
                n += 1
                ch <- n
            }

            if closeCh { 
              break // 关闭标记变为真，则退出循环
            }
        }
    }()
    return ch
}

// ... 剩余代码略 ...

登录后复制

这样，当调用 cancel() 时，<- ctx.done() 将在阻塞读取操作完成之前执行，并且协程将正确地退出。

以上就是Golang 中如何解决 context.Done() 在协程阻塞时无法执行的问题？的详细内容，更多请关注php中文网其它相关文章！