golang 协程 如何中断和恢复

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

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

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

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

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

package main
 
import (
    "fmt"
    "time"
)
 
func worker(done chan bool) {
    for {
        select {
        case <-done:
            fmt.Println("Worker stopped.")
            return // 退出函数，结束 goroutine
        default:
            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 main
 
import (
    "context"
    "fmt"
    "time"
)
 
func worker(ctx context.Context) {
    for {
        select {
        case <-ctx.Done(): // 当 ctx 被取消时，这里会被触发
            fmt.Println("Worker stopped.")
            return // 退出函数，结束 goroutine
        default:
            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 main
 
import (
    "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 的生命周期。