【Go - context 速览,场景与用法】

作用

context字面意思上下文,用于关联管理上下文,具体有如下几个作用

  1. 取消信号传递:可以用来传递取消信号,让一个正在执行的函数知道它应该提前终止。
  2. 超时控制:可以设定一个超时时间,自动取消超过执行时间的操作。
  3. 截止时间:与超时类似,但是是设定一个绝对时间点,而不是时间段。
  4. 值传递:可以安全地在请求的上下文中传递数据,避免了使用全局变量或者参数列表不断增长。

由上述看出,context有个重要用途,控制取消

场景与用法

示例1:HTTP 请求处理

在处理 HTTP 请求时,可以为每个请求创建一个 context,用于控制请求处理的整个生命周期。如果请求被取消或超时,依赖该 context 的操作也会被取消。

代码示例

package mainimport ("context""fmt""math/rand""net/http""time"
)func randomSleepAtMost2s() {rand.Seed(time.Now().UnixNano()) // 初始化随机数种子// 生成 0 到 2000 之间的随机整数(毫秒)randomMillis := rand.Intn(2000)// 转换为 time.Duration 类型,并乘以 time.MillisecondsleepDuration := time.Duration(randomMillis) * time.Millisecond// 随机睡眠fmt.Println("sleeping for", sleepDuration)time.Sleep(sleepDuration)
}func handler(w http.ResponseWriter, r *http.Request) {// 超时时间1sctx, cancel := context.WithTimeout(r.Context(), 1*time.Second)defer cancel()// 创建一个模拟正常处理完成的通道done := make(chan struct{})// 模拟异步处理逻辑go func() {// 模拟耗时操作// - 当随机睡眠超过1s时,会触发 ctx.Done(),取消请求// - 当随机睡眠不超过1s时,则会正常处理请求randomSleepAtMost2s()fmt.Println("request processed")close(done) // 处理完成,关闭通道}()// 模拟耗时操作select {case <-done:w.WriteHeader(http.StatusOK)w.Write([]byte("request processed successfully"))case <-ctx.Done():fmt.Println("request cancelled")http.Error(w, "request cancelled", http.StatusRequestTimeout)return}
}func main() {http.HandleFunc("/", handler)http.ListenAndServe(":8080", nil)
}

示例2 :数据库操作

数据库查询或操作可以接受一个 context 参数,允许查询在超时或被取消时立即停止,避免无谓的数据库资源占用。

代码示例

package mainimport ("context""database/sql""fmt""time"_ "github.com/lib/pq" // 假设使用 PostgreSQL
)func queryWithTimeout(ctx context.Context, db *sql.DB, query string) (*sql.Rows, error) {// 设置1s的超时时间ctx, cancel := context.WithTimeout(ctx, 1*time.Second)defer cancel()return db.QueryContext(ctx, query)
}func main() {// 连接数据库(示例)db, err := sql.Open("postgres", "your_connection_string")if err != nil {panic(err)}defer db.Close()ctx := context.Background()_, err = queryWithTimeout(ctx, db, "SELECT * FROM your_table")if err != nil {fmt.Println("Query failed:", err)}
}

示例3:取消协程

在启动多个 goroutine 进行并发操作时,可以通过 context 控制这些 goroutine 的生命周期,确保它们能够在必要时被正确取消。

代码示例

package mainimport ("context""fmt""time"
)func operation(ctx context.Context, id int) {select {case <-time.After(2 * time.Second):fmt.Printf("Operation %d completed\n", id)case <-ctx.Done():fmt.Printf("Operation %d cancelled\n", id)}
}func main() {ctx, cancel := context.WithCancel(context.Background())for i := 0; i < 5; i++ {go operation(ctx, i)}time.Sleep(1 * time.Second)cancel()                    // 取消所有协程操作time.Sleep(3 * time.Second) // 等待足够的时间以打印完日志,观察效果
}

示例4:跨服务调用

在微服务架构中,一个服务调用另一个服务时,可以通过 context 传递关于原始请求的信息,如请求ID,以便进行链路追踪。

代码示例

