在 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,您可以轻松管理协程的生命周期,确保在适当的时候取消或关闭协程。这对于防止协程泄漏和死锁至关重要。