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

哈喽！大家好，很高兴又见面了，我是golang学习网的一名作者，今天由我给大家带来一篇《Golang 中如何解决 context.Done() 在协程阻塞时无法执行的问题？》，本文主要会讲到等等知识点，希望大家一起学习进步，也欢迎大家关注、点赞、收藏、转发! 下面就一起来看看吧！

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

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

问题描述

考虑以下代码：

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() 在协程阻塞时无法执行的问题？》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！