如何在 Golang 函数中传递上下文信息？

学习知识要善于思考，思考，再思考！今天golang学习网小编就给大家带来《如何在 Golang 函数中传递上下文信息？》，以下内容主要包含等知识点，如果你正在学习或准备学习Golang，就都不要错过本文啦~让我们一起来看看吧，能帮助到你就更好了！

Golang 中可以使用 context 包来传递上下文信息：使用 context.Background() 创建一个新的上下文。使用 context.WithValue() 将值添加到上下文中。使用 ctx.Value() 获取上下文中存储的值。

如何在 Golang 函数中传递上下文信息？

在并发程序中传递上下文信息至关重要，它允许在不同 goroutine 中访问共享数据。这对于日志记录、追踪、错误处理和传递中间数据很有用。

使用 Context 包

Go 标准库提供了一个 context 包，它提供了用于管理上下文信息的类型和函数。

要创建一个新的 context.Context，请使用 context.Background()：

ctx := context.Background()

您可以使用 context.WithValue() 将值添加到上下文中：

ctx = context.WithValue(ctx, "key", "value")

您可以在任何地方获取上下文中存储的值：

value := ctx.Value("key")

实战案例

假设我们有一个 goroutine池，它执行并行任务。我们要在每个 goroutine 中记录任务的开始和结束时间。

package main

import (
    "context"
    "fmt"
    "time"
)

func main() {
    ctx, cancel := context.WithCancel(context.Background())
    defer cancel()

    // 创建 goroutine 池
    pool := make(chan func())
    for i := 0; i < 10; i++ {
        go func(ctx context.Context, i int) {
            fmt.Println("Task", i, "started")

            // 这里我们使用了上下文里的时间戳来计算任务耗时
            start := time.Now()
            ctx = context.WithValue(ctx, "start", start)

            // 模拟任务执行
            time.Sleep(time.Second)

            // 获取任务开始时间并计算耗时
            start = ctx.Value("start").(time.Time)
            fmt.Printf("Task %d ended, took %v\n", i, time.Since(start))
        }(ctx, i)
        pool <- nil
    }
    close(pool)
    time.Sleep(5 * time.Second)
}

在此示例中，我们使用 context.WithValue() 将 start 时间戳添加到上下文中。然后，在 goroutine 中，我们可以通过 ctx.Value() 获取开始时间戳，并计算任务的耗时。

注意事项

• 垃圾回收会清理上下文对象，因此确保在需要时进行传递。
• 不要将引用对象存储在上下文中，因为它可能会导致内存泄漏。
• 避免在多个 goroutine 中写入相同的上下文，因为这可能会导致竞争条件。

今天关于《如何在 Golang 函数中传递上下文信息？》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！