Golang標(biāo)準(zhǔn)庫——context

context

context包定義了上下文類型,該類型在API邊界之間以及進(jìn)程之間傳遞截止日期嘲碧,取消信號和其他請求范圍的值嘴纺。

對服務(wù)器的傳入請求應(yīng)創(chuàng)建一個(gè)Context邓深,而對服務(wù)器的傳出調(diào)用應(yīng)接受一個(gè)Context。它們之間的函數(shù)調(diào)用鏈必須傳播Context衙吩,可以選擇將其替換為使用WithCancel互妓,WithDeadline,WithTimeout或WithValue創(chuàng)建的派生Context坤塞。取消context后冯勉,從該Context派生的所有Context也會被取消。

WithCancel摹芙,WithDeadline和WithTimeout函數(shù)采用Context(父級)并返回派生的Context(子級)和CancelFunc灼狰。調(diào)用CancelFunc會取消該子代及其子代,刪除父代對該子代的引用浮禾,并停止所有關(guān)聯(lián)的計(jì)時(shí)器交胚。未能調(diào)用CancelFunc會使子代及其子代泄漏份汗,直到父代被取消或計(jì)時(shí)器觸發(fā)。審核工具檢查所有控制流路徑上是否都使用了CancelFuncs蝴簇。

使用Context的程序應(yīng)遵循以下規(guī)則杯活,以使各個(gè)包之間的接口保持一致,并啟用靜態(tài)分析工具來檢查上下文傳播:

  • 不要將Context存儲在結(jié)構(gòu)類型中军熏;
  • 將Context明確傳遞給需要它的每個(gè)函數(shù)轩猩;
  • Context應(yīng)該是第一個(gè)參數(shù),通常命名為ctx:
func DoSomething(ctx context.Context, arg Arg) error {
  // ... use ctx ...
}

即使函數(shù)允許荡澎,也不要傳遞nil Context均践。如果不確定使用哪個(gè)上下文,請傳遞context.TODO摩幔。

僅將Context值用于傳遞流程和API的請求范圍的數(shù)據(jù)彤委,而不用于將可選參數(shù)傳遞給函數(shù)。

可以將相同的Context傳遞給在不同goroutine中運(yùn)行的函數(shù)或衡。Context對于由多個(gè)goroutine同時(shí)使用是安全的焦影。

Variables

var Canceled = errors.New("context canceled")
var DeadlineExceeded error = deadlineExceededError{}

Canceled是Context.Err類型,取消Context時(shí)返回的錯(cuò)誤封断。

DeadlineExceeded是上下文的最后期限過去時(shí)Context.Err返回的錯(cuò)誤斯辰。

type CancelFunc

type CancelFunc func()

CancelFunc告訴操作停止工作。 CancelFunc不等待工作停止坡疼。 在第一個(gè)調(diào)用之后彬呻,隨后對CancelFunc的調(diào)用什么都不做。

type Context

type Context interface {

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

其中

  • Deadline方法需要返回當(dāng)前Context被取消的時(shí)間柄瑰,也就是完成工作的截止時(shí)間(deadline)闸氮;
  • Done方法需要返回一個(gè)Channel,這個(gè)Channel會在當(dāng)前工作完成或者上下文被取消之后關(guān)閉教沾,多次調(diào)用Done方法會返回同一個(gè)Channel蒲跨;
  • Err方法會返回當(dāng)前Context結(jié)束的原因,它只會在Done返回的Channel被關(guān)閉時(shí)才會返回非空的值授翻;
    • 如果當(dāng)前Context被取消就會返回Canceled錯(cuò)誤或悲;
    • 如果當(dāng)前Context超時(shí)就會返回DeadlineExceeded錯(cuò)誤;
  • Value方法會從Context中返回鍵對應(yīng)的值堪唐,對于同一個(gè)上下文來說巡语,多次調(diào)用Value并傳入相同的Key會返回相同的結(jié)果,該方法僅用于傳遞跨API和進(jìn)程間跟請求域的數(shù)據(jù)羔杨;

func Background

func Background() Context

Background返回一個(gè)非空的Context捌臊。 它永遠(yuǎn)不會被取消,沒有值兜材,也沒有期限理澎。 它通常由主要功能逞力,初始化和測試使用,并用作傳入請求的頂級Context糠爬。

func TODO

func TODO() Context

TODO返回一個(gè)非空的Context寇荧。 當(dāng)不清楚要使用哪個(gè)上下文或尚不可用時(shí)(因?yàn)樯形磾U(kuò)展周圍的功能以接受Context參數(shù)),代碼應(yīng)使用context.TODO执隧。 靜態(tài)分析工具可識別TODO揩抡,該工具可確定上下文是否在程序中正確傳播。

