在Go語言並發編程中，用一個goroutine來處理一個任務，而它又會創建多個goroutine來負責不同子任務的場景非常常見。如下圖
這些場景中，往往會需要在API邊界之間以及過程之間傳遞截止時間、取消信號或與其它請求相關的數據
誰是性能卡點呢？得通知它們任務取消了。
這時候就可以使用Context了。context包在Go1.7的時候被加入到官方庫中。
context包的內容可以概括為，一個接口，四個具體實現，還有六個函數。
Context接口提供了四個方法，下面是Context的接口
type Context interface {
Deadline() (deadline time.Time, ok bool)
Done() <-chan struct{}
Err() error
Value(key interface{}) interface{}
}
emptyCtx類型
emptyCtx本質上是一個整型, *emptyCtx對Context接口的實現，只是簡單的返回nil，false，實際上什么也沒做。如下代碼所示：
type emptyCtx int
func (*emptyCtx) Deadline() (deadline time.Time, ok bool) {
return
}
func (*emptyCtx) Done() <-chan struct{} {
return nil
}
func (*emptyCtx) Err() error {
return nil
}
func (*emptyCtx) Value(key interface{}) interface{} {
return nil
}
Background和TODO這兩個函數內部都會創建emptyCtx
var (
background = new(emptyCtx)
todo = new(emptyCtx)
)
func Background() Context {
return background
}
func TODO() Context {
return todo
}
其中Background主要用於在初始化時獲取一個Context(從代碼中可知本質是一個*emptyCtx，而emptCtx本質上是一個Int)，這就是Background()函數返回的變量結構。
而TODO()函數，官方文檔建議在本來應該使用外層傳遞的ctx而外層卻沒有傳遞的地方使用，就像函數名稱表達的含義一樣，留下一個TODO。
cancelCtx類型
再來看cancelCtx類型，cancleCtx定義如下
// cancelCtx可以被取消。 取消后，它也會取消所有實現取消方法的子級。
type cancelCtx struct {
Context
mu sync.Mutex // protects following fields
done chan struct{} // created lazily, closed by first cancel call
children map[canceler]struct{} // set to nil by the first cancel call
err error // set to non-nil by the first cancel call
}
func (c *cancelCtx) Value(key interface{}) interface{} {
if key == &cancelCtxKey {
return c
}
return c.Context.Value(key)
}
func (c *cancelCtx) Done() <-chan struct{} {
c.mu.Lock()
if c.done == nil {
c.done = make(chan struct{})
}
d := c.done
c.mu.Unlock()
return d
}
func (c *cancelCtx) Err() error {
c.mu.Lock()
err := c.err
c.mu.Unlock()
return err
}
這是一種可取消的Context，done用於獲取該Context的取消通知，children用於存儲以當前節點為根節點的所有可取消的Context，以便在根節點取消時，可以把它們一並取消，err用於存儲取消時指定的錯誤信息，而這個mu就是用來保護這幾個字段的鎖，以保障cancelCtx是線程安全的。
而WithCancel函數，可以把一個Context包裝為cancelCtx，並提供一個取消函數，調用它可以Cancel對應的Context
func WithCancel(parent Context) (ctx Context, cancel CancelFunc) {
if parent == nil {
panic("cannot create context from nil parent")
}
c := newCancelCtx(parent)
propagateCancel(parent, &c)
return &c, func() { c.cancel(true, Canceled) }
}
示例代碼：
ctx := context.Background()
ctx1, cancel := context.WithCancel(ctx)
timerCtx類型
再來看timerCtx,timerCtx定義如下
type timerCtx struct {
cancelCtx
timer *time.Timer // Under cancelCtx.mu.
deadline time.Time
}
它在cancelCtx的基礎上，又封裝了一個定時器和一個截止時間，這樣既可以根據需要主動取消，也可以在到達deadline時，通過timer來觸發取消動作。
要注意，這個timer也會由cancelCtx.mu來保護，確保取消操作也是線程安全的。
通過WithDeadline和WithTimeout函數，都可以創建timerCtx，區別是WithDeadline函數需要指定一個時間點，而WithTimeout函數接收一個時間段。
接下來，我們基於ctx1構造一個timerCtx
ctx := context.Background()
ctx1, cancel := context.WithCancel(ctx)
deadline := time.Now().Add(time.Second)
ctx2, cancel := context.WithDeadline(ctx1, deadline)
這個定時器會在deadline到達時，調用cancelCtx的取消函數，現在可以看到ctx2是基於ctx1創建的，而ctx1又是基於ctx創建的，基於每個Context可以創建多個Context，這樣就形成了一個Context樹，每個節點都可以有零個或多個子節點，可取消的Context都會被注冊到離它最近的、可取消的祖先節點中。對ctx2來說離它最新的、可取消的祖先節點是ctx1
所以在ctx1這里的children map中，會增加ctx2這組鍵值對
如果ctx2先取消，就只會影響到以它為根節點的Context，而如果ctx1先取消，就可以根據children map中的記錄，把ctx1子節點中所有可取消的Context全部Cancel掉。
最后來看valueCtx類型
valueCtx類型
首先來看valueCtx的定義
type valueCtx struct {
Context
key, val interface{}
}
它用來支持鍵值對打包，WithValue函數可以給Context附加一個鍵值對信息，這樣就可以通過Context傳遞數據了
var keyA string = "keyA"
ctx := context.Background()
ctxA := context.WithValue(ctx, keyA, "valA")
現在我們給ctx附加一個鍵值對keyA=>valA，變量ctxA也是Context接口類型，動態類型為*valueCtx，data指向一個valueCtx結構體，第一個字段是它的父級Context，key和val字段都是空接口類型，keyA的動態類型為string，動態值是string類型的變量keyA，val的動態類型同樣是string，動態值為valA，
下面我們再基於ctxA，附加一個key相等但val不相等的鍵值對keyA=>eggo，ctxC的動態值指向這樣一個valueCtx，父級Context自然是ctxA，key與ctxA中的相同，但是val的值與ctxA中的不相等
通過ctxC獲取kyA和keyC對應的值時會發現keyC覆蓋了keyA對應的val，要找到原因，就要先看看Value方法是怎么工作的
func (c *valueCtx) Value(key interface{}) interface{} {
if c.key == key {
return c.val
}
return c.Context.Value(key)
}
首先它會比較當前Context中的key是否等於要查找的key，因為keyA等於keyC，所以對keyA的查找會直接鎖定到ctxC這里的val，因而出現了子節點覆蓋父節點數據的情況，為了規避這種情況，最好不要直接使用string、int這些基礎類型作為Key，而是用自定義類型包裝一下，就像下面這樣，把keyA定義為keytypea類型，keyC定義為keytypec類型，這樣再次通過ctxC獲取keyA時，因為key的類型不相同，第一步key相等性比較不通過，就會委托父節點繼續查找，進而找到正確的val
所以說valueCtx之間通過Context字段形成了一個鏈表結構，使用Context傳遞數據時還要注意，Context本身本着不可改變(immutable)的模式設計的，所以不要試圖修改ctx里保存的值，在http、sql相關的庫中，都提供了對Context的支持，方便我們在處理請求時，實現超時自動取消，或傳遞請求相關的控制數據等等
了解了context包中，一個接口，四種具體實現，以及六個函數的基本情況，有助於我們理解Context的工作原理
整理自：