golang并发编程之context控制协程退出

在Golang中，Context是用于控制协程退出的工具。Context可以让我们在多个goroutine之间传递取消信号，并且能够及时终止协程的执行。它非常适用于一些需要及时停止的场景，比如网络请求、任务执行等。

使用Context的基本流程如下：

1. 使用context包创建一个Context对象。
2. 在需要控制的goroutine中，调用Context的Done方法来检查是否需要退出。
3. 如果需要退出，则退出当前goroutine。

下面是一个示例程序，演示如何使用Context来控制协程退出：

package main

import (
    "fmt"
    "time"
    "context"
)

func longRunningTask(ctx context.Context) {
    for {
        select {
        case <- ctx.Done():
            fmt.Println("longRunningTask exit")
            return
        default:
            // 执行一些耗时的操作
            time.Sleep(1 * time.Second)
            fmt.Println("longRunningTask running")
        }
    }
}

func main() {
    ctx, cancel := context.WithCancel(context.Background())
    go longRunningTask(ctx)

    // 运行一段时间后停止任务
    time.Sleep(5 * time.Second)
    cancel()
    fmt.Println("task was cancelled")
    time.Sleep(1 * time.Second)
}

在这个示例中，我们首先使用context包创建了一个Context对象ctx和一个取消函数cancel。然后我们启动了一个长时间运行的任务longRunningTask，并传入了Context对象作为参数。

在longRunningTask中，我们通过select语句来检查是否需要退出当前goroutine。如果收到了取消信号，则输出一条消息并返回函数；否则继续执行耗时操作。

在main函数中，我们运行了一个长时间的任务，并在5秒后调用cancel函数来发送取消信号。当longRunningTask接收到取消信号后，会退出函数，并输出一条消息。最后我们等待1秒钟，让程序有足够的时间来处理各种清理工作。

Context还可以用于超时控制、请求传递等其他场景。在实际使用中，我们需要根据具体的情况来选择合适的Context对象，并根据需要传递上下文信息和取消信号。同时，也需要注意在多个goroutine之间传递Context对象时的线程安全性。