func WithCancel

func WithCancel(parent Context) (ctx Context, cancel CancelFunc)

WithCancel返回具有新的“完成”通道的父級副本镀琉。 當(dāng)調(diào)用返回的cancel函數(shù)或關(guān)閉父上下文的Done通道時(shí)(以先發(fā)生的為準(zhǔn))峦嗤,關(guān)閉返回的上下文的Done通道。

取消此上下文將釋放與其關(guān)聯(lián)的資源屋摔,因此在此上下文中運(yùn)行的操作完成后烁设,代碼應(yīng)立即調(diào)用cancel。

func main() {
   gen := func(ctx context.Context) <-chan int {
      dst := make(chan int)
      n := 1
      go func() {
         for {
            select {
            case <-ctx.Done():
               return // returning not to leak the goroutine
            case dst <- n:
               n++
            }
         }
      }()
      return dst
   }

   ctx, cancel := context.WithCancel(context.Background())
   defer cancel() // cancel when we are finished consuming integers

   for n := range gen(ctx) {
      fmt.Println(n)
      if n == 5 {
         break
      }
   }
}

上面的示例代碼中钓试,gen函數(shù)在單獨(dú)的goroutine中生成整數(shù)并將它們發(fā)送到返回的通道装黑。 gen的調(diào)用者在使用生成的整數(shù)之后需要取消上下文,以免gen啟動的內(nèi)部goroutine發(fā)生泄漏弓熏。

func WithDeadline

func WithDeadline(parent Context, deadline time.Time) (Context, CancelFunc)

WithDeadline返回父上下文的副本恋谭,并將截止日期調(diào)整為不遲于d。 如果父母的截止日期早于d挽鞠,則WithDeadline(parent疚颊,d)在語義上等同于父母。 當(dāng)截止日期到期滞谢,調(diào)用返回的cancel函數(shù)或關(guān)閉父上下文的Done通道(以先到者為準(zhǔn))時(shí)串稀,將關(guān)閉返回的上下文的Done通道除抛。

取消此上下文將釋放與其關(guān)聯(lián)的資源狮杨,因此在此上下文中運(yùn)行的操作完成后,代碼應(yīng)立即調(diào)用cancel到忽。

func main() {
   d := time.Now().Add(50 * time.Millisecond)
   ctx, cancel := context.WithDeadline(context.Background(), d)

   // Even though ctx will be expired, it is good practice to call its
   // cancelation function in any case. Failure to do so may keep the
   // context and its parent alive longer than necessary.
   defer cancel()

   select {
   case <-time.After(1 * time.Second):
      fmt.Println("overslept")
   case <-ctx.Done():
      fmt.Println(ctx.Err())
   }
}

上面的代碼中橄教,定義了一個(gè)50毫秒之后過期的deadline,然后我們調(diào)用context.WithDeadline(context.Background(), d)得到一個(gè)上下文(ctx)和一個(gè)取消函數(shù)(cancel)喘漏,然后使用一個(gè)select讓主程序陷入等待:等待1秒后打印overslept退出或者等待ctx過期后退出护蝶。 因?yàn)閏tx50毫秒后就過期,所以ctx.Done()會先接收到值翩迈,上面的代碼會打印ctx.Err()取消原因持灰。

func WithTimeout

func WithTimeout(parent Context, timeout time.Duration) (Context, CancelFunc)

WithTimeout返回WithDeadline(parent, time.Now().Add(timeout))方法。

取消此Context 將釋放與之關(guān)聯(lián)的資源负饲,因此在此Context 中運(yùn)行的操作完成后堤魁,代碼應(yīng)立即調(diào)用cancel:

