下面由golang教學欄位來介紹golang中Context的使用場景,希望對需要的朋友有幫助!
golang中Context的使用場景
context在Go1.7之後就進入標準函式庫中了。它主要的用處如果用一句話來說,是在於控制goroutine的生命週期。當一個計算任務被goroutine承接了之後,由於某種原因(超時,或者強制退出)我們希望中止這個goroutine的計算任務,那麼就用得到這個Context了。
本文主要來盤一盤golang中context的一些使用場景:
場景一:RPC呼叫
在主goroutine上有4個RPC,RPC2/3/4是並行請求的,我們這裡希望在RPC2請求失敗之後,直接回傳錯誤,並且讓RPC3/4停止繼續計算。這時候,就使用的到Context。
這個的具體實作如下面的程式碼。
package main import ( "context" "sync" "github.com/pkg/errors" ) func Rpc(ctx context.Context, url string) error { result := make(chan int) err := make(chan error) go func() { // 进行RPC调用,并且返回是否成功,成功通过result传递成功信息,错误通过error传递错误信息 isSuccess := true if isSuccess { result <- 1 } else { err <- errors.New("some error happen") } }() select { case <- ctx.Done(): // 其他RPC调用调用失败 return ctx.Err() case e := <- err: // 本RPC调用失败,返回错误信息 return e case <- result: // 本RPC调用成功,不返回错误信息 return nil } } func main() { ctx, cancel := context.WithCancel(context.Background()) // RPC1调用 err := Rpc(ctx, "http://rpc_1_url") if err != nil { return } wg := sync.WaitGroup{} // RPC2调用 wg.Add(1) go func(){ defer wg.Done() err := Rpc(ctx, "http://rpc_2_url") if err != nil { cancel() } }() // RPC3调用 wg.Add(1) go func(){ defer wg.Done() err := Rpc(ctx, "http://rpc_3_url") if err != nil { cancel() } }() // RPC4调用 wg.Add(1) go func(){ defer wg.Done() err := Rpc(ctx, "http://rpc_4_url") if err != nil { cancel() } }() wg.Wait() }
當然我這裡使用了waitGroup來保證main函數在所有RPC呼叫完成之後才會退出。
在Rpc函數中,第一個參數是一個CancelContext, 這個Context形象的說,就是一個傳話筒,在創建CancelContext的時候,返回了一個聽聲器(ctx)和麥克風(cancel函數)。所有的goroutine都拿著這個聽聲器(ctx),當主goroutine想要告訴所有goroutine要結束的時候,透過cancel函數把結束的訊息告訴給所有的goroutine。當然所有的goroutine都需要內建處理這個聽聲器結束訊號的邏輯(ctx->Done())。我們可以看Rpc函式內部,透過一個select來判斷ctx的done和目前的rpc呼叫哪個先結束。
這個waitGroup和其中一個RPC呼叫就通知所有RPC的邏輯,其實有一個套件已經幫我們做好了。 errorGroup。具體這個errorGroup包的使用可以看這個包的test例子。
有人可能會擔心我們這裡的cancel()會被多次調用,context套件的cancel調用是冪等的。可以放心多次調用。
我們這裡不妨品一下,這裡的Rpc函數,實際上我們的這個例子裡面是一個「阻塞式」的請求,這個請求如果是使用http.Get或http.Post來實現,實際上Rpc函數的Goroutine結束了,內部的那個實際的http.Get卻沒有結束。所以,需要理解下,這裡的函數最好是「非阻塞」的,例如是http.Do,然後可以用某種方式中斷。例如像這篇文章Cancel http.Request using Context中的這個例子:
func httpRequest( ctx context.Context, client *http.Client, req *http.Request, respChan chan []byte, errChan chan error ) { req = req.WithContext(ctx) tr := &http.Transport{} client.Transport = tr go func() { resp, err := client.Do(req) if err != nil { errChan <- err } if resp != nil { defer resp.Body.Close() respData, err := ioutil.ReadAll(resp.Body) if err != nil { errChan <- err } respChan <- respData } else { errChan <- errors.New("HTTP request failed") } }() for { select { case <-ctx.Done(): tr.CancelRequest(req) errChan <- errors.New("HTTP request cancelled") return case <-errChan: tr.CancelRequest(req) return } } }
它使用了http.Client.Do,然後接收到ctx.Done的時候,透過呼叫transport.CancelRequest來進行結束。
我們也可以參考net/dail/DialContext
換而言之,如果你希望你實現的包是「可中止/可控制」的,那麼你在你包實現的函數裡面,最好是能接收一個Context函數,並且處理了Context.Done。
場景二:PipeLine
pipeline模式就是管線模型,管線模型,管線上的幾個工人,有n個產品,一個一個產品進行組裝。其實pipeline模型的實作和Context並無關係,沒有context我們也能用chan實作pipeline模型。但是對於整條管線的控制,則是需要使用上Context的。這篇文章Pipeline Patterns in Go的例子是非常好的說明。這裡就大致對這個程式碼進行下說明。
runSimplePipeline的管線工人有三個,lineListSource負責將參數一個個分割進行傳輸,lineParser負責將字串處理成int64,sink根據具體的值判斷這個資料是否可用。他們所有的返回值基本上都有兩個chan,一個用於傳遞數據,一個用於傳遞錯誤。 (<-chan string, <-chan error)輸入基本上也都有兩個值,一個是Context,用於傳聲控制的,一個是(in <-chan)輸入產品的。
我們可以看到,這三個工人的具體函數裡面,都使用switch處理了case <-ctx.Done()。這個就是生產線上的命令控制。
func lineParser(ctx context.Context, base int, in <-chan string) ( <-chan int64, <-chan error, error) { ... go func() { defer close(out) defer close(errc) for line := range in { n, err := strconv.ParseInt(line, base, 64) if err != nil { errc <- err return } select { case out <- n: case <-ctx.Done(): return } } }() return out, errc, nil }
場景三:超時請求
我們發送RPC請求的時候,往往希望對這個請求進行一個逾時的限制。當一個RPC請求超過10s的請求,自動斷開。當然我們使用CancelContext,也能實現這個功能(開啟一個新的goroutine,這個goroutine拿著cancel函數,當時間到了,就呼叫cancel函數)。
鑑於這個需求是非常常見的,context套件也實現了這個需求:timerCtx。具體實例化的方法是 WithDeadline 和 WithTimeout。
具體的timerCtx裡面的邏輯也就是透過time.AfterFunc來呼叫ctx.cancel的。
官方的例子:
package main import ( "context" "fmt" "time" ) func main() { ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond) defer cancel() select { case <-time.After(1 * time.Second): fmt.Println("overslept") case <-ctx.Done(): fmt.Println(ctx.Err()) // prints "context deadline exceeded" } }
在http的客戶端裡面加上timeout也是一個常見的辦法
uri := "https://httpbin.org/delay/3" req, err := http.NewRequest("GET", uri, nil) if err != nil { log.Fatalf("http.NewRequest() failed with '%s'\n", err) } ctx, _ := context.WithTimeout(context.Background(), time.Millisecond*100) req = req.WithContext(ctx) resp, err := http.DefaultClient.Do(req) if err != nil { log.Fatalf("http.DefaultClient.Do() failed with:\n'%s'\n", err) } defer resp.Body.Close()
在http服務端設定一個timeout如何做呢?
package main import ( "net/http" "time" ) func test(w http.ResponseWriter, r *http.Request) { time.Sleep(20 * time.Second) w.Write([]byte("test")) } func main() { http.HandleFunc("/", test) timeoutHandler := http.TimeoutHandler(http.DefaultServeMux, 5 * time.Second, "timeout") http.ListenAndServe(":8080", timeoutHandler) }
我们看看TimeoutHandler的内部,本质上也是通过context.WithTimeout来做处理。
func (h *timeoutHandler) ServeHTTP(w ResponseWriter, r *Request) { ... ctx, cancelCtx = context.WithTimeout(r.Context(), h.dt) defer cancelCtx() ... go func() { ... h.handler.ServeHTTP(tw, r) }() select { ... case <-ctx.Done(): ... } }
场景四:HTTP服务器的request互相传递数据
context还提供了valueCtx的数据结构。
这个valueCtx最经常使用的场景就是在一个http服务器中,在request中传递一个特定值,比如有一个中间件,做cookie验证,然后把验证后的用户名存放在request中。
我们可以看到,官方的request里面是包含了Context的,并且提供了WithContext的方法进行context的替换。
package main import ( "net/http" "context" ) type FooKey string var UserName = FooKey("user-name") var UserId = FooKey("user-id") func foo(next http.HandlerFunc) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { ctx := context.WithValue(r.Context(), UserId, "1") ctx2 := context.WithValue(ctx, UserName, "yejianfeng") next(w, r.WithContext(ctx2)) } } func GetUserName(context context.Context) string { if ret, ok := context.Value(UserName).(string); ok { return ret } return "" } func GetUserId(context context.Context) string { if ret, ok := context.Value(UserId).(string); ok { return ret } return "" } func test(w http.ResponseWriter, r *http.Request) { w.Write([]byte("welcome: ")) w.Write([]byte(GetUserId(r.Context()))) w.Write([]byte(" ")) w.Write([]byte(GetUserName(r.Context()))) } func main() { http.Handle("/", foo(test)) http.ListenAndServe(":8080", nil) }
在使用ValueCtx的时候需要注意一点,这里的key不应该设置成为普通的String或者Int类型,为了防止不同的中间件对这个key的覆盖。最好的情况是每个中间件使用一个自定义的key类型,比如这里的FooKey,而且获取Value的逻辑尽量也抽取出来作为一个函数,放在这个middleware的同包中。这样,就会有效避免不同包设置相同的key的冲突问题了。
以上是你知道golang中Context的使用場景有哪些嗎的詳細內容。更多資訊請關注PHP中文網其他相關文章!