如何通过通道来掔停长时间运行的操作

编程并不是一个机械性的工作，而是需要有思考，有创新的工作，语法是固定的，但解决问题的思路则是依靠人的思维，这就需要我们坚持学习和更新自己的知识。今天golang学习网就整理分享《如何通过通道来掔停长时间运行的操作》，文章讲解的知识点主要包括，如果你对Golang方面的知识点感兴趣，就不要错过golang学习网，在这可以对大家的知识积累有所帮助，助力开发能力的提升。

我正在研究一个使用 go 通道的示例，但遇到了问题。假设我们有一个运行大约 1 小时的方法。该方法执行各种操作，例如写入数据库、读取数据和发送电子邮件。我想让这个方法可以从外部中断，并且我正在尝试使用通道来实现这一点。

我遇到的问题是，由于没有从 <-cancelled 接收到信号，因此正在执行默认情况下的代码。但是，默认情况下的代码需要 5 秒才能执行，并且我在第 2 秒从主函数发送取消信号。

由于默认情况下的代码会继续执行，因此只有在默认块完成后才会捕获取消信号。

我们希望默认情况下的代码在收到取消信号时立即停止。

预先感谢您的帮助。祝你有美好的一天！

func worker(cancelled <-chan struct{}, wg *sync.WaitGroup) {
    defer wg.Done()
    for {
        select {
        case <-cancelled:
            fmt.Println("Worker cancelled")
            return
        default:
            fmt.Println("Working...")
            time.Sleep(5 * time.Second)
            fmt.Println("ended")
        }
    }
}

func main() {
    cancelled := make(chan struct{})
    var wg sync.WaitGroup
    wg.Add(1)
    go worker(cancelled, &wg)

    go func() {
        time.Sleep(2 * time.Second)
        cancelled <- struct{}{}
    }()

    wg.Wait()
    fmt.Println("Program completed")
}

正确答案

您应该使用 context.context 而不是通道。事实上，这种操作是使用 context 包的常见模式之一。 context 提供了使用同一上下文的 done 通道来停止多个 go 例程的能力。

select {
  case ctx.Done():
    return
  default:
    longRunningFunc(ctx)
}

如果 longrunningfunc 执行多个操作，则可以传递相同的上下文并让所有需要在收到信号时立即停止的函数监听。

但是，如果长时间运行的操作是单个操作，则无法在中间停止它。在这种情况下，考虑这样的操作是否应该在信号上运行还是作为默认情况运行会很有用。

终于介绍完啦！小伙伴们，这篇关于《如何通过通道来掔停长时间运行的操作》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！