我的编程空间,编程开发者的网络收藏夹
学习永远不晚

go语言context包功能及操作使用的方法

短信预约 -IT技能 免费直播动态提醒
省份

北京

  • 北京
  • 上海
  • 天津
  • 重庆
  • 河北
  • 山东
  • 辽宁
  • 黑龙江
  • 吉林
  • 甘肃
  • 青海
  • 河南
  • 江苏
  • 湖北
  • 湖南
  • 江西
  • 浙江
  • 广东
  • 云南
  • 福建
  • 海南
  • 山西
  • 四川
  • 陕西
  • 贵州
  • 安徽
  • 广西
  • 内蒙
  • 西藏
  • 新疆
  • 宁夏
  • 兵团
手机号立即预约

请填写图片验证码后获取短信验证码

看不清楚,换张图片

免费获取短信验证码

go语言context包功能及操作使用的方法

本篇内容介绍了“go语言context包功能及操作使用的方法”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!

Context包到底是干嘛用的?

我们会在用到很多东西的时候都看到context的影子,比如gin框架,比如grpc,这东西到底是做啥的?
大家都在用,没几个知道这是干嘛的,知其然而不知其所以然

原理说白了就是:

  • 当前协程取消了,可以通知所有由它创建的子协程退出

  • 当前协程取消了,不会影响到创建它的父级协程的状态

  • 扩展了额外的功能:超时取消、定时取消、可以和子协程共享数据

context原理

这就是context包的核心原理,链式传递context,基于context构造新的context

什么时候应该使用 Context?

  • 每一个 RPC 调用都应该有超时退出的能力,这是比较合理的 API 设计

  • 不仅仅 是超时,你还需要有能力去结束那些不再需要操作的行为

  • context.Context 是 Go 标准的解决方案

  • 任何函数可能被阻塞,或者需要很长时间来完成的,都应该有个 context.Context

如何创建 Context?

在 RPC 开始的时候,使用 context.Background()

有些人把在 main() 里记录一个 context.Background(),然后把这个放到服务器的某个变量里,然后请求来了后从这个变量里继承 context。这么做是不对的。直接每个请求,源自自己的 context.Background() 即可。

如果你没有 context,却需要调用一个 context 的函数的话,用 context.TODO()

如果某步操作需要自己的超时设置的话,给它一个独立的 sub-context(如前面的例子)

主协程通知有子协程,子协程又有多个子协程

