Context顾名思义上下文，可用于在整个申请上下文传值以及管制超时，本篇文章次要介绍Context的设计思路，以及根本应用形式。
Context 应用形式
构想有一个Go HTTP服务，在申请的整个解决链路，可能随时须要获取一些公共数据，如用户ID等，怎么办呢？通过参数呗，每个函数的第一个输出参数都是用户ID不就行了，如果再加一个公共数据呢？再加一个参数吗？如果将所有这些公共参数封装成一个构造体呢？这样貌似也能够。
不过Go语言自身就为咱们提供了context.Context（context.valueCtx），其能够存储一个key-value键值对，那不行啊，只能存储一个必定不够啊；怎么办，基于老的context.valueCtx对象再衍生新的context.valueCtx对象，两个context.valueCtx对象各能存储一个key-value键值对，想存储更多的数据，持续衍生就能够了。事例程序如下：
package main
import (
"context"
"fmt"
)
func main() {
ctx := context.Background()
ctx1 := context.WithValue(ctx, "k1", "v1")
ctx2 := context.WithValue(ctx1, "k2", "v2")
fmt.Println(ctx1.Value("k1"))
fmt.Println(ctx1.Value("k2")) //返回nil
fmt.Println(ctx2.Value("k1"))
fmt.Println(ctx2.Value("k2"))
}
// v1 <nil> v1 v2基于context.WithValue函数能够衍生新的context.valueCtx对象，同时传递须要存储的key-value键值对；留神第一个参数须要一个context.Context（这是一个接口，context.valueCtx实现了该接口）对象，context.Background函数可返回一个空的context.valueCtx对象。
仔细观察输入后果，ctx1对象只能获取到k1，ctx2对象能够获取到k1以及k2，因为ctx2对象是基于ctx1对象衍生进去的，也能够说ctx1对象是ctx2对象的父对象，而ctx2.Value既能够获取到本人存储的数据，也能获取到父对象存储的数据。最初值得一提的是，context.valueCtx存储的key-value键值对，类型都是interface{}，所以获取到数据之后个别须要进行类型转换能力应用。
既然通过context.Context就能实现key-value数据的存储，那就应用它呗，只须要我的项目中所有函数的第一个参数都是context.Context，就能实现在整个申请链路传值。
context.Context就这么点作用？当然不是，最罕用的还是它的超时管制性能。假如某项工作有工夫限度，最多执行3秒，超时后勾销工作的继续执行，这不很简略，通过定时器就能实现，那如果工作比较复杂，又分为多个子工作并启动了多个子协程别离执行呢，3秒超时后能同时完结整个工作吗，包含主工作以及多个子工作？这时候定时器能实现吗？比拟艰难。
上面程序展现了基于context.Context实现的工作超时管制。
package main
import (
"context"
"fmt"
"sync"
"time"
)
func main() {
ctx := context.Background()
//可勾销的context
ctx1, cancel := context.WithCancel(ctx)
//WaitGroup管制并发工作，主协程需期待子工作完结能力退出
wg := sync.WaitGroup{}
wg.Add(1)
go task(ctx1, &wg)
//三秒后勾销子工作
time.AfterFunc(time.Second*3, func() {
cancel()
})
//期待子工作完结
wg.Wait()
}
func task(ctx context.Context, wg *sync.WaitGroup) {
defer wg.Done()
for {
//context勾销时会敞开管道，从而可读，实现工作完结return
select {
case <-ctx.Done():
fmt.Println("context cancel and return")
return
default:
}
//子工作1秒周期执行一次，死循环
fmt.Println("exec task", time.Now())
time.Sleep(time.Second)
}
}
context.WithCancel函数返回可勾销的上下文（contetx.cancelCtx对象，同样实现了接口context.Context），函数返回两个值，第一个值就是contetx.cancelCtx对象，第二个值是一个函数，可用于完结该上下文，完结之后ctx.Done函数返回的管道变为可读的，所以子工作能够通过其判断上下文是否被完结，是否该完结当前任务。下面程序事例，定时器3秒超时后完结以后上下文，当然你也能够基于任何条件判断是否须要完结上下文。
超时管制绝对也是比较简单的，只须要主协程管制何时完结上下文，子工作只需监听ctx.Done管道，上下文完结后管道可读，从而完结所有子工作。
contetx.cancelCtx须要你本人基于定时器实现超时管制，Go语言还提供有两个函数，很不便实现context的超时管制：
//timeout之后，主动完结上下文
func WithTimeout(parent Context, timeout time.Duration) (Context, CancelFunc)
func WithDeadline(parent Context, d time.Time) (Context, CancelFunc)这两个函数都是返回新的上下文对象（contetx.timerCtx，同样实现了接口context.Context），函数返回两个值，第一个值就是contetx.timerCtx对象，第二个值是一个函数，同样可用于完结该上下文。也就是说，在timeout之后，会主动完结上下文，然而你也能够通过返回的CancelFunc完结该上下文。
实现原理
在介绍context的应用形式时，咱们提到了几个接口或者构造，如context.Context接口，context.valueCtx构造，contetx.cancelCtx构造，contetx.timerCtx构造，context的所有性能都是基于这几个类型实现的。
context.Context是一个接口，定义了4个根本办法：
type Context interface {
//返回上下文完结工夫，ok=false阐明没有定时完结性能
Deadline() (deadline time.Time, ok bool)
// func Stream(ctx context.Context, out chan<- Value) error {
// for {
// v, err := DoSomething(ctx)
// if err != nil {
// return err
// }
// select {
// case <-ctx.Done():
// return ctx.Err()
// case out <- v:
// }
// }
// }
//返回一个管道，上下文完结时会敞开该管道，从而可读
Done() <-chan struct{}
//如果Done管道敞开则Err不为空，用于示意谬误起因；否则为nil
Err() error
// func NewContext(ctx context.Context, u *User) context.Context {
// return context.WithValue(ctx, userKey, u)
// }
//
// // FromContext returns the User value stored in ctx, if any.
// func FromContext(ctx context.Context) (*User, bool) {
// u, ok := ctx.Value(userKey).(*User)
// return u, ok
// }
// 存取键值对数据
Value(key interface{}) interface{}
}从这四个办法也能看出，context.Context天生就是为了上下文传值以及管制超时的。另外通过下面的几个事例，你有没有发现context.WithXXX等函数，都是基于父context衍生出新的context，也就是说这些context存在父子关系（相似一棵树）；父context超时完结后，也会遍历完结其所有的context（如果子context可完结），顺次类推。
上面咱们先看看context.valueCtx的实现原理，其定义如下：
type valueCtx struct {
//父对象
Context
//存储key-value键值对
key, val any
}
func WithValue(parent Context, key, val any) Context {
return &valueCtx{parent, key, val}
}
func (c *valueCtx) Value(key any) any {
//间接返回
if c.key == key {
return c.val
}
//遍历父节点
return value(c.Context, key)
}
func value(c Context, key any) any {
//循环遍历父节点
for {
switch ctx := c.(type) {
case *valueCtx:
if key == ctx.key {
return ctx.val
}
//接续遍历父节点
c = ctx.Context
case *emptyCtx:
return nil
//省略其余类型context分支
default:
return c.Value(key)
}
}
}context.valueCtx的实现还是特地简略的，只是要留神其存储的key-value键值对类型都是interface{}，所以在获取到值对象后可能须要进行类型转换；另外，如果以后context对象没有获取到键值对，则遍历父对象获取，顺次类推。
contetx.cancelCtx构造用于实现可勾销的context对象，其定义如下：
type cancelCtx struct {
//父对象
Context
//理论类型为管道，敞开管道就是完结以后上下文
done atomic.Value // of chan struct{}, created lazily, closed by first cancel call
//子对象，在父对象完结时完结所有子对象
children map[canceler]struct{} // set to nil by the first cancel call
//上下文完结时，赋值err
err error // set to non-nil by the first cancel call
}这才对嘛，基于这些字段才能够实现context.Context接口，不然怎么实现ctx.Done以及ctx.Err办法呢？contetx.cancelCtx的实现如下：
func WithCancel(parent Context) (ctx Context, cancel CancelFunc) {
c := newCancelCtx(parent)
propagateCancel(parent, &c)
//返回一个函数CancelFunc，用于完结以后context对象
return &c, func() { c.cancel(true, Canceled) }
}
func propagateCancel(parent Context, child canceler) {
done := parent.Done()
select {
case <-done:
// 如果父对象已完结，完结以后context对象并返回
child.cancel(false, parent.Err())
return
default:
}
//判断父对象类型是否为contetx.cancelCtx
if p, ok := parentCancelCtx(parent); ok {
//关联父子context对象
p.children[child] = struct{}{}
} else {
//子协程解决：如果父对象完结，则敞开以后context对象
go func() {
select {
case <-parent.Done():
child.cancel(false, parent.Err())
case <-child.Done():
}
}()
}
}
func (c *cancelCtx) cancel(removeFromParent bool, err error) {
//敞开管道，用于告诉context完结了
d, _ := c.done.Load().(chan struct{})
close(d)
//完结所有子对象
for child := range c.children {
child.cancel(false, err)
}
}基于context.WithCancel衍生新的可勾销context对象时，留神要保护父子context对象之间的关系，而且在父对象完结后，也须要完结其所有子对象，顺次类推。另外，context.WithCancel返回的CancelFunc函数，主逻辑其实也就是敞开管道，以此传递context对象敞开信号。
contetx.timerCtx与contetx.cancelCtx还是比拟相似的，只不过内置了定时器而已，定时器触发时主动敞开context对象，并没有其余区别，这里就不再赘述。
再次强调，基于context实现工作的超时管制，在Go语言中十分常见，能够说到处都能看到，所以肯定要纯熟应用并理解其原理。
全链路追踪
全链路追踪是什么意思呢？构想你保护着一个Go HTTP服务，申请处理过程必定会记录一些日志，不然遇到问题怎么排查定位？可是，一个HTTP申请过程必定会记录多条日志，如何将这些日志汇总起来呢？要晓得日志文件里这些日志记录可是扩散开的。能够基于traceId实现，即所有日志记录都蕴含traceId字段，并且同一个申请打印的日志traceId雷同，这就是全链路追踪了；更进一步，如果还依赖了其余第三方服务，在向他们发动HTTP申请时，也能够携带traceId，第三方服务打印日志也携带该traceId，这样咱们甚至能汇总一个用户申请波及的多个服务间的日志（这只能汇总，还不能直观剖析其调用关系，个别还会有其余字段形容调用关系）。
为什么全链路追踪要放到context这一篇文章介绍呢？因为日志要记录traceId，可是traceId从哪来？上下文呗，也就是context了。所以你的Go服务，所有函数的第一个参数最好都是context.Context。大略应该是这个样子：
func Handler(ctx context.Context, req Req) (resp Resp, err error) {
//出错，记录error日志
if err != nil {
logger.Errorf(ctx, "xxxx error:%v", err)
return
}
}
func Errorf(ctx context.Context, tag, template string, args ...interface{}) {
logger.With("traceId", ctx.Value("traceId")).Errorf(template, args...)
}那如果已有的Go我的项目，参数的确没有context.Context呢？革新所有函数吗？这老本挺高的，当然也能够尝试革新。退而求其次，其实还有一个下下策，日志中记录协程ID，这样同一个协程记录的日志能够依据该ID汇总（子协程执行的工作就无奈汇总了）。
不过，Go语言貌似没有提供形式获取协程ID，的确没有（不倡议）。这里提一个开源组件（ https://github.com/petermatti… ），封装了协程ID的获取形式，当然底层也是基于线程本地存储获取的，还记得不，Go语言以后调度的协程g会保留在线程本地存储，有了g对象，是不是就能获取到协程ID。这里略微理解一下就能够。
func getg() *g
func Get() int64 {
return getg().goid
}
TEXT ·Get(SB),NOSPLIT,$0-8
//协程本地存储TLS
MOVQ (TLS), R14
MOVQ g_goid(R14), R13
MOVQ R13, ret+0(FP)
RET
总结
基于context实现工作的超时管制，在Go语言中十分常见，能够说到处都能看到，所以本篇文章介绍了context是如何在上下文传递key-value键值对，以及管制超时的。最初还简略形容了全链路追踪的含意，在日常Go我的项目开发中，肯定要记得，所有函数的第一个参数最好都是context.Context。
发表回复