package mainimport ("context""fmt""net/http""time"
)func callService(ctx context.Context, url string) {req, _ := http.NewRequest("GET", url, nil)req = req.WithContext(ctx)client := &http.Client{}resp, err := client.Do(req)if err != nil {fmt.Println("Request failed:", err)return}defer resp.Body.Close()fmt.Println("Response status:", resp.Status)
}func main() {ctx := context.Background()ctx = context.WithValue(ctx, "RequestID", "abc123")ctx, cancel := context.WithTimeout(ctx, 1*time.Second)defer cancel()callService(ctx, "http://example.com")
}

结语

通过这些场景和用法,可以看出 context 在 Go 中的重要性,特别是在需要控制管理请求生命周期时(控制取消)。

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

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

Swift学习入门,新手小白看过来

&#x1f604;作者简介&#xff1a; 小曾同学.com,一个致力于测试开发的博主⛽️&#xff0c;主要职责&#xff1a;测试开发、CI/CD 如果文章知识点有错误的地方&#xff0c;还请大家指正&#xff0c;让我们一起学习&#xff0c;一起进步。 &#x1f60a; 座右铭&#xff1a;不…

文本分类动转静预测错误分析和挖掘稀疏数据和建立新数据集.ipynb

import os import paddle from paddlenlp.transformers import AutoModelForSequenceClassification params_pathcheckpoint/text_classes/ output_pathoutput/text_class model AutoModelForSequenceClassification.from_pretrained(params_path) model.eval() # 转换为具…

(十三)Spring教程——依赖注入之工厂方法注入

1.工厂方法注入 工厂方法是在应用中被经常使用的设计模式&#xff0c;它也是控制反转和单例设计思想的主要实现方法。由于Spring IoC容器以框架的方式提供工厂方法的功能&#xff0c;并以透明的方式开放给开发者&#xff0c;所以很少需要手工编写基于工厂方法的类。正是因为工厂…

如何从网站获取表格数据

1.手动复制粘贴 最简单的方法是直接在网页上手动选择表格内容&#xff0c;然后复制粘贴到Excel或其他表格处理软件中。这种方法适用于表格较小且不经常更新的情况。 2.使用浏览器插件 有许多浏览器插件可以帮助从网页中提取表格数据&#xff0c;例如&#xff1a; -TableCapt…

SSRF过滤攻击

SSRF绕过&#xff1a; 靶场地址&#xff1a;重庆橙子科技SSRF靶场 这个是毫无过滤的直接读取&#xff0c;但是一般网站会设置有对SSRF的过滤&#xff0c;比如将IP地址过滤。 下面是常用的绕过方式&#xff1a; 1.环回地址绕过 http://127.0.0.1/flag.php http://017700…

相机怎么选(不推荐,只分析)

title: 相机怎么选 tags: [相机, 单反相机] categories: [其他, 相机] 最近准备购买&#xff0c;相机怎么选&#xff0c;我去搜索了许多文章&#xff0c;整理了一篇小白挑选技术篇&#xff0c;供大家参考。 分类 胶片相机 需要装入胶卷才能使用的相机&#xff0c;拍照后可直…

永磁同步电机无速度算法--非线性磁链观测器

非线性磁链观测器顾名思义观测器的状态变量为磁链值&#xff0c;观测的磁链值收敛于电机实际磁链值&#xff0c;观测器收敛。非线性是由于观测器存在sin和cos项&#xff0c;所以是非线性观测器 一、原理介绍 表贴式永磁同步电机αβ轴电压方程: 将公式变换 定义状态变量X: 定…

easy-ui nowrap

​​easy-ui​​ 是一个基于 jQuery 的前端框架&#xff0c;用于构建现代化的 Web 应用程序。它提供了丰富的组件和功能&#xff0c;简化了 Web 应用的开发。 ​​nowrap​​ 是 ​​easy-ui​​ 中的一个属性&#xff0c;用于控制表格列&#xff08;或其他容器&#xff09;中…

大模型日报 2024-07-28

大模型日报 2024-07-28 大模型资讯 Meta发布全球最大最强的开源AI模型Llama 3.1 摘要: Meta公司宣布其最新的开源AI模型Llama 3.1具备最先进的能力&#xff0c;被称为全球最大和最强的AI模型。 阿里巴巴和腾讯迅速整合Meta的Llama 3.1 AI模型 摘要: 阿里巴巴和腾讯迅速将Meta的…

