上下文取消不退出

Question

预计：大约完成后完成2秒

实际：无限期运行。

不明白可能导致它无限期运行的原因。

package main

import (
    "context"
    "fmt"
    "time"
)

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

    for i := range generator(ctx) {
        select {
        case <-time.After(2 * time.Second):
            cancel()
            return
        default:
            fmt.Println(i)
        }
    }
}

func generator(ctx context.Context) <-chan int {
    ch := make(chan int)

    go func() {
        count := 0
        for {
            select {
            case <-ctx.Done():
                return
            case ch <- count:
                count++
            }
        }
    }()

    return ch
}

Answer 1

主要问题是，从generator(ctx)返回的频道发出的值几乎与您阅读时一样快。

由time.After(2 * time.Second)创建的频道几乎立即被丢弃，并且每次迭代通过生成器创建一个新的超时频道。

如果你做了一个小改动;在循环外部创建超时通道，然后将它放在您将看到它开始工作的select子句中。

timeout := time.After(2 * time.Second)
for i := range generator(ctx) {
    select {
    case <-timeout:
        cancel()
        return
    default:
        fmt.Println(i)
    }
}

https://play.golang.org/p/zb3wn5FJuK