石頭智能洗地機A10 Plus體驗：雙向自清潔治好了我的懶癌
一、前言和介紹專為家庭請假懶人而生的石頭科技在近日又帶來了自己的全新旗艦新品，石頭智能洗地機A10 Plus。從這個產品名上就不難看出，這次石頭推出的并不是常見的掃地機器
你好，我是四哥。
前一篇文章從源碼的角度詳細介紹了 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://m.www897cc.com/showinfo-119-2167-0.html細數 Context 使用場景
聲明：本網頁內容旨在傳播知識，若有侵權等問題請及時與本網聯系，我們將在第一時間刪除處理。郵件：2376512515@qq.com
上一篇： 云計算開發：Python3-isdecimal()方法詳解
下一篇： 數據分析八大模型：漏斗模型
Copyright ? 2016-2023 天津谷騏科技有限公司 版權所有 sitemap.xml
違法及侵權請聯系：2376512515@qq.com 津ICP備18001702號
津公網安備 12010102000574號