之前有聊過 golang 的協(xié)程，我發(fā)覺似乎還很理論，特別是在并發(fā)安全上，所以特結(jié)合網(wǎng)上的一些例子，來試驗下go routine中 的 channel, select, context 的妙用。
場景-微服務(wù)調(diào)用
我們用 gin（一個web框架） 作為處理請求的工具，需求是這樣的：
一個請求 X 會去并行調(diào)用 A, B, C 三個方法，并把三個方法返回的結(jié)果加起來作為 X 請求的 Response。
但是我們這個 Response 是有時間要求的（不能超過3秒的響應(yīng)時間），可能 A, B, C 中任意一個或兩個，處理邏輯十分復(fù)雜，或者數(shù)據(jù)量超大，導(dǎo)致處理時間超出預(yù)期，那么我們就馬上切斷，并返回已經(jīng)拿到的任意個返回結(jié)果之和。
我們先來定義主函數(shù)：
func main() { r := gin.New() r.GET("/calculate", calHandler) http.ListenAndServe(":8008", r)}非常簡單，普通的請求接受和 handler 定義。其中 calHandler 是我們用來處理請求的函數(shù)。
分別定義三個假的微服務(wù)，其中第三個將會是我們超時的哪位~
func microService1() int { time.Sleep(1*time.Second) return 1}func microService2() int { time.Sleep(2*time.Second) return 2}func microService3() int { time.Sleep(10*time.Second) return 3}接下來，我們看看 calHandler 里到底是什么
func calHandler(c *gin.Context) { ...}要點1--并發(fā)調(diào)用
直接用 go 就好了嘛~
所以一開始我們可能就這么寫：
go microService1()go microService2()go microService3()
很簡單有沒有，但是等等，說好的返回值我怎么接呢？
為了能夠并行地接受處理結(jié)果，我們很容易想到用 channel 去接。
所以我們把調(diào)用服務(wù)改成這樣：
var resChan = make(chan int, 3) // 因為有3個結(jié)果，所以我們創(chuàng)建一個可以容納3個值的 int channel。go func() { resChan <- microService1()}()go func() { resChan <- microService2()}()go func() { resChan <- microService3()}()有東西接，那也要有方法去算，所以我們加一個一直循環(huán)拿 resChan 中結(jié)果并計算的方法：
var resContainer, sum intfor { resContainer = <-resChan sum += resContainer}這樣一來我們就有一個 sum 來計算每次從 resChan 中拿出的結(jié)果了。
要點2--超時信號
還沒結(jié)束，說好的超時處理呢？
為了實現(xiàn)超時處理，我們需要引入一個東西，就是 context，什么是 context ?
我們這里只使用 context 的一個特性，超時通知（其實這個特性完全可以用 channel 來替代）。
可以看在定義 calHandler 的時候我們已經(jīng)將 c *gin.Context 作為參數(shù)傳了進來，那我們就不用自己在聲明了。
gin.Context 簡單理解為貫穿整個 gin 聲明周期的上下文容器，有點像是分身，亦或是量子糾纏的感覺。
有了這個 gin.Context， 我們就能在一個地方對 context 做出操作，而其他正在使用 context 的函數(shù)或方法，也會感受到 context 做出的變化。
ctx, _ := context.WithTimeout(c, 3*time.Second) //定義一個超時的 context
只要時間到了，我們就能用 ctx.Done() 獲取到一個超時的 channel(通知)，然后其他用到這個 ctx 的地方也會停掉，并釋放 ctx。
一般來說，ctx.Done() 是結(jié)合 select 使用的。
所以我們又需要一個循環(huán)來監(jiān)聽 ctx.Done()
for { select { case <- ctx.Done(): // 返回結(jié)果}現(xiàn)在我們有兩個 for 了，是不是能夠合并下？
for { select { case resContainer = <-resChan: sum += resContainer fmt.Println("add", resContainer) case <- ctx.Done(): fmt.Println("result:", sum) return }}誒嘿，看上去不錯。
不過我們怎么在正常完成微服務(wù)調(diào)用的時候輸出結(jié)果呢？
看來我們還需要一個 flag
var count intfor { select { case resContainer = <-resChan: sum += resContainer count ++ fmt.Println("add", resContainer) if count > 2 { fmt.Println("result:", sum) return } case <- ctx.Done(): fmt.Println("timeout result:", sum) return }}我們加入一個計數(shù)器，因為我們只是調(diào)用3次微服務(wù)，所以當 count 大于2的時候，我們就應(yīng)該結(jié)束并輸出結(jié)果了。
要點3--并發(fā)中的等待
上面的計時器是一種偷懶的方法，因為我們知道了調(diào)用微服務(wù)的次數(shù)，如果我們并不知道，或者之后還要添加呢？
手動每次改 count 的判斷閾值會不會太沙雕了？這時候我們就要加入 sync 包了。
我們將會使用的 sync 的一個特性是 WaitGroup。它的作用是等待一組協(xié)程運行完畢后，執(zhí)行接下去的步驟。
我們來改下之前微服務(wù)調(diào)用的代碼塊：
var success = make(chan int, 1) // 成功的通道標識wg := sync.WaitGroup{} // 創(chuàng)建一個 waitGroup 組wg.Add(3) // 我們往組里加3個標識，因為我們要運行3個任務(wù)go func() { resChan <- microService1() wg.Done() // 完成一個，Done()一個}()go func() { resChan <- microService2() wg.Done()}()go func() { resChan <- microService3() wg.Done()}()wg.Wait() // 直到我們前面三個標識都被 Done 了，否則程序一直會阻塞在這里success <- 1 // 我們發(fā)送一個成功信號到通道中既然我們有了 success 這個信號，那么再把它加入到監(jiān)控 for 循環(huán)中，并做些修改，刪除原來 count 判斷的部分。
go func() { for { select { case resContainer = <-resChan: sum += resContainer fmt.Println("add", resContainer) case <- success: fmt.Println("result:", sum) return case <- ctx.Done(): fmt.Println("result:", sum) return } }}() 三個 case，分工明確，一個用來拿服務(wù)輸出的結(jié)果并計算，一個用來做最終的完成輸出，一個是超時輸出。
同時我們將這個循環(huán)監(jiān)聽，也作為協(xié)程運行。
至此，所有的主要代碼都完成了。下面是完全版
package mainimport ( "context" "fmt" "net/http" "sync" "time" "github.com/gin-gonic/gin")// 一個請求會觸發(fā)調(diào)用三個服務(wù)，每個服務(wù)輸出一個 int,// 請求要求結(jié)果為三個服務(wù)輸出 int 之和// 請求返回時間不超過3秒，大于3秒只輸出已經(jīng)獲得的 int 之和func calHandler(c *gin.Context) { var resContainer, sum int var success, resChan = make(chan int), make(chan int, 3) ctx, _ := context.WithTimeout(c, 3*time.Second) go func() { for { select { case resContainer = <-resChan: sum += resContainer fmt.Println("add", resContainer) case <- success: fmt.Println("result:", sum) return case <- ctx.Done(): fmt.Println("result:", sum) return } } }() wg := sync.WaitGroup{} wg.Add(3) go func() { resChan <- microService1() wg.Done() }() go func() { resChan <- microService2() wg.Done() }() go func() { resChan <- microService3() wg.Done() }() wg.Wait() success <- 1 return}func main() { r := gin.New() r.GET("/calculate", calHandler) http.ListenAndServe(":8008", r)}func microService1() int { time.Sleep(1*time.Second) return 1}func microService2() int { time.Sleep(2*time.Second) return 2}func microService3() int { time.Sleep(10*time.Second) return 3}上面的程序只是簡單描述了一個調(diào)用其他微服務(wù)超時的處理場景。
實際過程中還需要加很多很多調(diào)料，才能保證接口的對外完整性。
以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持VEVB武林網(wǎng)。
新聞熱點
疑難解答
