摸魚心法第一章——和配置文件說拜拜
為了能摸魚我們團隊做了容器化，但是帶來的問題是服務配置文件很麻煩，然后大家在群里進行了“親切友好”的溝通圖片圖片圖片圖片對比就對比，簡單對比下獨立配置中心和k8s作為配
你好，我是四哥。
前一篇文章從源碼的角度詳細介紹了 Context 的實現原理，但是還沒有提到 Context 的使用場景，今天我們一起來看下：
1.請求鏈路傳值。
傳值使用方式如下：
func func1(ctx context.Context) {
ctx = context.WithValue(ctx, "k1", "v1")
func2(ctx)
}
func func2(ctx context.Context) {
fmt.Println("func2:",ctx.Value("k1").(string))
ctx = context.WithValue(ctx, "k2", "v2")
func3(ctx)
}
func func3(ctx context.Context) {
fmt.Println("func3:",ctx.Value("k1").(string))
fmt.Println("func3:",ctx.Value("k2").(string))
}
func main() {
ctx := context.Background()
func1(ctx)
}
我們在 func1() 通過函數 WithValue() 設置了一個鍵值對 k1-v1，在 func2() 可以獲取到 func1() 設置的鍵值對，如果調用 func3() 時把這個 ctx 繼續傳入的話，在 func3() 中依然還是可以獲取到 k1-v1。
但是在 func1() 中獲取不到 func2() 設置的鍵值對 k2-v2，因為 context 只能自上而下攜帶值，這點需要注意。
2.取消耗時操作，及時釋放資源。
使用 channel + select 的機制：
func func1() error {
respC := make(chan int) // 起消息通知作用
// 處理邏輯
go func() {
time.Sleep(time.Second * 3) // 模擬處理業務邏輯
respC close(respC)
}()
// 判斷是否超時
select {
case r := <-respC:
fmt.Printf("Resp: %d ", r)
return nil
case <-time.After(time.Second * 2): // 超過設置的時間就報錯
fmt.Println("catch timeout")
return errors.New("timeout")
}
}
func main() {
err := func1()
fmt.Printf("func1 error: %v ", err)
}
上面的方式平時也會用到，通過 context 怎么實現呢?
下面來看下如何使用 context 進行主動取消、超時取消。
主動取消：
func func1(ctx context.Context, wg *sync.WaitGroup) error {
defer wg.Done()
respC := make(chan int)
go func() {
time.Sleep(time.Second * 5) // 模擬業務邏輯處理
respC }()
// 取消機制
select {
case <-ctx.Done():
fmt.Println("cancel")
return errors.New("cancel")
case r := <-respC:
fmt.Println(r)
return nil
}
}
func main() {
wg := &sync.WaitGroup{}
ctx, cancel := context.WithCancel(context.Background())
wg.Add(1)
go func1(ctx, wg)
time.Sleep(time.Second * 2)
cancel() // 主動取消
wg.Wait() // 等待 goroutine 退出
}
超時取消：
func func1(ctx context.Context) {
resp := make(chan int)
go func() {
time.Sleep(time.Second * 5) // 模擬處理邏輯
resp }()
// 超時機制
select {
case <-ctx.Done():
fmt.Println("ctx timeout")
fmt.Println(ctx.Err())
case <-resp:
fmt.Println("done")
}
return
}
func main() {
ctx, cancel := context.WithTimeout(context.Background(), time.Second*2)
defer cancel()
func1(ctx)
}3.防止 goroutine 泄露。
引自【深度解密 Go 語言之 context[1]】
func gen() ch := make(chan int)
go func() {
var n int
for {
ch n++
time.Sleep(time.Second)
}
}()
return ch
}
這是一個可以生成無限整數的協程，但如果我只需要它產生的前 5 個數，那么就會發生 goroutine 泄漏：
func main() {
for n := range gen() {
fmt.Println(n)
if n == 5 {
break
}
}
// ……
}
當 n == 5 的時候，直接 break 掉。那么 gen 函數的協程就會執行無限循環，永遠不會停下來。發生了 goroutine 泄漏。
用 context 改進這個例子：
func gen(ctx context.Context) ch := make(chan int)
go func() {
var n int
for {
select {
case <-ctx.Done():
return
case ch n++
time.Sleep(time.Second)
}
}
}()
return ch
}
func main() {
ctx, cancel := context.WithCancel(context.Background())
defer cancel() // 避免其他地方忘記 cancel，且重復調用不影響
for n := range gen(ctx) {
fmt.Println(n)
if n == 5 {
cancel()
break
}
}
// ……
}
增加一個 context，在 break 前調用 cancel 函數，取消 goroutine。gen 函數在接收到取消信號后，直接退出，系統回收資源。
總結
這篇文章列出的幾個例子是 context 最基本的使用場景，其他框架、第三包基本上都是從這幾種用法擴展的，所以非常有必要掌握基礎用法。
另外希望這篇文章能給你帶來幫助，如果文中有理解錯誤之處或者你還想到其他用法，可以在留言區留言，一定回復!抱團學習不孤單!
參考資料
[1]深度解密Go語言之context: https://qcrao.com/2019/06/12/dive-into-go-context/
本文鏈接：http://www.www897cc.com/showinfo-119-2167-0.html細數 Context 使用場景
聲明：本網頁內容旨在傳播知識，若有侵權等問題請及時與本網聯系，我們將在第一時間刪除處理。郵件：2376512515@qq.com
上一篇： 云計算開發：Python3-isdecimal()方法詳解
下一篇： 數據分析八大模型：漏斗模型
Copyright ? 2016-2023 天津谷騏科技有限公司 版權所有 sitemap.xml
違法及侵權請聯系：2376512515@qq.com 津ICP備18001702號
津公網安備 12010102000574號