Golang 中 context 执行了Cancel，但是<- ctx.Done() 没有执行？

Question

package main

import (
    "context"
    "fmt"
)

func gen(ctx context.Context) <-chan interface{} {
    ch := make(chan interface{})
    go func() {
        n := 0
        for {
            select {
            case <-ctx.Done():
                fmt.Println("done")
            default:
                n += 1
                ch <- n
            }
        }

    }()
    return ch
}

func main() {
    ctx, cancel := context.WithCancel(context.Background())

    for n := range gen(ctx) {
        fmt.Println(n)
        if n == 5 {
            break
        }
    }
    defer cancel()
}

期望 能打印出 "done"

Answer 1

应该是阻塞在 ch<-n 了。

cancel() 如果执行的比较晚，在 cancel 之前第六个 ch<-n 就已经执行了的话，就阻塞了。（main 逻辑相对比较长，比如还有一个 Println ，所以大概率 cancel 会比较晚。

可以试试这个：

package main

import (
    "context"
    "fmt"
)

func gen(ctx context.Context) <-chan interface{} {
    ch := make(chan interface{})
    go func() {
        n := 0
        for {
            select {
            case <-ctx.Done():
                fmt.Println("done")
                close(ch) // 关闭 channel ，range 结束
                return
            default:
                n += 1
                ch <- n
            }
        }

    }()
    return ch
}

func main() {
    ctx, cancel := context.WithCancel(context.Background())

    for n := range gen(ctx) {
        fmt.Println(n)
        if n == 5 {
            cancel()
            // break 不能 break，否则没有从 ch 中的读取，就会一直阻塞在 ch <- n
        }
    }
}

这个依然会打印 6 个，因为 cancel 的时候第六个 ch<-n 已经执行了。

Answer 2

你的代码有几个问题：

1. 不应该在case default: 内再写ch, 它会永远阻塞在default条件内而无法选择其它case条件，　应该换成 case ch <- n:
2. case <- ctx.Done 应该直接返回，否则会死循环
3. main　退出时，子协程会被破退出，导致不能打印done; main 应该使用wg.Wait() 等待子协程退出

package main

import (
    "context"
    "fmt"
    "sync"
)

func gen(ctx context.Context, wg *sync.WaitGroup) <-chan interface{} {
    ch := make(chan interface{})
    go func() {
        wg.Add(1)
        defer wg.Done()
        n := 1
        for {
            select {
            case <-ctx.Done():
                fmt.Println("goroutine done")
                return
            case ch <- n:
                n++
            }
        }

    }()
    return ch
}

func main() {
    ctx, cancel := context.WithCancel(context.Background())
    var wg sync.WaitGroup
    for n := range gen(ctx, &wg) {
        fmt.Println(n)
        if n == 5 {
            break
        }
    }
    cancel()
    wg.Wait()
    println("main done")
}