Golang 函数并发编程如何通过 context 管理协程生命周期？-Golang-PHP中文网

Golang 函数并发编程如何通过 context 管理协程生命周期？

王林

发布： 2024-09-21 19:12:02

原创

795人浏览过

在 go 中，context 用于管理协程生命周期，避免协程泄漏或死锁。使用 context 的步骤包括：创建一个 context，指定取消信号或截止时间。在协程中使用 context，通过 select case 语句监听 context 的信号。在适当的时候取消 context，即可关闭协程。通过 context 管理协程生命周期，可以确保在需要时关闭协程，提高并发编程的效率和稳定性。

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

以上就是Golang 函数并发编程如何通过 context 管理协程生命周期？的详细内容，更多请关注php中文网其它相关文章！