Golang 中如何解决 context.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
}

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

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
}

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