Go语言中的Context

news2025/4/22 10:38:30

目录

Go语言中的Context

1. Context的基本概念

1.1 Context的核心作用

2. Context的基本用法

2.1 创建Context

背景Context

可取消的Context

带有超时的Context

2.2 在Goroutine间传递Context

2.3 获取Context的值

为Context添加自定义数据

访问Context中的值

3. Context的高级用法

3.1 Context链

3.2 多个Context的选择

3.3 Context的使用规范

4. Context的最佳实践

4.1 在HTTP处理中使用Context

4.2 在数据库查询中使用Context

4.3 在多层函数调用中传递Context

4.4 使用Context进行资源释放

5. Context的替代方案

5.1 使用通道传递取消信号

5.2 使用 ErrGroup 进行错误处理

6. 总结

Go语言中的Context

在Go语言中，context是一个重要的功能，用于在多个goroutine之间传递取消信号、超时控制和请求相关的上下文信息。它是Go语言并发编程中的一个关键组件，能够有效地管理不同任务之间的协作和资源释放。本文将详细探讨context的功能、用法及其在实际开发中的应用场景。

1. Context的基本概念

context，即上下文，在Go语言中是一个接口，定义了四个方法：CancelFunc, Deadline, Done, 和 Err。它主要用于在不同的goroutine之间传递取消信号和上下文信息。

以下是context.Context接口的定义：

type Context interface {
    Deadline() (deadline time.Time, ok bool)
    Done() <-chan struct{}
    Err() error
    Value(key interface{}) interface{}
}

1.1 Context的核心作用

  1. 取消信号：context可以在父goroutine中创建，并传递给子goroutine。当父goroutine完成时，可以通过调用CancelFunc取消子goroutine的执行。
  2. 超时控制：context可以设置一个超时时间，确保子goroutine在指定时间内完成任务，防止无限等待。
  3. 上下文信息：context可以携带一些请求相关的信息，比如用户ID、请求ID、开始时间等，方便在不同的goroutine中访问和使用。

2. Context的基本用法

2.1 创建Context

context可以通过context.Background()和context.WithCancel等方法创建。常见的创建方式如下：

背景Context

所有的context都应该从context.Background()开始，这是整个上下文树的根节点。

ctx = context.Background()
可取消的Context

使用context.WithCancel创建一个可取消的context。

ctx, cancel := context.WithCancel(context.Background())
defer cancel()
带有超时的Context

使用context.WithDeadline或context.WithTimeout创建一个带有超时时间的context。

// 使用Deadline
ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(5*time.Second))
defer cancel()

// 使用Timeout
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()

2.2 在Goroutine间传递Context

context的设计初衷是在线性调用链中传递。当启动一个新的goroutine时，应将context传递给该goroutine。

package main

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

func worker(ctx context.Context) {
    select {
    case <-ctx.Done():
        fmt.Println("Worker: Context已取消")
    case <-time.After(5 * time.Second):
        fmt.Println("Worker: 完成任务")
    }
}

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

    go worker(ctx)

    fmt.Println("Main: 等待5秒后取消Context")
    time.Sleep(3 * time.Second)
    cancel()
    fmt.Println("Main: Context已取消")
}

在上述代码中，main函数和worker函数共享同一个context。当main函数调用cancel()时，worker函数会通过ctx.Done()信号知道已被取消。

2.3 获取Context的值

context还可以携带键值对的数据，通过Value(key interface{})方法获取。

为Context添加自定义数据

使用context.WithValue将自定义数据添加到context中。

ctx = context.WithValue(context.Background(), "requestID", "12345")
访问Context中的值

在需要访问context值的位置，调用Value(key)方法，并传入相应的键。

requestID, ok := ctx.Value("requestID").(string)
if ok {
    fmt.Printf("Request ID: %s\n", requestID)
}

需要注意的是，Value方法返回的是一个interface{}类型，需要进行类型断言才能使用。

3. Context的高级用法

3.1 Context链

context可以形成链条结构，每个子context继承自父context，并添加额外的值或取消操作。

ctxBackground := context.Background()
ctxWithValue := context.WithValue(ctxBackground, "requestID", "12345")
ctxWithCancel := context.WithCancel(ctxWithValue)

在Context链中，子context会继承父context的值，同时也可以有自己的值和取消操作。

3.2 多个Context的选择

在多个context同时存在时，通常需要使用select语句来处理多个Done()信号。

select {
case <-ctx1.Done():
    handleCancel(ctx1)
case <-ctx2.Done():
    handleCancel(ctx2)
default:
    // 进行其他操作
}

