引言

現(xiàn)在團(tuán)隊(duì)里幾乎所有的代碼都需要經(jīng)過(guò) Code Review（代碼審查）之后，才允許合入主分支。作為 CR 負(fù)責(zé)人之一，在 CR 中看到了不少不適合的問(wèn)題，也看到了不少值得學(xué)習(xí)的點(diǎn)。筆者決定從今天開(kāi)始，一點(diǎn)一滴地記錄這些做法、經(jīng)驗(yàn)、教訓(xùn)，以饗讀者。

這系列文章，我都會(huì)先拋出一句話規(guī)范說(shuō)明，然后解釋問(wèn)題的背景，最后再給出正確的規(guī)范。

一句話規(guī)范

• 當(dāng)使用 context.Context 類(lèi)型保存 KV 對(duì)時(shí), key 不能使用原生類(lèi)型，而應(yīng)該使用派生類(lèi)型

問(wèn)題背景

我們知道，可以利用 context.Context 類(lèi)型來(lái)存一些自定義的鍵值對(duì)——當(dāng)然了，需要保存新的 context 對(duì)象：

ctx = context.WithValue(ctx, someKey, someValue)

很快就可以注意到 key 的參數(shù)類(lèi)型是 any，也就是 Go 1.17 之前的 interface{}。“可能是為了能夠使用 int 吧？” ——初學(xué)者很可能會(huì)這么想。

在實(shí)際的 CR 中，可以看到很多人都使用 string 類(lèi)型作為 key，比如這是一個(gè)非常典型的例子：

ctx = context.WithValue(ctx, "openid", userOpenID)

存在問(wèn)題

如果你使用了 VSCode，并且一鍵安裝了 Go 開(kāi)發(fā)工具包，那么 VSCode 大概率也安裝了 golangci-lint 工具。此時(shí)，上面的這段代碼會(huì)喜提一個(gè) warning：

SA1029: should not use built-in type string as key for value; define your own type to avoid collisions (staticcheck)

告警信息雖然是英文，但很容易理解：

• 不應(yīng)該使用內(nèi)建類(lèi)型作為 KV 中 key 的類(lèi)型，而應(yīng)該使用自定義的類(lèi)型來(lái)避免沖突。

為什么要這么做呢？很簡(jiǎn)單，現(xiàn)代軟件都是團(tuán)隊(duì)開(kāi)發(fā)的，多模塊相互耦合，互相協(xié)作。在一個(gè) ctx 對(duì)象的整個(gè)生命周期中，它需要經(jīng)過(guò)多個(gè)邏輯 / 模塊的洗禮，每一個(gè)模塊都可能使用 ctx 來(lái)存儲(chǔ)相應(yīng)信息。

假設(shè) user 模塊，它使用 ctx 類(lèi)型緩存了用戶的 openid 字段。這個(gè)邏輯沒(méi)什么問(wèn)題。然后這個(gè) ctx（和代碼邏輯）繼續(xù)往后走，大家約定，就使用這個(gè) "openid" 來(lái)存儲(chǔ)。

有一天，來(lái)了一個(gè)緊急需求，比如說(shuō)要做一個(gè)群聊功能，盡可能復(fù)用老代碼減少開(kāi)發(fā)?；蛟S group 模塊就利用了 user 模塊的代碼。好巧不巧，從其他團(tuán)隊(duì)過(guò)來(lái)支援的開(kāi)發(fā)同學(xué)，也使用了 "openid" 這個(gè) key，來(lái)存儲(chǔ)群主的 openid。結(jié)果繞了一圈，這位同學(xué)發(fā)現(xiàn)：咦怎么這群主的 openid 老變成別人的 openid？擱這群主輪流做是吧？

解決方法

可能有人覺(jué)得：那我把 key 的定義統(tǒng)一收集起來(lái)規(guī)定不就行了？解決一個(gè)問(wèn)題總有上中下策，這是個(gè)辦法，但是一個(gè)下下下策。軟件工程主打一個(gè)分而治之，在沒(méi)有必要的情況下，盡可能避免集中式的管理。

最上策的解決方法簡(jiǎn)單而言，就是使用自定義的類(lèi)型作為 key 的類(lèi)型。我們看看下面的代碼：

type myString string
func main() {
    ctx := context.Background()
    ctx = context.WithValue(ctx, "openid", "不是群主")
    ctx = context.WithValue(ctx, myString("openid"), "群主")
    fmt.Println(ctx.Value("openid"))
    fmt.Println(ctx.Value(myString("openid")))
}

兩行輸出：

不是群主
群主

這就非常清晰了，盡管底層類(lèi)型相同（都是 string 類(lèi)型），但是經(jīng)過(guò) type 定義之后，Go 是作為完全不同的 key 來(lái)處理的。針對(duì)具體類(lèi)型自定義 key 類(lèi)型之后，很好地解決了同名 key 沖突的問(wèn)題。

不過(guò)呢，如果你并不需要使用同一個(gè) key 類(lèi)型，存儲(chǔ)多個(gè)不同 value，那么上面的模式還只能說(shuō)是中策，真正的上策是這么做的：

type myString struct{}

func main() {
    ctx := context.Background()
    ctx = context.WithValue(ctx, "openid", "不是群主")
    ctx = context.WithValue(ctx, myString{}, "群主")

    fmt.Println(ctx.Value("openid"))
    fmt.Println(ctx.Value(myString{}))
}

我還記得我當(dāng)年第一次看到這個(gè)模式的時(shí)候，我就是上圖的這個(gè)表情。是的，struct{} 類(lèi)型也可以作為 KV 的 key 類(lèi)型——當(dāng)然了，也應(yīng)該定義為自定義類(lèi)型。

使用 struct{} 的好處可是大大的多：首先，這個(gè)類(lèi)型在 Go 中原則上是不占內(nèi)存空間和 gc 開(kāi)銷(xiāo)的，可以提升性能；其次，這少了開(kāi)發(fā)者額外 “寫(xiě)一個(gè) key” 的時(shí)間（類(lèi)型往往可以通過(guò) IDE 快速補(bǔ)全），大大提高了敲代碼的速度呀。

典型例子

使用 context WithValue 方法，有一個(gè)很典型的例子，就是在 ctx 中存入一個(gè) trace ID，用于跟蹤整個(gè)調(diào)用鏈。那么，我們可以包裝一個(gè) traceid 包，比較規(guī)范的寫(xiě)法是這樣的：

// Package traceid 用于在 context 中維護(hù) trace ID
package traceid
import "context"
// WithTraceID 往 context 中存入 trace ID
func WithTraceID(ctx context.Context, traceID string) context.Context {
    return context.WithValue(ctx, traceIDKey{}, traceID)
}
// TraceID 從 context 中提取 trace ID
func TraceID(ctx context.Context) string {
    v := context.Value(ctx, traceIDKey{})
    id, _ := v.(string)
    return id
}
type traceIDKey struct{}

以上就是Go語(yǔ)言規(guī)范context 類(lèi)型的key用法示例解析的詳細(xì)內(nèi)容，更多關(guān)于Go context類(lèi)型key的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

最新評(píng)論