doujiaozhan2413 2017-09-24 03:49
浏览 33
已采纳

上下文取消不会退出

Expected: To be done after approx. 2 seconds

Actual: Runs indefinitely.

Don't understand what could be causing it to run indefinitely.

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
}
  • 写回答

1条回答 默认 最新

  • duanlu9816 2017-09-24 04:44
    关注

    The main issue is that your channel returned from generator(ctx) emits values almost as fast as you can read them.

    The channel created by time.After(2 * time.Second) is discarded almost immediately, and you create a new timeout channel every iteration through the generator.

    If you make one small change; create the timeout channel outside the loop, and then put it in the select clause you'll see it begin to work.

    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

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?