func main() {
   ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)
   defer cancel()

   select {
   case <-time.After(1 * time.Second):
      fmt.Println("overslept")
   case <-ctx.Done():
      fmt.Println(ctx.Err()) // prints "context deadline exceeded"
   }
}

func WithValue

func WithValue(parent Context, key, val interface{}) Context

WithValue返回父項(xiàng)的副本喂链,其中與key關(guān)聯(lián)的值為val。

僅將Context值用于傳遞流程和API的請求范圍的數(shù)據(jù)妥泉,而不用于將可選參數(shù)傳遞給函數(shù)椭微。

提供的密鑰必須具有可比性,并且不能為字符串類型或任何其他內(nèi)置類型盲链,以避免使用上下文在程序包之間發(fā)生沖突蝇率。 WithValue的用戶應(yīng)定義自己的密鑰類型。 為了避免在分配給接口{}時(shí)進(jìn)行分配刽沾,上下文鍵通常具有具體的類型struct {}本慕。 另外,導(dǎo)出的上下文鍵變量的靜態(tài)類型應(yīng)該是指針或接口侧漓。

func main() {
   type favContextKey string

   f := func(ctx context.Context, k favContextKey) {
      if v := ctx.Value(k); v != nil {
         fmt.Println("found value:", v)
         return
      }
      fmt.Println("key not found:", k)
   }

   k := favContextKey("language")
   ctx := context.WithValue(context.Background(), k, "Go")

   f(ctx, k)
   f(ctx, favContextKey("color"))
}
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末间狂,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子火架,更是在濱河造成了極大的恐慌鉴象,老刑警劉巖,帶你破解...
    沈念sama閱讀 219,427評論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件何鸡,死亡現(xiàn)場離奇詭異纺弊,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)骡男,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,551評論 3 395
  • 文/潘曉璐 我一進(jìn)店門淆游,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人隔盛,你說我怎么就攤上這事犹菱。” “怎么了吮炕?”我有些...
    開封第一講書人閱讀 165,747評論 0 356
  • 文/不壞的土叔 我叫張陵腊脱,是天一觀的道長。 經(jīng)常有香客問我龙亲,道長陕凹,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,939評論 1 295
  • 正文 為了忘掉前任鳄炉,我火速辦了婚禮杜耙,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘拂盯。我一直安慰自己佑女,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,955評論 6 392
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著团驱,像睡著了一般簸呈。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上店茶,一...
    開封第一講書人閱讀 51,737評論 1 305
  • 那天蜕便,我揣著相機(jī)與錄音,去河邊找鬼贩幻。 笑死轿腺,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的丛楚。 我是一名探鬼主播族壳,決...
    沈念sama閱讀 40,448評論 3 420
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼趣些!你這毒婦竟也來了仿荆?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,352評論 0 276
  • 序言:老撾萬榮一對情侶失蹤坏平,失蹤者是張志新(化名)和其女友劉穎拢操,沒想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體舶替,經(jīng)...
    沈念sama閱讀 45,834評論 1 317
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡令境,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,992評論 3 338
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了顾瞪。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片舔庶。...
    茶點(diǎn)故事閱讀 40,133評論 1 351
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖陈醒,靈堂內(nèi)的尸體忽然破棺而出惕橙,到底是詐尸還是另有隱情,我是刑警寧澤钉跷,帶...
    沈念sama閱讀 35,815評論 5 346
  • 正文 年R本政府宣布弥鹦,位于F島的核電站,受9級特大地震影響尘应,放射性物質(zhì)發(fā)生泄漏惶凝。R本人自食惡果不足惜吼虎,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,477評論 3 331
  • 文/蒙蒙 一犬钢、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧思灰,春花似錦玷犹、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,022評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽坯屿。三九已至,卻和暖如春巍扛,著一層夾襖步出監(jiān)牢的瞬間领跛,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,147評論 1 272
  • 我被黑心中介騙來泰國打工撤奸, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留吠昭,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 48,398評論 3 373
  • 正文 我出身青樓胧瓜,卻偏偏與公主長得像矢棚,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個(gè)殘疾皇子府喳,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,077評論 2 355

推薦閱讀更多精彩內(nèi)容