什么是数据挖掘?为什么要进行数据挖掘?

数据挖掘可以表达为“从数据中挖掘知识”&#xff0c;或者“知识提取”&#xff0c;亦或是“数据/模式分析”等。 也有一些人把之称为“数据中的知识发现&#xff08;KDD&#xff09;”。 我们写论文时很常用的一句开头语就时“随着信息时代的发展”&#xff0c;其实&#xff…

Milvus 向量数据库进阶系列丨部署形态选型

本系列文章介绍 在和社区小伙伴们交流的过程中&#xff0c;我们发现大家最关心的问题从来不是某个具体的功能如何使用&#xff0c;而是面对一个具体的实战场景时&#xff0c;如何选择合适的向量数据库解决方案或最优的功能组合。在 “Milvus 向量数据库进阶” 这个系列文章中&…

Java实现打印功能

JPG图片文件格式打印实现 打印JPG图片格式的文件&#xff0c;本次采用的Java原生的打印方式。 public static void main(String[] argv) throws Exception {File file new File("E:\\a.jpg");String printerName "HP MFP M436 PCL6";//打印机名包含字串…

kafka 首选读副本

一、kafka 首选读副本是哪个版本开始加入的&#xff1f;作用是什么 Kafka的首选读副本&#xff08;Preferred Read Replica&#xff09;功能是在2.4版本中引入的。这个功能的主要作用是允许客户端从指定的副本而不是分区的领导者&#xff08;Leader&#xff09;读取数据&#…

vite+vue3项目,开发时候正常,打包后router-view不渲染

这是个很奇怪的问题&#xff0c;但是基本上命名或者引入文件的方式导致的。要么文件名与系统的某些标签名一样&#xff0c;要么就是routes写成了routers。还有一种就是导入方式错误的 错误截图&#xff1a; 正确引入截图&#xff1a;

Flink大状态作业调优——DataStream篇

一、Flink 状态&#xff08;State&#xff09;简介 在流式计算中有些操作一次处理一个独立的事件(比如解析一个事件), 有些操作却需要记住多个事件的信息(比如窗口操作)。那些需要记住多个事件信息的操作就是有状态的。流式计算分为无状态计算和有状态计算两种情况。状态可以理…

Word题库转Excel关键字快速查询题库

一、前言 内部培训会有Word版本题库&#xff0c;考核时如果使用Word、Excel、PDF等文档进行关键字查询题目体验不佳。so&#xff0c;撸个软件吧&#xff01;   20240728更新&#xff1a;支持更多题库类型。 二、Word题库转Excel 1、Word题库格式要求 内容格式要求事例题目…

在 Postman 中设置全局 token

目录 问题描述解决方案 问题描述 在使用 Postman 进行接口测试时&#xff0c;经常会遇到在 Header 中添加 token 的情况。当接口数量较多时&#xff0c;需要为每个接口进行设置&#xff0c;而且当 token 失效时需要重新获取并设置&#xff0c;这样一来效率较低。 解决方案 下…

【开发学习笔记】什么是Springboot?

Spring Boot是由Pivotal团队提供的基于Spring的全新框架&#xff0c;其设计目的是用来简化新Spring应用的初始搭建以及开发过程。Spring Boot通过特定的配置方式&#xff0c;减少了开发人员需要定义的样板化配置&#xff0c;从而提高了开发效率。以下是对Spring Boot的详细解释…

wpf中开发独立模块功能和左侧1个菜单的框架演示

此篇文章适用于有一定经验的wpf开发者&#xff0c;并且是团队协作开发模式&#xff0c;并且业务与此比较相近的话&#xff0c;用起来非常的方便。 我们在开发wpf程序的时候&#xff0c;比如1&#xff0c;2个人&#xff0c;那肯定随便怎么开发&#xff0c;都没事&#xff0c;代码…

Matlab M_map工具箱绘制Interrupted Mollweide Projection

GMT自带了许多的地图投影&#xff0c;但是对于Interrupted Mollweide投影效果却不好。 作为平替的m_map工具箱中带有的投影类型可完美解决这一问题。 Interrupted Mollweide Projection长这样 全球陆地 全球海洋 使用Matlab工具箱m_map展示全球海平面变化的空间分布 addpath(…