如何在 Golang 中使用 context 实现函数回调- 技术经验 -卓越飞翔博客

context 简化了并发回调函数的管理。它提供了以下方法来检测取消和存储值：deadline() 返回剩余时间，done() 返回关闭信道，err() 返回取消原因，value() 使用键检索值。

如何在 Golang 中使用 context 实现函数回调

如何在 Golang中使用 Context 实现函数回调

背景

在 Golang 中，回调函数允许我们在一个函数完成执行后执行另一个函数。然而，在并发应用程序中，管理回调的执行顺序和取消变得很棘手。 Context 提供了一个轻量级的机制来处理此类场景。

立即学习“go语言免费学习笔记（深入）”；

Context

Context 是一个接口，定义了如下方法：

Deadline()：返回到截止日期剩余的时间。
Done()：返回一个信道，当Context被取消时，该信道将关闭。
Err()：返回Context被取消的原因。
Value(key interface{}) interface{}：使用给定键检索Context 中存储的值。

使用 Context 实现函数回调

要使用 Context 实现函数回调，我们可以执行以下步骤：

在调用函数之前，创建一个 Context：

ctx := context.Background()

将 Context 作为参数传递给函数：

func myFunction(ctx context.Context) {
    // ...
}

在函数中，使用 Context 中的方法来检测取消：

select {
case <-ctx.Done():
    // 取消已被触发，返回
}

在需要时，将值存储在 Context 中：

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

实战案例：异步 HTTP 请求

我们可以在异步 HTTP 请求中使用 Context 来管理回调。例如：

func asynchronousHTTPClient(ctx context.Context, url string) (*http.Response, error) {
    httpClient := &http.Client{}
    req, _ := http.NewRequest(http.MethodGet, url, nil)
    req = req.WithContext(ctx)

    resp, err := httpClient.Do(req)
    if err != nil {
        return nil, err
    }
    return resp, nil
}

在这种情况下，我们可以创建一个 Context，并在请求执行后将其传递给 HTTP 客户端。如果在请求处理期间 Context 被取消，则请求将被放弃，并返回一个 context.Canceled 错误。

结论

使用 Context 可以简化并发回调函数的管理。这使得我们能够在并发环境中构建可靠且响应迅速的应用程序。

相关推荐