如何使用通道通知 goroutine 正确退出

Question

注意：我用谷歌搜索了这个主题，几乎阅读了我能找到的所有内容，但仍然无法获得正确/合理/生产就绪的答案。

基本上所有的答案都是相似的，就像这个：how to stop a groutine，都是相同的模式，没有例外：真正的工作是fmt.Println(1)打印一些东西，或者只是// Do other stuff，

但是如果将实际工作保留在for select default case branch，那么它将被执行多次，打印一些东西是可以的，但显然它还没有准备好进行实际工作。

我能想象的唯一有效方法是将真正的工作放在案例分支上，然后向该案例发送 only 一个 信号以通知它开始，就像这里：playground，但只是感觉想，同样使用这种方法，它是否会产生一些潜在的问题？

添加代码片段以准确显示我想要实现的目标：https://play.golang.org/p/7xjjiW7XdoQ，我想在客户端关闭连接时实现，然后立即终止我的处理程序，释放资源，然后无条件退出。

Answer 1

您希望在多个位置检查取消信号，而不是使用for select 循环。但是为了不阻塞，您仍然必须使用select（默认情况为空），这使它有点尴尬。我使用的是 context.Context 而不是“普通取消通道”，但想法是一样的：

func doWork(ctx context.Context) {
    fmt.Println("Doing some work 1")
    time.Sleep(time.Second * 5)
    // check is the task cancelled
    select {
    case <-ctx.Done():
        fmt.Println("cancelled at checkpoint 1")
        return
    default:
    }
    fmt.Println("Doing some work 2")
    time.Sleep(time.Second * 5)
    // check is the task cancelled
    select {
    case <-ctx.Done():
        fmt.Println("cancelled at checkpoint 2")
        return
    default:
    }
    fmt.Println("Doing some work 3")
    time.Sleep(time.Second * 5)
}

func main() {
    var wg sync.WaitGroup
    wg.Add(1)
    ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
    defer cancel()
    go func() {
        doWork(ctx)
        wg.Done()
    }()
    wg.Wait()
    fmt.Println("done")
}