golang 里出現(xiàn)多 goroutine 的場景很常見, 最常用的兩種方式就是 WaitGroup 和 Context, 今天我們了解一下 Context 的應(yīng)用場景
使用場景
場景一: 多goroutine執(zhí)行超時(shí)通知
并發(fā)執(zhí)行的業(yè)務(wù)中最常見的就是有協(xié)程執(zhí)行超時(shí), 如果不做超時(shí)處理就會出現(xiàn)一個(gè)僵尸進(jìn)程, 這累計(jì)的多了就會有一陣手忙腳亂了, 所以我們要在源頭上就避免它們
看下面這個(gè)示例:
package main
import (
"context"
"fmt"
"time"
)
/**
同一個(gè)content可以控制多個(gè)goroutine, 確保線程可控, 而不是每新建一個(gè)goroutine就要有一個(gè)chan去通知他關(guān)閉
有了他代碼更加簡潔
*/
func main() {
fmt.Println("run demo \n\n\n")
demo()
}
func demo() {
ctx, cancel := context.WithTimeout(context.Background(), 9*time.Second)
go watch(ctx, "[線程1]")
go watch(ctx, "[線程2]")
go watch(ctx, "[線程3]")
index := 0
for {
index++
fmt.Printf("%d 秒過去了 \n", index)
time.Sleep(1 * time.Second)
if index > 10 {
break
}
}
fmt.Println("通知停止監(jiān)控")
// 其實(shí)此時(shí)已經(jīng)超時(shí), 協(xié)程已經(jīng)提前退出
cancel()
// 防止主進(jìn)程提前退出
time.Sleep(3 * time.Second)
fmt.Println("done")
}
func watch(ctx context.Context, name string) {
for {
select {
case -ctx.Done():
fmt.Printf("%s 監(jiān)控退出, 停止了...\n", name)
return
default:
fmt.Printf("%s goroutine監(jiān)控中... \n", name)
time.Sleep(2 * time.Second)
}
}
}
使用 context.WithTimeout() 給文本流設(shè)置一個(gè)時(shí)間上限, 結(jié)合 for+select 去接收消息. 當(dāng)執(zhí)行超時(shí),或手動關(guān)閉都會給 -ctx.Done() 發(fā)送消息,而且所有使用同一個(gè) context 都會收到這個(gè)通知, 免去了一個(gè)一個(gè)通知的繁瑣代碼
場景二: 類似web服務(wù)器中的session
比如在php中(沒用swoole擴(kuò)展), 一個(gè)請求進(jìn)來, 從 $_REQUEST $_SERVER 能獲取到的是有關(guān)這一條請求的所有信息, 哪怕是使用全局變量也是給這一個(gè)請求來服務(wù)的, 是線程安全的
但是 golang 就不一樣了, 因?yàn)槌绦虮旧砭湍芷鹨粋€(gè) web sever, 因此就不能隨便使用全局變量了, 不然就是內(nèi)存泄露警告. 但是實(shí)際業(yè)務(wù)當(dāng)中需要有一個(gè)類似session 的東西來承載單次請求的信息, 舉一個(gè)具體的例子就是: 給每次請求加一個(gè) uniqueID 該如何處理? 有了這個(gè) uniqueID, 請求的所有日志都能帶上它, 這樣排查問題的時(shí)候方便追蹤一次請求發(fā)生了什么
如下:
func demo2() {
pCtx, pCancel := context.WithCancel(context.Background())
pCtx = context.WithValue(pCtx, "parentKey", "parentVale")
go watch(pCtx, "[父進(jìn)程1]")
go watch(pCtx, "[父進(jìn)程2]")
cCtx, cCancel := context.WithCancel(pCtx)
go watch(cCtx, "[子進(jìn)程1]")
go watch(cCtx, "[子進(jìn)程2]")
fmt.Println(pCtx.Value("parentKey"))
fmt.Println(cCtx.Value("parentKey"))
time.Sleep(10 * time.Second)
fmt.Println("子進(jìn)程關(guān)閉")
cCancel()
time.Sleep(5 * time.Second)
fmt.Println("父進(jìn)程關(guān)閉")
pCancel()
time.Sleep(3 * time.Second)
fmt.Println("done")
}
最開始的 context.WithCancel(context.Background()) 中 context.Background() 就是一個(gè)新建的 context, 利用 context 能繼承的特性, 可以將自己的程序構(gòu)建出一個(gè) context 樹, context 執(zhí)行 cancel() 將影響到當(dāng)前 context 和子 context, 不會影響到父級.
同時(shí) context.WithValue 也會給 context 帶上自定義的值, 這樣 uniqueID 就能輕松的傳遞了下去, 而不是一層層的傳遞參數(shù), 改func什么的
對于 context 很值得參考的應(yīng)用有:
Context 相關(guān) func 和接口
繼承 context 需要實(shí)現(xiàn)如下四個(gè)接口
type Context interface {
Deadline() (deadline time.Time, ok bool)
Done() -chan struct{}
Err() error
Value(key interface{}) interface{}
}
當(dāng)使用的時(shí)候不需要實(shí)現(xiàn)接口, 因?yàn)楣俜桨镆呀?jīng)基于 emptyCtx 實(shí)現(xiàn)了一個(gè), 調(diào)用方法有
var (
background = new(emptyCtx)
todo = new(emptyCtx)
)
// 這個(gè)是最初始的ctx, 之后的子ctx都是繼承自它
func Background() Context {
return background
}
// 不清楚context要干嘛, 但是就得有一個(gè)ctx的用這個(gè)
func TODO() Context {
return todo
}
繼承用的函數(shù)
func WithCancel(parent Context) (ctx Context, cancel CancelFunc)
func WithDeadline(parent Context, deadline time.Time) (Context, CancelFunc)
func WithTimeout(parent Context, timeout time.Duration) (Context, CancelFunc)
func WithValue(parent Context, key, val interface{}) Context
- WithCancel 返回一個(gè)帶 cancel 函數(shù)的ctx,
- WithDeadline 在到達(dá)指定時(shí)間時(shí)自動執(zhí)行 cancel()
- WithTimeout 是 WithDeadline的殼子, 區(qū)別就是這個(gè)函數(shù)是多少時(shí)間過后執(zhí)行 cancel
func WithTimeout(parent Context, timeout time.Duration) (Context, CancelFunc) {
return WithDeadline(parent, time.Now().Add(timeout))
}
WithValue 繼承父類ctx時(shí)順便帶上一個(gè)值
以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。
您可能感興趣的文章:- 快速解決Golang Map 并發(fā)讀寫安全的問題
- 淺談golang并發(fā)操作變量安全的問題
- golang高并發(fā)限流操作 ping / telnet
- golang gin 框架 異步同步 goroutine 并發(fā)操作
- Golang 實(shí)現(xiàn)分片讀取http超大文件流和并發(fā)控制
- golang-gin-mgo高并發(fā)服務(wù)器搭建教程
- golang 限制同一時(shí)間的并發(fā)量操作
- golang并發(fā)編程的實(shí)現(xiàn)
- Golang 并發(fā)以及通道的使用方式
