Go1.21发布！Context函数更强大，开发更加轻松！

2024-01-11 23:14:19

Context：Go 1.21 中控制并发任务的神器

简介

在 Go 语言中，Context 是一个至关重要的数据结构，它允许程序员传递上下文信息并控制任务的执行。随着 Go 1.21 版本的发布，Context 函数得到了大幅增强，引入了取消原因和回调函数支持。本文将深入探讨这些增强功能，展示它们如何为开发人员提供对任务执行的更精细控制。

取消原因

在 Go 1.21 之前，当一个 Context 被取消时，开发人员只能调用 Context.Done() 方法来了解该事实。然而，这个方法不会提供取消的原因。随着 Go 1.21 的发布，开发人员可以通过 Context.Err() 方法获取取消原因。

取消原因有助于识别触发取消操作的具体事件或错误。这对于调试和理解并发任务的行为非常有用。

回调函数

在 Go 1.21 之前，当一个 Context 被取消时，开发人员只能通过注册一个 Context.CancelFunc 函数来执行某些操作。但是，这个函数只能执行一次。

在 Go 1.21 中，Context.AddCallback() 方法允许注册多个回调函数，并且这些函数可以多次执行。这提供了更大的灵活性，允许开发人员根据 Context 取消状态执行各种操作。

示例代码

以下示例代码演示了如何在 Go 1.21 中使用取消原因和回调函数：

package main

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

func main() {
    // 创建一个 Context
    ctx, cancel := context.WithCancel(context.Background())

    // 注册一个回调函数
    ctx = context.WithValue(ctx, "key", "value")

    // 启动一个 goroutine，并在 Context 被取消时执行回调函数
    go func() {
        select {
        case <-ctx.Done():
            fmt.Println("Context cancelled:", ctx.Err())
        }
    }()

    // 取消 Context
    cancel()

    // 等待 goroutine 退出
    time.Sleep(100 * time.Millisecond)
}