在 go 中，context 用于管理协程生命周期，避免协程泄漏或死锁。使用 context 的步骤包括：创建一个 context，指定取消信号或截止时间。在协程中使用 context，通过 select case 语句监听 context 的信号。在适当的时候取消 context，即可关闭协程。通过 context 管理协程生命周期，可以确保在需要时关闭协程，提高并发编程的效率和稳定性。
Go 函数并发编程：使用 context 管理协程生命周期
在 Go 中使用协程可以有效提升并行处理效率。然而，管理协程的生命周期至关重要，以避免协程泄漏或死锁。Context 是管理协程生命周期的强大工具。
什么是 context？
Context 是一个存储键值对的 struct，它在协程之间传递，用于传递取消请求或其他控制信息。
立即学习“go语言免费学习笔记（深入）”；
如何使用 context 管理协程生命周期？
Go 提供了几个包来操作 context：
-
context：包含Context类型和相关辅助函数。 -
context/cancel：提供创建带取消功能的 context 的工具。 -
context/deadline：提供创建带截止时间的 context 的工具。
要使用 context 管理协程生命周期，请按照以下步骤操作：
- 创建一个 context：
import (
"context"
"fmt"
"time"
)
func main() {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()- 启动协程并从 context 中读取信号：
go func() {
for {
select {
case <-ctx.Done():
fmt.Println("协程已取消")
return
default:
// 做点事
fmt.Println("协程正在运行")
}
}
}()
time.Sleep(5 * time.Second)
cancel()
}实战案例
以下是一个使用 context 控制协程生命周期的实战案例：
package main
import (
"context"
"fmt"
"log"
"net/http"
"time"
)
func main() {
// 创建一个带有 5 秒截止时间的 context
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
// 启动 HTTP 服务器并在 context 被取消时自动关闭
go func() {
srv := &http.Server{Addr: ":8080"}
http.Handle("/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { fmt.Fprintf(w, "Hello, World!") }))
err := srv.ListenAndServe()
if err != nil && err != http.ErrServerClosed {
log.Fatal(err)
}
}()
// 5 秒后取消 context，导致 HTTP 服务器关闭
time.Sleep(5 * time.Second)
cancel()
}通过使用 context，您可以轻松管理协程的生命周期，确保在适当的时候取消或关闭协程。这对于防止协程泄漏和死锁至关重要。