package mainimport (    "context"    "fmt"    "time")func main() {    ctx, cancel := context.WithCancel(context.Background())    //缓冲通道预先放置10个消息    messages := make(chan int, 10)    defer close(messages)    for i := 0; i < 10; i++ {        messages <- i    }    //启动3个子协程消费messages消息    for i := 1; i <= 3; i++ {        go child(i, ctx, messages)    }    time.Sleep(3 * time.Second) //等待子协程接收一半的消息    cancel() //结束前通知子协程    time.Sleep(2 * time.Second) //等待所有的子协程输出    fmt.Println("主协程结束")}//从messages通道获取信息,当收到结束信号的时候不再接收func child(i int, ctx context.Context, messages <-chan int) {    //基于父级的context建立context    newCtx, _ := context.WithCancel(ctx)    go childJob(i, "a", newCtx)    go childJob(i, "b", newCtx)Consume:    for {        time.Sleep(1 * time.Second)        select {        case <-ctx.Done():            fmt.Printf("[%d]被主协程通知结束...\n", i)            break Consume        default:            fmt.Printf("[%d]接收消息: %d\n", i, <-messages)        }    }}//任务func childJob(parent int, name string, ctx context.Context) {    for {        time.Sleep(1 * time.Second)        select {        case <-ctx.Done():            fmt.Printf("[%d-%v]被结束...\n", parent, name)            return        default:            fmt.Printf("[%d-%v]执行\n", parent, name)        }    }}

运行结果如下

go语言context包功能及操作使用的方法

可以看到,改成context包还是顺利的通过子协程退出了
主要修改了几个地方,再ctx向下传递

go语言context包功能及操作使用的方法

基于上层context再构建当前层级的context

go语言context包功能及操作使用的方法

监听context的退出信号,

go语言context包功能及操作使用的方法

这就是context包的核心原理,链式传递context,基于context构造新的context

context核心接口

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

 Deadline返回绑定当前context的任务被取消的截止时间;如果没有设定期限,将返回ok == false。

Done 当绑定当前context的任务被取消时,将返回一个关闭的channel;如果当前context不会被取消,将返回nil。

Err 如果Done返回的channel没有关闭,将返回nil;如果Done返回的channel已经关闭,将返回非空的值表示任务结束的原因。如果是context被取消,Err将返回Canceled;如果是context超时,Err将返回DeadlineExceeded。

Value 返回context存储的键值对中当前key对应的值,如果没有对应的key,则返回nil。

emptyCtx结构体

实现了context接口,emptyCtx没有超时时间,不能取消,也不能存储额外信息,所以emptyCtx用来做根节点,一般用Background和TODO来初始化emptyCtx

Backgroud

通常用于主函数,初始化以及测试,作为顶层的context

context.Background()

TODO

不确定使用什么用context的时候才会使用

valueCtx结构体

type valueCtx struct{ Context key, val interface{} }

valueCtx利用Context的变量来表示父节点context,所以当前context继承了父context的所有信息
valueCtx还可以存储键值。

WithValue向context添加值

可以向context添加键值

func WithValue(parent Context, key, val interface{}) Context {    if key == nil {        panic("nil key")    }    if !reflect.TypeOf(key).Comparable() {        panic("key is not comparable")    }    return &amp;valueCtx{parent, key, val}}

添加键值会返回创建一个新的valueCtx子节点

Value向context取值

func (c *valueCtx) Value(key interface{}) interface{} {    if c.key == key {        return c.val    }    return c.Context.Value(key)}

可以用来获取当前context和所有的父节点存储的key

如果当前的context不存在需要的key,会沿着context链向上寻找key对应的值,直到根节点

示例

package mainimport ("context""fmt""time")func main() {ctx := context.WithValue(context.Background(), "name1", "root1")//第一层go func(parent context.Context) {ctx = context.WithValue(parent, "name2", "root2")//第二层go func(parent context.Context) {ctx = context.WithValue(parent, "name3", "root3")//第三层go func(parent context.Context) {//可以获取所有的父类的值fmt.Println(ctx.Value("name1"))fmt.Println(ctx.Value("name2"))fmt.Println(ctx.Value("name3"))//不存在fmt.Println(ctx.Value("name4"))}(ctx)}(ctx)}(ctx)time.Sleep(1 * time.Second)fmt.Println("end")}

运行结果

go语言context包功能及操作使用的方法

可以看到,子context是可以获取所有父级设置过的key

WithCancel可取消的context

用来创建一个可取消的context,返回一个context和一个CancelFunc,调用CancelFunc可以触发cancel操作。

package mainimport ("context""fmt""time")func main() {ctx, cancel := context.WithCancel(context.Background())//第一层go func(parent context.Context) {ctx, _ := context.WithCancel(parent)//第二层go func(parent context.Context) {ctx, _ := context.WithCancel(parent)//第三层go func(parent context.Context) {waitCancel(ctx, 3)}(ctx)waitCancel(ctx, 2)}(ctx)waitCancel(ctx, 1)}(ctx)// 主线程给的结束时间time.Sleep(2 * time.Second)cancel() // 调用取消contexttime.Sleep(1 * time.Second)}func waitCancel(ctx context.Context, i int) {for {time.Sleep(time.Second)select {case <-ctx.Done():fmt.Printf("%d end\n", i)returndefault:fmt.Printf("%d do\n", i)}}}

结果:

go语言context包功能及操作使用的方法

cancelCtx结构体

type cancelCtx struct {    Context    mu sync.Mutex    done chan struct{}    children map[canceler]struct{}    err error}type canceler interface {    cancel(removeFromParent bool, err error)    Done() &lt;-chan struct{}}

WithDeadline-超时取消context

返回一个基于parent的可取消的context,并且过期时间deadline不晚于所设置时间d

WithTimeout-超时取消context

创建一个定时取消context,和WithDeadline差不多,WithTimeout是相对时间

timerCtx结构体

timerCtx是基于cancelCtx的context精英,是一种可以定时取消的context,过期时间的deadline不晚于所设置的时间d

示例:

package mainimport ("context""fmt""time")func main() {// 设置超时时间ctx, _ := context.WithTimeout(context.Background(), 2*time.Second)//第一层go func(parent context.Context) {ctx, _ := context.WithCancel(parent)//第二层go func(parent context.Context) {ctx, _ := context.WithCancel(parent)//第三层go func(parent context.Context) {waitCancel(ctx, 3)}(ctx)waitCancel(ctx, 2)}(ctx)waitCancel(ctx, 1)}(ctx)<-ctx.Done()// 给时间调用endtime.Sleep(time.Second)}func waitCancel(ctx context.Context, i int) {for {time.Sleep(time.Second)select {case <-ctx.Done():fmt.Printf("%d end\n", i)returndefault:fmt.Printf("%d do\n", i)}}}

运行结果:

1 do3 do2 do1 end3 end2 end

可以看到,虽然我们没有调用cancel方法,5秒后自动调用了,所有的子goroutine都已经收到停止信号

总结核心原理

  • Done方法返回一个channel

  • 外部通过调用<-channel监听cancel方法

  • cancel方法会调用close(channel)
    当调用close方法的时候,所有的channel再次从通道获取内容,会返回零值和false

res,ok := &lt;-done:
  • 过期自动取消,使用了time.AfterFunc方法,到时调用cancel方法

  c.timer = time.AfterFunc(dur, func() {   c.cancel(true, DeadlineExceeded)  })

“go语言context包功能及操作使用的方法”的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识可以关注编程网网站,小编将为大家输出更多高质量的实用文章!

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

go语言context包功能及操作使用的方法

下载Word文档到电脑,方便收藏和打印~

下载Word文档

猜你喜欢

go语言context包功能及操作使用的方法

本篇内容介绍了“go语言context包功能及操作使用的方法”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!Context包到底是干嘛用的?我
2023-06-30

Python语言功能中的宏编程语言的操作方法

这篇文章给大家介绍Python语言功能中的宏编程语言的操作方法,内容非常详细,感兴趣的小伙伴们可以参考借鉴,希望对大家能有所帮助。Python语言之所以被广泛的应用,其中的主要原因是因为Python语言的功能就有简捷,快速,相关性能好等特点
2023-06-17

Go语言time包的时间常用操作方法有哪些

本篇内容主要讲解“Go语言time包的时间常用操作方法有哪些”,感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷,实用性强。下面就让小编来带大家学习“Go语言time包的时间常用操作方法有哪些”吧!Now():获取当前本地的时间impor
2023-07-04

Rxjava功能操作符的使用方法详解

Rxjava功能个人感觉很好用,里面的一些操作符很方便,Rxjava有:被观察者,观察者,订阅者,被观察者通过订阅者订阅观察者,从而实现观察者监听被观察者返回的数据下面把Rxjava常用的模型代码列出来,还有一些操作符的运用:依赖:comp
2023-05-30

Go语言Slice作为函数参数的使用方法

这篇文章主要讲解了“Go语言Slice作为函数参数的使用方法”,文中的讲解内容简单清晰,易于学习与理解,下面请大家跟着小编的思路慢慢深入,一起来研究和学习“Go语言Slice作为函数参数的使用方法”吧!目录前言问题与解析典型问题其它疑问1其
2023-06-20

c语言中回调函数的使用方法及作用是什么

本篇内容介绍了“c语言中回调函数的使用方法及作用是什么”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!目录前言一、通过这节课程你能掌握以下知识
2023-06-20

win7系统中文语言包下载及使用方法的示例分析

这篇文章给大家分享的是有关win7系统中文语言包下载及使用方法的示例分析的内容。小编觉得挺实用的,因此分享给大家做个参考,一起跟随小编过来看看吧。具体教程:1.Win7中文语言包下载(Win7 64位):Chinese (Simplifie
2023-06-27

使用路由器功能实现主机跨网访问的操作方法

  路由器是互联网络中必不可少的网络设备之一,路由器是一种连接多个网络或网段的网络设备,它能将不同网络或网段之间的数据信息进行"翻译",以使它们能够相互"读"懂对方的数据,从而构成一个更大的网络,使整个网络互通数据,提高数据传输效率。在这里,小编为大家分享的教程是:使用路由器功能实现主机跨网访问的操作方法。  我们应该
使用路由器功能实现主机跨网访问的操作方法
2024-04-18

编程热搜

  • Python 学习之路 - Python
    一、安装Python34Windows在Python官网(https://www.python.org/downloads/)下载安装包并安装。Python的默认安装路径是:C:\Python34配置环境变量:【右键计算机】--》【属性】-
    Python 学习之路 - Python
  • chatgpt的中文全称是什么
    chatgpt的中文全称是生成型预训练变换模型。ChatGPT是什么ChatGPT是美国人工智能研究实验室OpenAI开发的一种全新聊天机器人模型,它能够通过学习和理解人类的语言来进行对话,还能根据聊天的上下文进行互动,并协助人类完成一系列
    chatgpt的中文全称是什么
  • C/C++中extern函数使用详解
  • C/C++可变参数的使用
    可变参数的使用方法远远不止以下几种,不过在C,C++中使用可变参数时要小心,在使用printf()等函数时传入的参数个数一定不能比前面的格式化字符串中的’%’符号个数少,否则会产生访问越界,运气不好的话还会导致程序崩溃
    C/C++可变参数的使用
  • css样式文件该放在哪里
  • php中数组下标必须是连续的吗
  • Python 3 教程
    Python 3 教程 Python 的 3.0 版本,常被称为 Python 3000,或简称 Py3k。相对于 Python 的早期版本,这是一个较大的升级。为了不带入过多的累赘,Python 3.0 在设计的时候没有考虑向下兼容。 Python
    Python 3 教程
  • Python pip包管理
    一、前言    在Python中, 安装第三方模块是通过 setuptools 这个工具完成的。 Python有两个封装了 setuptools的包管理工具: easy_install  和  pip , 目前官方推荐使用 pip。    
    Python pip包管理
  • ubuntu如何重新编译内核
  • 改善Java代码之慎用java动态编译

目录