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

处理 golang context 取消时阻塞的

在 golang 中，使用 context.context 管理协程的取消非常普遍。但是，在某些情况下，即使调用了 cancel 方法，

问题描述

考虑以下代码：

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() 时，如果协程中

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

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() 时，

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