3.3 Context的使用规范

  1. 避免作为结构体的字段：context不应该作为结构体的字段，而是应该通过函数参数传递。
  2. 不应长时间持有context：context是用于短期的取消和超时控制，不应长时间持有，特别是在函数之间传递。
  3. 避免将context存储在全局变量中：全局变量会导致context的生命周期难以控制，增加资源泄漏的风险。
  4. 使用context管理资源：利用context的Done()信号，释放不再需要的资源，如文件句柄、网络连接等。

4. Context的最佳实践

4.1 在HTTP处理中使用Context

在处理HTTP请求时，context可以用来传递请求相关的信息，并在出现错误或超时时及时取消后续操作。

package main

import (
    "context"
    "fmt"
    "net/http"
)

func handler(w http.ResponseWriter, r *http.Request) {
    ctx := r.Context()
    requestID := ctx.Value("requestID")
    fmt.Printf("处理请求 ID: %s\n", requestID)
    // 处理具体业务逻辑
}

4.2 在数据库查询中使用Context

context可以用于设置数据库查询的超时时间，避免长时间阻塞。

package main

import (
    "context"
    "database/sql"
    "fmt"
    "time"
)

func queryDatabase(ctx context.Context) {
    query := "SELECT * FROM mytable"
    ctxTimeout, cancel := context.WithTimeout(ctx, 5*time.Second)
    defer cancel()

    rows, err := db.QueryContext(ctxTimeout, query)
    if err != nil {
        fmt.Printf("查询失败: %v\n", err)
        return
    }
    defer rows.Close()

    // 处理查询结果
}

4.3 在多层函数调用中传递Context

在多层函数调用中，始终将context作为第一个参数传递，确保取消信号和超时能够正确传播。

package main

import (
    "context"
    "fmt"
)

func outerFunction(ctx context.Context) {
    innerFunction(ctx)
}

func innerFunction(ctx context.Context) {
    // 使用ctx进行操作
    fmt.Println("内层函数: 使用传递过来的Context")
}

4.4 使用Context进行资源释放

通过context的Done()信号，可以在需要时及时释放资源，如关闭文件、断开连接等。

package main

import (
    "context"
    "fmt"
    "os"
)

func processFile(ctx context.Context, filename string) {
    file, err := os.Open(filename)
    if err != nil {
        fmt.Printf("打开文件失败: %v\n", err)
        return
    }

    select {
    case <-ctx.Done():
        fmt.Println("Context取消，关闭文件")
        file.Close()
        return
    default:
        fmt.Println("开始处理文件")
        // 处理文件内容
    }
}

5. Context的替代方案

虽然context是Go语言标准库提供的最佳解决方案，但在某些特定场景下，开发者可能会寻求其他替代方案。以下是几种常见的替代方案：

5.1 使用通道传递取消信号

除了context，开发者还可以通过通道传递取消信号。

package main

import (
    "fmt"
)

func worker(done <-chan struct{}) {
    select {
    case <-done:
        fmt.Println("Worker: 已取消")
    }
}

func main() {
    done := make(chan struct{})
    go worker(done)
    fmt.Println("Main: 等待3秒后取消")
    time.Sleep(3 * time.Second)
    done <- struct{}{}
}

5.2 使用 ErrGroup 进行错误处理

在处理多个子任务时，可以使用errgroup.Group来管理每个任务的错误，并在任意一个任务失败时取消整个组。

package main

import (
    "context"
    "fmt"
    "sync/errgroup"
)

func worker(ctx context.Context) error {
    // 执行具体的工作
    return nil
}

func main() {
    ctx := context.Background()
    g, egctx := errgroup.WithContext(ctx)
    
    for i := 0; i < 5; i++ {
        g.Go(func() error {
            return worker(egctx)
        })
    }

    if err := g.Wait(); err != nil {
        fmt.Printf("错误: %v\n", err)
        return
    }
}

6. 总结

context是Go语言中用于在多个goroutine之间传递取消信号、超时控制和上下文信息的重要机制。通过合理使用context，开发者可以更高效地管理并发任务，确保资源的及时释放和程序的健壮性。在实际开发中，遵循context的使用规范和最佳实践，能够显著提升代码的可维护性和性能。

无论是处理HTTP请求、数据库查询，还是在多层函数调用中传递信息，context都能发挥其独特的作用。

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.coloradmin.cn/o/2340042.html

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈，一经查实，立即删除！

相关文章

一文了解相位阵列天线中的真时延

一文了解相位阵列天线中的真时延

本文要点 真时延是宽带带相位阵列天线的关键元素之一。 真时延透过在整个信号频谱上应用可变相移来消除波束斜视现象。 在相位阵列中使用时延单元或电路板&#xff0c;以提供波束控制和相移。 市场越来越需要更快、更可靠的通讯网络&#xff0c;而宽带通信系统正在努力满…
阅读更多...
linux学习 5 正则表达式及通配符

