面試官:Context攜帶數據是線程安全的嗎?
本文轉載自微信公眾號「Golang夢工廠」,作者AsongGo。轉載本文請聯系Golang夢工廠公眾號。
哈嘍,大家好,我是asong。最近一個群里看到一個有趣的八股文,問題是:使用context攜帶的value是線程安全的嗎?這道題其實就是考察面試者對context實現原理的理解,如果不知道context的實現原理,很容易答錯這道題,所以本文我們就借著這道題,再重新理解一遍context攜帶value的實現原理。
context攜帶value是線程安全的嗎?
先說答案,context本身就是線程安全的,所以context攜帶value也是線程安全的,寫個簡單例子驗證一下:
func main() {
ctx := context.WithValue(context.Background(), "asong", "test01")
go func() {
for {
_ = context.WithValue(ctx, "asong", "test02")
}
}()
go func() {
for {
_ = context.WithValue(ctx, "asong", "test03")
}
}()
go func() {
for {
fmt.Println(ctx.Value("asong"))
}
}()
go func() {
for {
fmt.Println(ctx.Value("asong"))
}
}()
time.Sleep(10 * time.Second)
}
程序正常運行,沒有任何問題,接下來我們就來看一下為什么context是線程安全的!!!
為什么線程安全?
context包提供兩種創建根context的方式:
- context.Backgroud()
- context.TODO()
又提供了四個函數基于父Context衍生,其中使用WithValue函數來衍生context并攜帶數據,每次調用WithValue函數都會基于當前context衍生一個新的子context,WithValue內部主要就是調用valueCtx類:
func WithValue(parent Context, key, val interface{}) Context {
if parent == nil {
panic("cannot create context from nil parent")
}
if key == nil {
panic("nil key")
}
if !reflectlite.TypeOf(key).Comparable() {
panic("key is not comparable")
}
return &valueCtx{parent, key, val}
}
valueCtx結構如下:
type valueCtx struct {
Context
key, val interface{}
}
valueCtx繼承父Context,這種是采用匿名接口的繼承實現方式,key,val用來存儲攜帶的鍵值對。
通過上面的代碼分析,可以看到添加鍵值對不是在原context結構體上直接添加,而是以此context作為父節點,重新創建一個新的valueCtx子節點,將鍵值對添加在子節點上,由此形成一條context鏈。
獲取鍵值過程也是層層向上調用直到最終的根節點,中間要是找到了key就會返回,否會就會找到最終的emptyCtx返回nil。畫個圖表示一下:
image-20220207214507921
總結:context添加的鍵值對一個鏈式的,會不斷衍生新的context,所以context本身是不可變的,因此是線程安全的。
總結
本文主要是想帶大家回顧一下context的實現原理,面試中面試官都喜歡隱晦提出問題,所以這就需要我們有很扎實的基本功,一不小心就會掉入面試官的陷阱,要處處小心哦~
好啦,本文到這里就結束了,我是asong,我們下期見。
創建了讀者交流群,歡迎各位大佬們踴躍入群,一起學習交流。入群方式:關注公眾號獲取。更多學習資料請到公眾號領取。