Go语言通知协程退出(取消)的几种方式 - 知乎

GoLang之goroutine底层系列二(goroutine的创建、让出、恢复)_golang goroutine-CSDN博客

在 Go 语言中，协程（也称为 goroutine）是通过 go 关键字启动的轻量级线程。由于 goroutine 的调度是由 Go 运行时管理的，直接停止一个正在执行的 goroutine 是不可能的，这与操作系统线程不同。但是，你可以通过一些策略来实现“中断”和“恢复”的行为，尽管这不是传统意义上的停止和恢复。

1. 使用通道（Channel）进行信号传递

最常用的方法是使用通道（channel）来安全地中断 goroutine。你可以在另一个 goroutine 中向该 goroutine 发送一个信号（例如，一个特殊的值或者一个结构体），然后在被中断的 goroutine 中定期检查这个通道以决定是否退出循环或函数。

package mainimport ("fmt""time"
)func worker(done chan bool) {for {select {case <-done:fmt.Println("Worker stopped.")return // 退出函数，结束 goroutinedefault:fmt.Println("Working...")time.Sleep(500 * time.Millisecond) // 模拟工作}}
}func main() {done := make(chan bool)go worker(done)time.Sleep(2 * time.Second) // 等待一段时间done <- true               // 发送信号到 worker，要求它停止time.Sleep(1 * time.Second) // 等待 worker 完全停止
}

2. 使用 context 包

context 包是 Go 标准库中处理请求范围的值、取消信号和截止时间的标准方式。它非常适合用来控制 goroutine 的生命周期。

示例代码：

package mainimport ("context""fmt""time"
)func worker(ctx context.Context) {for {select {case <-ctx.Done(): // 当 ctx 被取消时，这里会被触发fmt.Println("Worker stopped.")return // 退出函数，结束 goroutinedefault:fmt.Println("Working...")time.Sleep(500 * time.Millisecond) // 模拟工作}}
}func main() {ctx, cancel := context.WithCancel(context.Background())go worker(ctx)time.Sleep(2 * time.Second) // 等待一段时间cancel()                    // 取消 context，要求 worker 停止time.Sleep(1 * time.Second) // 等待 worker 完全停止
}

3. 使用 sync.Once 或 sync.Mutex 控制退出逻辑的执行一次

如果你需要在某些情况下只安全地执行退出逻辑一次，可以使用 sync.Once 来确保。这对于确保资源清理代码只运行一次非常有用。但是，这种方法更多地是关于如何安全地执行退出逻辑，而不是真正地“中断”一个正在执行的 goroutine。

示例代码：

package mainimport ("fmt""sync""time"
)var once sync.Once
var running = true // 控制循环是否继续运行的标志位func worker() {for running { // 使用 running 作为循环条件来控制何时退出循环体中的代码块。fmt.Println("Working...")time.Sleep(500 * time.Millisecond) // 模拟工作}
}func stopWorker() { // 安全地停止 worker 的函数。使用 sync.Once 确保只调用一次。once.Do(func() { // 使用 sync.Once 来确保这个操作只执行一次。running = false // 设置 running 为 false，让循环在下次迭代时退出。})
}

在上面的代码中，你可以通过调用 stopWorker() 来停止 worker() 函数。但是，这种方法更多地是控制何时退出循环而不是真正地“中断”一个正在执行的 goroutine。在实践中，通常推荐使用通道或 context 来管理 goroutine 的生命周期。