使用context.WithCancel可取消goroutine，调用cancel()后所有监听该context的goroutine通过ctx.Done()收到信号并退出。
在Golang中，
context
包提供了一种优雅的方式来管理并发任务的生命周期，包括取消、超时和传递请求相关的值。它允许你构建更健壮、更可控的并发程序。
使用
context
可以控制并发任务的启动、停止和传递请求范围的数据。
如何使用Context取消一个goroutine？
context.WithCancel
函数是实现取消goroutine的关键。它返回一个
context
和一个
CancelFunc
。调用
CancelFunc
会取消
context
，所有监听该
context
的goroutine都会收到取消信号。
package mainimport ( "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
。启动了三个
worker
goroutine，每个goroutine都在循环中执行任务，并监听
context.Done()
通道。当调用
cancel()
函数时，
context
被取消，所有
worker
goroutine收到信号并退出。 我个人觉得这种方式比直接使用channel发送信号更加优雅，尤其是当你需要传递多个取消信号的时候。
立即学习“go语言免费学习笔记（深入）”；
Context的超时控制是如何实现的？
context.WithTimeout
函数允许你设置一个超时时间。当超过指定时间后，
context
会自动取消。这对于防止goroutine无限期地阻塞非常有用。
package mainimport ( "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 mainimport ( "context" "fmt")func processRequest(ctx context.Context) { userID := ctx.Value("userID") fmt.Printf("处理请求，用户ID: %vn", 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 mainimport ( "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
context
的使用，可以帮助你构建更健壮、更可控的并发程序。
以上就是Golang使用context控制并发任务生命周期的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1403042.html
微信扫一扫 
支付宝扫一扫 