linux学习 5 正则表达式及通配符

重心应该放在通配符的使用上 正则表达式 正则表达式是用于 文本匹配和替换 的强大工具 介绍两个交互式的网站来学习正则表达式 regexlearn 支持中文 regexone 还有一个在线测试的网址 regex101 基本规则 符号作用示例.匹配任何字符除了换行a.b -> axb/a,b[abc]匹配字符…
阅读更多...
基于超启发鲸鱼优化算法的混合神经网络多输入单输出回归预测模型 HHWOA-CNN-LSTM-Attention

基于超启发鲸鱼优化算法的混合神经网络多输入单输出回归预测模型 HHWOA-CNN-LSTM-Attention

基于超启发鲸鱼优化算法的混合神经网络多输入单输出回归预测模型 HHWOA-CNN-LSTM-Attention 随着人工智能技术的飞速发展&#xff0c;回归预测任务在很多领域得到了广泛的应用。尤其在金融、气象、医疗等领域&#xff0c;精确的回归预测模型能够为决策者提供宝贵的参考信息。为…
阅读更多...
Android RK356X TVSettings USB调试开关

Android RK356X TVSettings USB调试开关

Android RK356X TVSettings USB调试开关 平台概述操作-打开USB调试实现源码补充说明 平台 RK3568 Android 11 概述 RK3568 是瑞芯微&#xff08;Rockchip&#xff09;推出的一款高性能处理器&#xff0c;支持 USB OTG&#xff08;On-The-Go&#xff09;和 USB Host 功能。US…
阅读更多...
消息队列知识点详解

消息队列知识点详解

消息队列场景 什么是消息队列 可以把消息队列理解一个使用队列来通信的组件&#xff0c;它的本质是交换机队列的模式&#xff0c;实现发送消息&#xff0c;存储消息&#xff0c;消费消息的过程。 我们通常说的消息队列&#xff0c;MQ其实就是消息中间件&#xff0c;业界中比较…
阅读更多...
序列号绑定的SD卡坏了怎么办？

序列号绑定的SD卡坏了怎么办？

在给SD卡烧录程序的时候&#xff0c;大家发现有的卡是无法烧录的&#xff0c;如&#xff1a;复印机的SD卡不能被复制通常涉及以下几个技术原因&#xff0c;可能与序列号绑定、加密保护或硬件限制有关&#xff1a; 一、我们以复印机的系统卡为例来简单讲述一下 序列号或硬件绑定…
阅读更多...
使用SystemWeaver生成SOME/IP ETS ARXML的完整实战指南

使用SystemWeaver生成SOME/IP ETS ARXML的完整实战指南

使用SystemWeaver生成SOME/IP ETS ARXML的完整实战指南 前些天发现了一个巨牛的人工智能学习网站&#xff0c;通俗易懂&#xff0c;风趣幽默&#xff0c;可以分享一下给大家。点击跳转到网站。 https://www.captainbed.cn/ccc 一、SystemWeaver与SOME/IP基础认知 1.1 SystemWe…
阅读更多...
Flutter 状态管理 Riverpod

Flutter 状态管理 Riverpod

Android Studio版本 Flutter SDK 版本 将依赖项添加到您的应用 flutter pub add flutter_riverpod flutter pub add riverpod_annotation flutter pub add dev:riverpod_generator flutter pub add dev:build_runner flutter pub add dev:custom_lint flutter pub add dev:riv…
阅读更多...
【HarmonyOS 5】VisionKit人脸活体检测详解

【HarmonyOS 5】VisionKit人脸活体检测详解

【HarmonyOS 5】VisionKit人脸活体检测详解 一、VisionKit人脸活体检测是什么&#xff1f; VisionKit是HamronyOS提供的场景化视觉服务工具包。 华为将常见的解决方案&#xff0c;通常需要三方应用使用SDK进行集成。华为以Kit的形式集成在HarmoyOS系统中&#xff0c;方便三方…
阅读更多...
Pycharm（九）函数的闭包、装饰器

Pycharm（九）函数的闭包、装饰器

目录 一、函数参数 二、闭包 三、装饰器 一、函数参数 def func01():print("func01 shows as follows") func01() # 函数名存放的是函数所在空间的地址 print(func01)#<function func01 at 0x0000023BA9FC04A0> func02func01 print(func02)#<function f…
阅读更多...
【深度学习】详解矩阵乘法、点积，内积，外积、哈达玛积极其应用|tensor系列02

【深度学习】详解矩阵乘法、点积，内积，外积、哈达玛积极其应用|tensor系列02

