快捷導(dǎo)航

Golang實(shí)現(xiàn)協(xié)程超時(shí)控制的方式總結(jié)

更新時(shí)間：2023年05月24日 10:31:49 作者：tracy小貓

我們知道，go協(xié)程如果不做好處理，很容易造成內(nèi)存泄漏，所以對goroutine做超時(shí)控制，才能有效避免這種情況發(fā)生，本文為大家整理了兩個(gè)常見的Golang超時(shí)控制方法，需要的可以收藏一下

Timer+Select

time.After 實(shí)現(xiàn)超時(shí)控制

利用 time.After 啟動(dòng)了一個(gè)異步的定時(shí)器，返回一個(gè) channel，當(dāng)超過指定的時(shí)間后，該 channel 將會(huì)接受到信號(hào)。
啟動(dòng)了子協(xié)程，函數(shù)執(zhí)行結(jié)束后，將向 channel ch 發(fā)送結(jié)束信號(hào)。
使用 select 阻塞等待 done 或 time.After 的信息，若超時(shí)，輸出timeout，若沒有超時(shí)，則輸出done。

func TestContext12(t *testing.T) {
    ch := make(chan struct{}, 1)
    go func() {
       fmt.Println("running...")
       time.Sleep(3 * time.Second)
       ch <- struct{}{}
   }()
    select {
       case <-ch:
       fmt.Println("done")
       case <-time.After(2 * time.Second):
       fmt.Println("timeout")
   }
}

output:

=== RUN TestContext12
running...
timeout
--- PASS: TestContext12 (3.01s)
PASS

這里要注意一個(gè)點(diǎn)，就是你創(chuàng)建ch 的時(shí)候，必須是帶緩沖的。如果不帶，在并發(fā)的情況下會(huì)怎樣呢？

NewTimer

也可以新建一個(gè)NewTimer，timer.C返回也是一個(gè)channel

func TestContext5(t *testing.T) {
   timer := time.NewTimer(time.Duration(time.Millisecond * 900))
   ch := make(chan struct{}, 1)
   go func() {
      fmt.Println("running...")
      time.Sleep(3 * time.Second)
      ch <- struct{}{}
   }()
   select {
   case <-ch:
      fmt.Println("done")
   case <-timer.C:
      fmt.Println("timeout")
   }
}

output:

=== RUN TestContext5
running...
timeout
--- PASS: TestContext5 (0.90s)
PASS

留一個(gè)坑

為什么不需要關(guān)閉ch 管道？什么時(shí)候關(guān)閉ch?

Context+Select

第二種方案是利用 context,context.WithTimeout。
它接受一個(gè)Context和一個(gè)超時(shí)時(shí)間作為參數(shù)，返回一個(gè)子Context和一個(gè)取消函數(shù)CancelFunc。
取消函數(shù)CancelFunc將釋放子Context與之有關(guān)的資源，因此在子Context中的相關(guān)操作一旦完成，應(yīng)該立即調(diào)用取消函數(shù)CancelFunc。

func TestContext6(t *testing.T) {
   ch := make(chan string)
   timeout, cancel := context.WithTimeout(context.Background(), 2*time.Second)
   defer cancel()
   go func() {
      time.Sleep(time.Second * 3)
      ch <- "done"
   }()
   select {
   case res := <-ch:
      fmt.Println(res)
   case <-timeout.Done():
      fmt.Println("timout", timeout.Err())
   }
}

利用 context.WithTimeou 返回啟動(dòng)了一個(gè)異步的定時(shí)器，返回一個(gè) channel，當(dāng)超過指定的時(shí)間后，該 channel 將會(huì)接受到信號(hào)。
啟動(dòng)了子協(xié)程，函數(shù)執(zhí)行結(jié)束后，將向 channel ch 發(fā)送結(jié)束信號(hào)。
使用 select 阻塞等待 done 或 time.Done的信息，若超時(shí)，輸出timeout，若沒有超時(shí)，則輸出done。

output:

=== RUN TestContext6
timout context deadline exceeded
--- PASS: TestContext6 (2.00s)
PASS

context+并發(fā)協(xié)程

并發(fā)協(xié)程使用WaitGroup阻塞主協(xié)程保證生產(chǎn)協(xié)程和消費(fèi)協(xié)程正常執(zhí)行完成。

func TestContext3(t *testing.T) {
   withTimeout, cancelFunc := context.WithTimeout(context.Background(), time.Second*1)
   waitGroup := sync.WaitGroup{}
   waitGroup.Add(2)
   go func() { //協(xié)程1
      time.Sleep(time.Second)
      fmt.Println("finished 1")
      waitGroup.Done()
   }()
   go func() { //協(xié)程2
      time.Sleep(time.Second * 2)
      fmt.Println("finished 2")
      waitGroup.Done()
   }()
   go func() {
      select {
      case <-withTimeout.Done():
         fmt.Println("timeout")
         return
      default:
         waitGroup.Wait()
         cancelFunc()
         fmt.Println("finished all")
         return //結(jié)束監(jiān)聽協(xié)程
      }
   }()
   <-withTimeout.Done()
}

開啟協(xié)程1，等待1秒輸出。協(xié)程2，等待2秒輸出。
context.WithTimeout等待1秒后返回函數(shù)。
1秒后，協(xié)程一輸出。但是協(xié)程二沒輸出，主線程不等待了。直接返回。

output:

=== RUN TestContext3
finished 1
--- PASS: TestContext3 (1.01s)
PASS

這里多一個(gè)問題，留個(gè)坑！

goroutine如何超時(shí)控制并發(fā)輸出a,b,c并且按照a,b,c的順序輸出？

到此這篇關(guān)于Golang實(shí)現(xiàn)協(xié)程超時(shí)控制的方式總結(jié)的文章就介紹到這了,更多相關(guān)Golang協(xié)程超時(shí)控制內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: