Go語言實(shí)現(xiàn)超時(shí)的三種方法實(shí)例
前言
超時(shí),指一個(gè)協(xié)程A開啟另一個(gè)協(xié)程B,A會(huì)阻塞等待B一段指定的時(shí)間,例如:5秒,A通知B結(jié)束(也有可能不通知,讓B繼續(xù)運(yùn)行)。也就是說,A就不愿意阻塞等待太久。
Go語言有多種方法實(shí)現(xiàn)這種超時(shí),我總結(jié)出3種:
方法一:用兩個(gè)通道 + A協(xié)程sleep
一個(gè)通道用來傳數(shù)據(jù),一個(gè)用來傳停止信號(hào)。
package main import ( "fmt" "time" ) // 老師視頻里的生產(chǎn)者消費(fèi)者 func main() { //知識(shí)點(diǎn): 老師這里用了兩個(gè)線程,一個(gè)用個(gè)傳數(shù)據(jù),一個(gè)用來傳關(guān)閉信號(hào) messages := make(chan int, 10) done := make(chan bool) defer close(messages) // consumer go func() { ticker := time.NewTicker(1 * time.Second) for range ticker.C { select { case <-done: fmt.Println("child process interrupt...") // 數(shù)據(jù)還沒收完,就被停止了。 return default: fmt.Printf("receive message:%d\n", <-messages) } } }() // producer for i := 0; i < 10; i++ { messages <- i } // 5秒后主線程關(guān)閉done通道 time.Sleep(5 * time.Second) close(done) time.Sleep(1 * time.Second) fmt.Println("main process exit!") }
程序輸出如下:
receive message:0
receive message:1
receive message:2
receive message:3
child process interrupt...
main process exit!
方法二:使用Timer(定時(shí)器)
這種方法也方法一類似,只不過是用一個(gè)Timer代替通道。
package main import ( "fmt" "time" ) //知識(shí)點(diǎn): // 1) 多通道 // 2) 定時(shí)器 func main() { ch1 := make(chan int, 10) go func(ch chan<- int) { // 假設(shè)子協(xié)程j是一個(gè)耗時(shí)操作,例如訪問網(wǎng)絡(luò),要10秒后才會(huì)有數(shù)據(jù) time.Sleep(10 * time.Second) ch <- 1 }(ch1) timer := time.NewTimer(5 * time.Second) // 設(shè)置定時(shí)器的超時(shí)時(shí)間,主線程只等5秒 fmt.Println("select start....") // 知識(shí)點(diǎn):主協(xié)程等待子線程,并有超時(shí)機(jī)制 select { case <-ch1: fmt.Println("從channel 1 收到一個(gè)數(shù)字") case <-timer.C: // 定時(shí)器也是一個(gè)通道 fmt.Println("5秒到了,超時(shí)了,main協(xié)程不等了") } fmt.Println("done!") }
程序輸出如下:
select start....
5秒到了,超時(shí)了,main協(xié)程不等了
done!
方法三:使用context.WithTimeout
下面的例子比較復(fù)雜,基于 Channel 編寫一個(gè)簡單的單協(xié)程生產(chǎn)者消費(fèi)者模型。
要求如下:
1)隊(duì)列:隊(duì)列長度 10,隊(duì)列元素類型為 int
2)生產(chǎn)者:每 1 秒往隊(duì)列中放入一個(gè)類型為 int 的元素,隊(duì)列滿時(shí)生產(chǎn)者可以阻塞
3)消費(fèi)者:每2秒從隊(duì)列中獲取一個(gè)元素并打印,隊(duì)列為空時(shí)消費(fèi)者阻塞
4)主協(xié)程30秒后要求所有子協(xié)程退出。
5)要求優(yōu)雅退出,即消費(fèi)者協(xié)程退出前,要先消費(fèi)完所有的int
6)通過入?yún)⒅С謨煞N運(yùn)行模式:
- wb(溫飽模式)生產(chǎn)速度快過消費(fèi)速度、
- je(饑餓模式)生產(chǎn)速度慢于消費(fèi)速度
context.WithTimeout見第87行。
package main import ( "context" "flag" "fmt" "sync" "time" ) // 課后練習(xí) 1.2 // 基于 Channel 編寫一個(gè)簡單的單協(xié)程生產(chǎn)者消費(fèi)者模型。 // 要求如下: // 1)隊(duì)列:隊(duì)列長度 10,隊(duì)列元素類型為 int // 2)生產(chǎn)者:每 1 秒往隊(duì)列中放入一個(gè)類型為 int 的元素,隊(duì)列滿時(shí)生產(chǎn)者可以阻塞 // 3)消費(fèi)者:每2秒從隊(duì)列中獲取一個(gè)元素并打印,隊(duì)列為空時(shí)消費(fèi)者阻塞 // 4)主協(xié)程30秒后要求所有子協(xié)程退出。 // 5)要求優(yōu)雅退出,即消費(fèi)者協(xié)程退出前,要先消費(fèi)完所有的int。 // 知識(shí)點(diǎn): // 1) 切片的零值也是可用的。 // 2) context.WithTimeout var ( wg sync.WaitGroup p Producer c Consumer ) type Producer struct { Time int Interval int } type Consumer struct { Producer } func (p Producer) produce(queue chan<- int, ctx context.Context) { go func() { LOOP: for { p.Time = p.Time + 1 queue <- p.Time fmt.Printf("生產(chǎn)者進(jìn)行第%d次生產(chǎn),值:%d\n", p.Time, p.Time) time.Sleep(time.Duration(p.Interval) * time.Second) select { case <-ctx.Done(): close(queue) break LOOP } } wg.Done() }() } func (c Consumer) consume(queue <-chan int, ctx context.Context) { go func() { LOOP: for { c.Time++ val := <-queue fmt.Printf("-->消費(fèi)者進(jìn)行第%d次消費(fèi),值:%d\n", c.Time, val) time.Sleep(time.Duration(c.Interval) * time.Second) select { case <-ctx.Done(): //remains := new([]int) //remains := []int{} var remains []int // 知識(shí)點(diǎn):切片的零值也是可用的。 for val = range queue { remains = append(remains, val) fmt.Printf("-->消費(fèi)者: 最后一次消費(fèi), 值為:%v\n", remains) break LOOP } } } wg.Done() }() } func main() { wg.Add(2) // 知識(shí)點(diǎn):context.Timeout timeout := 30 ctx, _ := context.WithTimeout(context.Background(), time.Duration(timeout)*time.Second) queue := make(chan int, 10) p.produce(queue, ctx) fmt.Println("main waiting...") wg.Wait() fmt.Println("done") } /* 啟動(dòng)命令: $ go run main/main.go -m wb $ go run main/main.go -m je */ func init() { // 解析程序入?yún)ⅲ\(yùn)行模式 mode := flag.String("m", "wb", "請(qǐng)輸入運(yùn)行模式:\nwb(溫飽模式)生產(chǎn)速度快過消費(fèi)速度、\nje(饑餓模式)生產(chǎn)速度慢于消費(fèi)速度)") flag.Parse() p = Producer{} c = Consumer{} if *mode == "wb" { fmt.Println("運(yùn)行模式:wb(溫飽模式)生產(chǎn)速度快過消費(fèi)速度") p.Interval = 1 // 每隔1秒生產(chǎn)一次 c.Interval = 5 // 每隔5秒消費(fèi)一次 // p = Producer{Interval: 1} // c = Consumer{Interval: 5} // 這一行會(huì)報(bào)錯(cuò),為什么? } else { fmt.Println("運(yùn)行模式:je(饑餓模式)生產(chǎn)速度慢于消費(fèi)速度") p.Interval = 5 // 每隔5秒生產(chǎn)一次 c.Interval = 1 // 每隔1秒消費(fèi)一次 } }
wb(溫飽模式)生產(chǎn)速度快過消費(fèi)速度,輸出如下:
運(yùn)行模式:wb(溫飽模式)生產(chǎn)速度快過消費(fèi)速度
生產(chǎn)者: 第1次生產(chǎn), 值為:1
-->消費(fèi)者: 第1次消費(fèi), 值為:1
生產(chǎn)者: 第2次生產(chǎn), 值為:2
生產(chǎn)者: 第3次生產(chǎn), 值為:3
生產(chǎn)者: 第4次生產(chǎn), 值為:4
生產(chǎn)者: 第5次生產(chǎn), 值為:5
-->消費(fèi)者: 第2次消費(fèi), 值為:2
生產(chǎn)者: 第6次生產(chǎn), 值為:6
生產(chǎn)者: 第7次生產(chǎn), 值為:7
生產(chǎn)者: 第8次生產(chǎn), 值為:8
生產(chǎn)者: 第9次生產(chǎn), 值為:9
生產(chǎn)者: 第10次生產(chǎn), 值為:10
-->消費(fèi)者: 第3次消費(fèi), 值為:3
生產(chǎn)者: 第11次生產(chǎn), 值為:11
生產(chǎn)者: 第12次生產(chǎn), 值為:12
生產(chǎn)者: 第13次生產(chǎn), 值為:13
-->消費(fèi)者: 第4次消費(fèi), 值為:4
生產(chǎn)者: 第14次生產(chǎn), 值為:14
-->消費(fèi)者: 第5次消費(fèi), 值為:5
生產(chǎn)者: 第15次生產(chǎn), 值為:15
生產(chǎn)者: 第16次生產(chǎn), 值為:16
-->消費(fèi)者: 第6次消費(fèi), 值為:6
main waiting
生產(chǎn)者: 第17次生產(chǎn), 值為:17
-->消費(fèi)者: 最后一次消費(fèi), 值為:[7 8 9 10 11 12 13 14 15 16 17]
-- done --
je(饑餓模式)生產(chǎn)速度慢于消費(fèi)速度,輸出如下:
運(yùn)行模式:je(饑餓模式)生產(chǎn)速度慢于消費(fèi)速度
-->消費(fèi)者: 第1次消費(fèi), 值為:1
生產(chǎn)者: 第1次生產(chǎn), 值為:1
生產(chǎn)者: 第2次生產(chǎn), 值為:2
-->消費(fèi)者: 第2次消費(fèi), 值為:2
生產(chǎn)者: 第3次生產(chǎn), 值為:3
-->消費(fèi)者: 第3次消費(fèi), 值為:3
生產(chǎn)者: 第4次生產(chǎn), 值為:4
-->消費(fèi)者: 第4次消費(fèi), 值為:4
生產(chǎn)者: 第5次生產(chǎn), 值為:5
-->消費(fèi)者: 第5次消費(fèi), 值為:5
生產(chǎn)者: 第6次生產(chǎn), 值為:6
-->消費(fèi)者: 第6次消費(fèi), 值為:6
main waiting
-->消費(fèi)者: 第7次消費(fèi), 值為:0
附:go 實(shí)現(xiàn)超時(shí)退出
之前手寫rpc框架的時(shí)候,吃多了網(wǎng)絡(luò)超時(shí)處理的苦,今天偶然發(fā)現(xiàn)了實(shí)現(xiàn)超時(shí)退出的方法,MARK
func AsyncCall() { ctx, cancel := context.WithTimeout(context.Background(), time.Duration(time.Millisecond*800)) defer cancel() go func(ctx context.Context) { // 發(fā)送HTTP請(qǐng)求 }() select { case <-ctx.Done(): fmt.Println("call successfully!!!") return case <-time.After(time.Duration(time.Millisecond * 900)): fmt.Println("timeout!!!") return } } //2 func AsyncCall() { ctx, cancel := context.WithTimeout(context.Background(), time.Duration(time.Millisecond * 800)) defer cancel() timer := time.NewTimer(time.Duration(time.Millisecond * 900)) go func(ctx context.Context) { // 發(fā)送HTTP請(qǐng)求 }() select { case <-ctx.Done(): timer.Stop() timer.Reset(time.Second) fmt.Println("call successfully!!!") return case <-timer.C: fmt.Println("timeout!!!") return } } //3 func AsyncCall() { ctx := context.Background() done := make(chan struct{}, 1) go func(ctx context.Context) { // 發(fā)送HTTP請(qǐng)求 done <- struct{}{} }() select { case <-done: fmt.Println("call successfully!!!") return case <-time.After(time.Duration(800 * time.Millisecond)): fmt.Println("timeout!!!") return } }
總結(jié)
到此這篇關(guān)于Go語言實(shí)現(xiàn)超時(shí)的三種方法的文章就介紹到這了,更多相關(guān)Go語言實(shí)現(xiàn)超時(shí)方法內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
使用Go語言開發(fā)一個(gè)高并發(fā)系統(tǒng)
高并發(fā)系統(tǒng)是指能同時(shí)支持眾多用戶請(qǐng)求,處理大量并行計(jì)算的系統(tǒng),這篇文章主要為大家詳細(xì)介紹了如何使用Go語言開發(fā)一個(gè)高并發(fā)系統(tǒng),感興趣的小伙伴可以了解下2023-11-11Go并發(fā)讀寫文件、分片寫、分片下載文件的實(shí)現(xiàn)示例
讀寫文件在很多項(xiàng)目中都可以用到,本文主要介紹了Go并發(fā)讀寫文件、分片寫、分片下載文件的實(shí)現(xiàn)示例,具有一定的參考價(jià)值,感興趣的可以了解一下2024-01-01VSCode1.4 搭建Golang的開發(fā)調(diào)試環(huán)境(遇到很多問題)
這篇文章主要介紹了VSCode1.4 搭建Golang的開發(fā)調(diào)試環(huán)境(遇到很多問題),文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-04-04golang實(shí)現(xiàn)頁面靜態(tài)化操作的示例代碼
這篇文章主要介紹了golang實(shí)現(xiàn)頁面靜態(tài)化操作的示例代碼,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-02-02goroutine?泄漏和避免泄漏實(shí)戰(zhàn)示例
這篇文章主要為大家介紹了goroutine?泄漏和避免泄漏實(shí)戰(zhàn)示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-12-12