博主简介&#xff1a;努力学习的22级计算机科学与技术本科生一枚&#x1f338;博主主页&#xff1a; Yaoyao2024往期回顾&#xff1a;【深度学习】你真的理解张量了吗&#xff1f;|标量、向量、矩阵、张量的秩|01每日一言&#x1f33c;: “脑袋想不明白的&#xff0c;就用脚想”…
阅读更多...
MH2103系列coremark1.0跑分数据和优化，及基于arm2d的优化应用

MH2103系列coremark1.0跑分数据和优化，及基于arm2d的优化应用

CoreMark 1.0 介绍 CoreMark 是由 EEMBC&#xff08;Embedded Microprocessor Benchmark Consortium&#xff09;组织于 2009 年推出的一款用于衡量嵌入式系统 CPU 或 MCU 性能的标准基准测试工具。它旨在替代陈旧的 Dhrystone 标准&#xff08;Dhrystone 容易受到各种libc不同…
阅读更多...
Flowith AI，解锁下一代「知识交易市场」

Flowith AI，解锁下一代「知识交易市场」

前言 最近几周自媒体号都在疯狂推Manus&#xff0c;看了几篇测评后&#xff0c;突然在某个时间节点&#xff0c;在特工的文章下&#xff0c;发现了很小众的Flowith。 被这段评论给心动到&#xff0c;于是先去注册了下账号。一翻探索过后&#xff0c;发现比我想象中要有趣的多&…
阅读更多...
SpringBoot企业级开发之【文章分类-新增文章分类】

SpringBoot企业级开发之【文章分类-新增文章分类】

看一下新增文章的需求&#xff1a; 接口文档&#xff1a; 开发思路&#xff1a; 先在controller下去创建add方法&#xff0c;方法内导入Service类获取add的结果&#xff1b;再在Service接口下去创建add的方法&#xff1b;然后在Service实现类下去实现方法的作用&#xff0c;且导…
阅读更多...
【AI News | 20250421】每日AI进展

【AI News | 20250421】每日AI进展

AI Repos 1、langgraph-mcp-agents 基于LangGraph的AI智能体系统&#xff0c;集成了MCP&#xff0c;使AI助手能访问各种数据源和API。提供了Streamlit网页界面&#xff0c;方便与LangGraph和MCP工具交互。可以通过界面动态添加、删除以及配置MCP工具&#xff0c;无需重启应用&…
阅读更多...
牛客 | OJ在线编程常见输入输出练习

牛客 | OJ在线编程常见输入输出练习

1.只有输出 言归正传&#xff0c;本张试卷总共包括18个题目&#xff0c;包括了笔试情况下的各种输入输出。 第一题不需要输入&#xff0c;仅需输出字符串 Hello Nowcoder! 即可通过。 #include <iostream> using namespace std; int main(){string s "Hello Nowco…
阅读更多...
python生成动态库在c++中调用

python生成动态库在c++中调用

一.Windows下生成动态库.pyd 在setup.py的同目录下使用python setup.py build_ext --inplace 二.在vscode的c中使用.pyd文件&#xff08;动态库&#xff09; 1&#xff09;配置python的环境 python -c "import sys; print(sys.executable)" #确定python安装位置 2…
阅读更多...
OpenCV基础函数学习4

OpenCV基础函数学习4

【大纲笔记见附件pdf】 目录 一、基于OpenCV的形态学操作 二、基于OpenCV的直方图处理 三、基于OpenCV霍夫变换 四、基于OpenCV模板匹配 一、基于OpenCV的形态学操作 二、基于OpenCV的直方图处理 三、基于OpenCV霍夫变换 四、基于OpenCV模板匹配
阅读更多...
Nginx反向代理用自定义Header参数

Nginx反向代理用自定义Header参数

【啰嗦两句】 也不知道为啥&#xff0c;我仅仅想在Nginx的反向代理中使用自己定义的“x-api-key”做Header参数&#xff0c;却发现会被忽略&#xff0c;网上搜的资料都是说用“proxy_set_header”&#xff0c;却只愿意介绍最基本的几个参数&#xff0c;你懂的&#xff0c;那些资…
阅读更多...
详解机器学习各算法的优缺点！！

详解机器学习各算法的优缺点！！

在机器学习这个 “工具库” 里&#xff0c;算法就像各种各样的工具&#xff0c;每一种都有自己的 “脾气” 和 “特长”。有些算法擅长找规律&#xff0c;有些算法能快速分类&#xff0c;还有些在处理复杂数据时特别厉害。 而且&#xff0c;就像锤子适合敲钉子、螺丝刀适合拧螺…
阅读更多...
推荐文章
最新文章

Copyright @ 2022~2023