使用context.withcancel可取消goroutine，调用cancel()后所有监听该context的goroutine通过ctx.done()收到信号并退出。
在Golang中，
context包提供了一种优雅的方式来管理并发任务的生命周期，包括取消、超时和传递请求相关的值。它允许你构建更健壮、更可控的并发程序。
使用
context可以控制并发任务的启动、停止和传递请求范围的数据。
如何使用Context取消一个goroutine？
context.WithCancel函数是实现取消goroutine的关键。它返回一个
context和一个
CancelFunc。调用
CancelFunc会取消
context，所有监听该
context的goroutine都会收到取消信号。
package main
import (
"context"
"fmt"
"time"
)
func worker(ctx context.Context, id int) {
for {
select {
case <-ctx.Done():
fmt.Printf("Worker %d: 任务取消\n", id)
return
default:
fmt.Printf("Worker %d: 正在工作中...\n", id)
time.Sleep(time.Second)
}
}
}
func main() {
ctx, cancel := context.WithCancel(context.Background())
// 启动多个worker goroutine
for i := 1; i <= 3; i++ {
go worker(ctx, i)
}
// 模拟一段时间后取消任务
time.Sleep(3 * time.Second)
fmt.Println("准备取消所有worker...")
cancel()
// 等待一段时间，确保所有worker都已退出
time.Sleep(time.Second)
fmt.Println("所有worker已退出，程序结束")
}在这个例子中，我们创建了一个带有取消功能的
context。启动了三个
workergoroutine，每个goroutine都在循环中执行任务，并监听
context.Done()通道。当调用
cancel()函数时，
context被取消，所有
workergoroutine收到信号并退出。 我个人觉得这种方式比直接使用channel发送信号更加优雅，尤其是当你需要传递多个取消信号的时候。
立即学习“go语言免费学习笔记（深入）”；
Context的超时控制是如何实现的？
context.WithTimeout函数允许你设置一个超时时间。当超过指定时间后，
context会自动取消。这对于防止goroutine无限期地阻塞非常有用。
package main
import (
"context"
"fmt"
"time"
)
func longRunningTask(ctx context.Context) {
select {
case <-time.After(5 * time.Second):
fmt.Println("任务完成")
case <-ctx.Done():
fmt.Println("任务超时取消")
}
}
func main() {
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
defer cancel() // 确保即使任务提前完成，cancel也会被调用
go longRunningTask(ctx)
select {
case <-ctx.Done():
fmt.Println("主程序检测到任务超时")
case <-time.After(6 * time.Second): // 稍微长于longRunningTask，确保其完成或超时
fmt.Println("主程序结束")
}
}在这个例子中，
longRunningTask模拟一个需要5秒才能完成的任务。我们使用
context.WithTimeout设置了3秒的超时时间。如果任务在3秒内没有完成，
context将被取消，
longRunningTask会收到取消信号并退出。 注意
defer cancel()，这是一个好习惯，即使任务提前完成，也应该调用
cancel释放资源。
如何在Context中传递请求相关的值？
context.WithValue函数允许你在
context中存储键值对。这些值可以在goroutine之间传递，例如请求ID、用户信息等。
package main
import (
"context"
"fmt"
)
func processRequest(ctx context.Context) {
userID := ctx.Value("userID")
fmt.Printf("处理请求，用户ID: %v\n", userID)
}
func main() {
ctx := context.WithValue(context.Background(), "userID", "12345")
processRequest(ctx)
}在这个例子中，我们在
context中存储了
userID。
processRequest函数可以从
context中获取
userID并使用它。 需要注意的是，
context.Value的键应该是可比较的类型，通常是自定义类型，以避免与其他包冲突。 此外，不要在
context中传递可选参数，应该使用函数参数来传递。
Context的父子关系是如何工作的？
context可以有父子关系。子
context继承父
context的值、截止时间和取消信号。如果父
context被取消，所有子
context也会被取消。
package main
import (
"context"
"fmt"
"time"
)
func childTask(ctx context.Context) {
select {
case <-ctx.Done():
fmt.Println("子任务被取消")
case <-time.After(2 * time.Second):
fmt.Println("子任务完成")
}
}
func main() {
ctx, cancel := context.WithCancel(context.Background())
go childTask(ctx)
time.Sleep(1 * time.Second)
fmt.Println("取消父任务")
cancel()
time.Sleep(1 * time.Second) // 等待子任务退出
}在这个例子中，
childTask的
context是父
context的子
context。当父
context被取消时，
childTask也会收到取消信号。 这种父子关系使得你可以构建复杂的并发任务树，并统一管理它们的生命周期。
Context的最佳实践是什么？
- 始终传递
context
作为函数的第一个参数，尤其是当函数会启动goroutine时。 - 使用
context.TODO
作为顶级context
的初始值，表示你还不清楚需要传递什么值。 - 不要在
context
中存储必需的参数，应该使用函数参数来传递。 context
是不可变的，不要修改它。- 确保在任务完成后调用
cancel
函数，释放资源。 - 使用自定义类型作为
context.Value
的键，避免与其他包冲突。 - 理解
context
的取消和超时机制，并合理地使用它们来管理goroutine的生命周期。
总而言之，
context是Golang并发编程中一个非常强大的工具。 掌握
context的使用，可以帮助你构建更健壮、更可控的并发程序。
