Go语言的context包从放弃到入门
以下内容由伟大的诗人、哲学家chenqionghe吐血传授,希望能帮助到你,谢谢~
一、Context包到底是干嘛用的
我们会在用到很多东西的时候都看到context的影响,比如grpc,这东西到底是做啥的?
大家都在用,没几个知道这是干嘛的,知其然而不知其所以然,
谁都在CRUD,谁都觉得if else就完了,有代码能copy我也行,原理啥啥不懂不重要,反正就是一把梭
原理说白了就是:
- 当前协程取消了,可以通知所有由由它的子协程退出
- 当前协程取消了,不会影响到他父级协程的状态
- 扩展了一些其他的功能:超时取消、定时取消、给以给子协程传递数据
二、主协程退出通知子协程示例演示
主协程通知子协程退出
如下代码展示了,通过一个叫done的channel通道达到了这样的效果
package main
import (
"fmt"
"time"
)
func main() {
done := make(chan string)
//缓冲通道预先放置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, done, messages)
}
time.Sleep(3 * time.Second) //等待子协程接收一半的消息
close(done) //结束前通知子协程
time.Sleep(2 * time.Second) //等待所有的子协程输出
fmt.Println("主协程结束")
}
//从messages通道获取信息,当收到结束信号的时候不再接收
func child(i int, done <-chan string, messages <-chan int) {
Consume:
for {
time.Sleep(1 * time.Second)
select {
case <-done:
fmt.Printf("[%d]被主协程通知结束...\n", i)
break Consume
default:
fmt.Printf("[%d]接收消息: %d\n", i, <-messages)
}
}
}
运行结束如下
这里,我们用一个channel的关闭做到了通知所有的消费到一半的子协程退出。
问题来了,如果子协程又要启动它的子协程,这可咋整?
主协程通知有子协程,子协程又有多个子协程
这是可哲学问题,我们还是得建立一个叫done的channel来监测
下面演示一下这种操作,再在每个child方法里启动多个job,如下
全量代码贴出来
package main
import (
"fmt"
"time"
)
func main() {
done := make(chan string)
//缓冲通道预先放置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, done, messages)
}
time.Sleep(3 * time.Second) //等待子协程接收一半的消息
close(done) //结束前通知子协程
time.Sleep(2 * time.Second) //等待所有的子协程输出
fmt.Println("主协程结束")
}
//从messages通道获取信息,当收到结束信号的时候不再接收
func child(i int, done <-chan string, messages <-chan int) {
newDone := make(chan string)
defer close(newDone)
go childJob(i, "a", newDone)
go childJob(i, "b", newDone)
Consume:
for {
time.Sleep(1 * time.Second)
select {
case <-done:
fmt.Printf("[%d]被主协程通知结束...\n", i)
break Consume
default:
fmt.Printf("[%d]接收消息: %d\n", i, <-messages)
}
}
}
//任务
func childJob(parent int, name string, done <-chan string) {
for {
time.Sleep(1 * time.Second)
select {
case <-done:
fmt.Printf("[%d-%v]被结束...\n", parent, name)
return
default:
fmt.Printf("[%d-%v]执行\n", parent, name)
}
}
}
运行结果如下
问题来了,如果job里再启动自己的goroutine,这样没完没了的建立done的通道有点恶心,这时候context包就来了!
我们先把上面的代码改成context包的方式
package main
import (
"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)
}
}
}
运行结果如下
可以看到,改成context包还是顺利的通过子协程退出了
主要修改了几个地方,再ctx向下传递
基于上层context再构建当前层级的context
监听context的退出信号,
这就是context包的核心原理,链式传递context,基于context构造新的context
三、Context包的核心接口和方法
context接口
context是一个接口,主要包含以下4个方法
- Deadline
返回当前context任务被取消的时间,没有设定返回ok返回false - Done
当绑定当前的context任务被取消时,将返回一个关闭的channel - Err
Done返回的channel没有关闭,返回nil;
Done返回的channel已经关闭,返回非空值表示任务结束的原因;
context被取消,返回Canceled。
context超时,DeadlineExceeded - Value
返回context存储的键
emptyCtx结构体
实现了context接口,emptyCtx没有超时时间,不能取消,也不能存储额外信息,所以emptyCtx用来做根节点,一般用Background和TODO来初始化emptyCtx
Backgroud
通常用于主函数,初始化以及测试,作为顶层的context
TODO
不确定使用什么用context的时候才会使用
valueCtx结构体
type valueCtx struct{ Context key, val interface{} }
valueCtx利用Context的变量来表示父节点context,所以当前context继承了父context的所有信息
valueCtx还可以存储键值。
Value
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对应的值,直到根节点
WithValue
可以向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 &valueCtx{parent, key, val}
}
添加键值会返回创建一个新的valueCtx子节点
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() <-chan struct{}
}
和valueCtx类似,有一个context做为父节点,
变量done表示一个channel,用来表示传递关闭;
children表示一个map,存储了当前context节点为下的子节点
err用来存储错误信息表示任务结束的原因
WithCancel
用来创建一个可取消的context,返回一个context和一个CancelFunc,调用CancelFunc可以触发cancel操作。
timerCtx结构体
timerCtx是基于cancelCtx的context精英,是一种可以定时取消的context,过期时间的deadline不晚于所设置的时间d
WithDeadline
返回一个基于parent的可取消的context,并且过期时间deadline不晚于所设置时间d
WithTimeout
创建一个定时取消context,和WithDeadline差不多,WithTimeout是相对时间
四、总结核心原理
- Done方法返回一个channel
- 外部通过调用<-channel监听cancel方法
- cancel方法会调用close(channel)
当调用close方法的时间,所有的channel再次从通道获取内容,会返回零值和false
如
res,ok := <-done:
- 过期自动取消,相当于返回了一个函数,到时调用cancel方法
如下
c.timer = time.AfterFunc(dur, func() {
c.cancel(true, DeadlineExceeded)
})
授人以渔不如授人以渔,知其然也知其所以然,让我们共同构建美丽新世界,让人与自然更加和谐,就是这样,giao~
Go语言的context包从放弃到入门的更多相关文章
- 深度解密Go语言之context
目录 什么是 context 为什么有 context context 底层实现原理 整体概览 接口 Context canceler 结构体 emptyCtx cancelCtx timerCtx ...
- Go语言 context包源码学习
你必须非常努力,才能看起来毫不费力! 微信搜索公众号[ 漫漫Coding路 ],一起From Zero To Hero ! 前言 日常 Go 开发中,Context 包是用的最多的一个了,几乎所有函数 ...
- Golang Context 包详解
Golang Context 包详解 0. 引言 在 Go 语言编写的服务器程序中,服务器通常要为每个 HTTP 请求创建一个 goroutine 以并发地处理业务.同时,这个 goroutine 也 ...
- golang context包
go context标准库 context包在Go1.7版本时加入到标准库中.其设计目标是给Golang提供一个标准接口来给其他任务发送取消信号和传递数据.其具体作用为: 可以通过context发送取 ...
- R语言︱H2o深度学习的一些R语言实践——H2o包
每每以为攀得众山小,可.每每又切实来到起点,大牛们,缓缓脚步来俺笔记葩分享一下吧,please~ --------------------------- R语言H2o包的几个应用案例 笔者寄语:受启发 ...
- Go语言基础之包
Go语言基础之包 在工程化的Go语言开发项目中,Go语言的源码复用是建立在包(package)基础之上的.本文介绍了Go语言中如何定义包.如何导出包的内容及如何导入其他包. Go语言的包(packag ...
- go使用context包避免goroutine泄露问题
go是带内存自动回收的特性,因此内存一般不会泄漏.但是Goroutine确存在泄漏的情况,同时泄漏的Goroutine引用的内存同样无法被回收. 下面的程序中后台Goroutine向管道输入自然数序列 ...
- golang中的context包
标准库的context包 从设计角度上来讲, golang的context包提供了一种父routine对子routine的管理功能. 我的这种理解虽然和网上各种文章中讲的不太一样, 但我认为基本上还是 ...
- R语言:recommenderlab包的总结与应用案例
R语言:recommenderlab包的总结与应用案例 1. 推荐系统:recommenderlab包整体思路 recommenderlab包提供了一个可以用评分数据和0-1数据来发展和测试推荐算 ...
随机推荐
- Pandas_基础_全
Pandas基础(全) 引言 Pandas是基于Numpy的库,但功能更加强大,Numpy专注于数值型数据的操作,而Pandas对数值型,字符串型等多种格式的表格数据都有很好的支持. 关于Numpy的 ...
- 也谈模块加载,吐槽CMD
先吐槽CMD,不要没头没脑的搞出个CMD,没意思. 大家都看AMD好了,异步模块加载机制,CMD并没有改变这个模式. 模块加载的关口就是getCurrentScript,每次define被调用的时候, ...
- Java 内功修炼 之 数据结构与算法(二)
一.二叉树补充.多叉树 1.二叉树(非递归实现遍历) (1)前提 前面一篇介绍了 二叉树.顺序二叉树.线索二叉树.哈夫曼树等树结构. 可参考:https://www.cnblogs.com/l-y-h ...
- golang 简单工厂模式
package kit //golang简单工厂模式 //go 语言没有构造函数一说,所以一般会定义NewXXX函数来初始化相关类. NewXXX 函数返回接口时就是简单工厂模式,也就是说Golang ...
- Java编码和字符集(详解)
[1]什么是编码? [2]通过生活案例: [3]由权威机构形成的编码表才可以称之为:字符集 ASCII 英文字符集 用一个字节的7位表示 IOS8859-1 西欧字符集 用一个字节的8位表示 GB23 ...
- 循序渐进VUE+Element 前端应用开发(27)--- 数据表的动态表单设计和数据存储
在我们一些系统里面,有时候会需要一些让用户自定义的数据信息,一般这些可以使用扩展JSON进行存储,不过每个业务表的显示项目可能不一样,因此需要根据不同的表单进行设计,然后进行对应的数据存储.本篇随笔结 ...
- mysql 不常用备忘
# group_concat 函数语法: group_concat( [DISTINCT] 要连接的字段 [Order BY 排序字段 ASC/DESC] [Separator '分隔符'] ...
- bWAPP----HTML OS Command Injection - Blind
OS Command Injection - Blind 先上代码,他判断了win还是linux然后进行了ping但是结果并没有返回. 1 <div id="main"> ...
- 一遍记住Java常用的八种排序算法
1.直接插入排序 经常碰到这样一类排序问题:把新的数据插入到已经排好的数据列中. 将第一个数和第二个数排序,然后构成一个有序序列 将第三个数插入进去,构成一个新的有序序列. 对第四个数.第五个数--直 ...
- Codeforces1009F Dominant Indices
dsu on tree 题目链接 点我跳转 题目大意 给定一棵以 \(1\) 为根,\(n\) 个节点的树.设\(d(u,x)\) 为 \(u\) 子树中到 \(u\) 距离为 \(x\) 的节点数. ...