golang的協程上下文的具體使用

2022-04-11 19:01:20

go協程上下文context

golang的context 主要用來在 goroutine 之間傳遞上下文資訊，包括：取消訊號、超時時間、截止時間、k-v 等

context是golang1.17版本之後才出的特性

上下文解決的問題

協程間的通訊

例如web應用中，每一個請求都由一個協程去處理。當然處理處理請求的這個協程，一般我們還會起一些其他的協程，用來處理其他的業務，比如運算元據庫，生份驗證、檔案讀寫等。這些協程是獨立的，我們在當前的協程中無法感知到其他的協程執行的情況怎麼樣了。實用通道channel可以實現通訊功能

context中context.WithValue()本質上也是通過channel來實現通訊

子協程的超時處理

同樣例如web應用當中，我們主程序是一直常駐記憶體的。每一個請求都由一個協程去處理，在處理業務的過程中可能會起另外的協程去處理其他的業務，當子協程出現了異常或者阻塞了，無法向上一級的協程反饋資訊，主協程接受不到反饋也會阻塞。上下文可以很好的解決這個問題，context可以實現子協程或子孫協程的超時退出或定時退出

上下文的幾個方法和用法

context.WithCancel(context.Context)

WithCancel()方法傳入一個上下文空範例，直接用context.Background()即可，返回一個上下文和一個取消函數。呼叫cancal()會向其他協程傳遞訊號，收到訊號後子協程就可以做關閉或其他處理

package main

import (
    "context"
    "fmt"
    "time"
)
func worker(ctx context.Context) {
    for {
        select {
        case <-ctx.Done():
            return
        default:
        }
        fmt.Println("worker...")
        time.Sleep(time.Second * 1)
    }
}
func main() {
    ctx, cancal := context.WithCancel(context.Background())
    go worker(ctx)
    time.Sleep(time.Second * 5)
    cancal()
    fmt.Println("over ...")

}

context.WithTimeout(context.Context，timeout)

定義一個會超時的上下文，範例化後倒計時就開始，到時間會自動呼叫cancel()函數通知子協程,也可以手動呼叫cancel()通知。如果子協程中還有子協程，繼續使用這個上下文，當主協程發出取消訊號時每一個使用了這個上下文的都會收到通知

package main

import (
    "context"
    "fmt"
    "time"
)

func worker(ctx context.Context) {
    for {
        select {
        case <-ctx.Done():
            return
        default:
        }
        fmt.Println("worker...")
        time.Sleep(time.Second * 1)
    }
}

func main() {
    ctx, cancal := context.WithTimeout(context.Background(), time.Second*2)
    go worker(ctx)
    time.Sleep(time.Second * 5)
    cancal()
    fmt.Println("over ...")

}

context.WithDeadline(context.Context，(絕對時間)timeout)

定義一個會超時的上下文，與Timeout不同在於，傳入的時間是一個絕對時間。到了指定的時間會自動呼叫cancel()函數通知子協程,也可以手動呼叫cancel()通知。如果子協程中還有子協程，繼續使用這個上下文，當主協程發出取消訊號時每一個使用了這個上下文的都會收到通知

package main

import (
    "context"
    "fmt"
    "time"
)

func worker(ctx context.Context) {
    for {
        select {
        case <-ctx.Done():
            return
        default:
        }
        fmt.Println("worker...")
        time.Sleep(time.Second * 1)
    }
}

func main() {
    ctx, cancal := context.WithDeadline(context.Background(), time.Now().Add(3 * time.Second))
    go worker(ctx)
    time.Sleep(time.Second * 5)
    cancal()
    fmt.Println("over ...")

}

協程間的上下文通訊：context.WithValue()

先看一下這段程式碼

package main

import (
    "context"
    "fmt"
    "time"
)

type CTXKEY string

func worker(ctx context.Context) {
    // 在子協程中獲取上下文資訊
    num, ok := ctx.Value(CTXKEY("num")).(string)
    if !ok {
        fmt.Println("invalid trace code")
    }
    fmt.Println(num)
    for {
        select {
        case <-ctx.Done():
            return
        default:
        }
        fmt.Println("worker...")
        time.Sleep(time.Second * 1)
    }
}

func main() {
    ctx, cancal := context.WithDeadline(context.Background(), time.Now().Add(3*time.Second))
    // 利用上下文傳一個 num = 1234567
    // 範例化一個上下文
    ctx = context.WithValue(ctx, CTXKEY("num"), "1234567")
    go worker(ctx)
    time.Sleep(time.Second * 5)
    cancal()
    fmt.Println("over ...")
}

通過上下文實現協程間的通訊,如果專案大，為了避免變數的汙染，原則上：上下文通訊所用的key需要自定義一個型別

type traceCode string;context.WithValue(context.Context,key,value)

到此這篇關於golang的協程上下文的文章就介紹到這了,更多相關golang的協程上下文內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！

golang的協程上下文的具體使用

go協程上下文context

上下文解決的問題

上下文的幾個方法和用法

熱門文章