如果 goroutine 正在运行，则停止它

Question

我有一个与How to stop a goroutine 类似的问题，但稍有不同。我不确定 goroutine 是否正在运行。

var quit = make(chan bool)

func f1() {
    go func() {
        t := time.NewTimer(time.Minute)
        select {
        case <-t.C:
            // Do stuff
        case <-quit:
            return
        }
    }()
}

func f2() {
    quit <- true
}

如果在f1() 之后不到一分钟调用了f2()，则goroutine 返回。但是，如果它在一分钟后被调用，goroutine 将已经返回，f2() 将阻塞。 我希望 f2() 在 goroutine 正在运行时取消它，否则什么也不做。

我在这里想要实现的是当且仅当它没有在创建后的一分钟内被取消时才执行任务。

说明：

• 没有什么可以阻止f2() 被多次调用。
• 一次只运行一个 goroutine。 f1() 的调用者将确保每分钟调用它不超过一次。

Answer 1

使用上下文。

使用可能被取消的上下文运行f1。 使用关联的取消函数运行 f2。

func f1(ctx context.Context) {
    go func(ctx context.Context) {
        t := time.NewTimer(time.Minute)
        select {
        case <-t.C:
            // Do stuff
        case <-ctx.Done():
            return
        }
    }(ctx)
}

func f2(cancel context.CancelFunc) {
    cancel()
}

然后，为了协调这两个功能，您可以这样做：

    ctx, cancel := context.WithCancel(context.Background())
    f1(ctx)
    f2(cancel)

您还可以尝试使用 context.WithTimeout 函数来合并外部定义的超时。

如果你不知道是否有一个 goroutine 已经在运行，你可以像上面一样初始化 ctx 和 cancel 变量，但不要将它们传递给任何东西。这避免了检查nil。

请记住将 ctx 和 cancel 视为要复制的变量，而不是引用，因为您不希望多个 goroutine 共享内存 - 这可能会导致竞争条件。

Answer 2

您可以将缓冲区大小设置为 1。这意味着您可以向它发送一个值而不会阻塞，即使该值没有立即（或根本没有）接收到。

var quit = make(chan bool, 1)

我认为最佳答案更好，这只是另一种可以转化为其他情况的解决方案。