处理 golang context 取消时阻塞的
在 golang 中，使用 context.context 管理协程的取消非常普遍。但是，在某些情况下，即使调用了 cancel 方法，
问题描述
考虑以下代码：
立即学习“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() 时，如果协程中
要解决这个问题，考虑在协程中关闭管道或其他阻塞操作。例如：
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() 时，
