快捷導(dǎo)航

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

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

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

Timer+Select

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

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

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

NewTimer

也可以新建一個NewTimer，timer.C返回也是一個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

留一個坑

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

Context+Select

第二種方案是利用 context,context.WithTimeout。
它接受一個Context和一個超時時間作為參數(shù)，返回一個子Context和一個取消函數(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 返回啟動了一個異步的定時器，返回一個 channel，當超過指定的時間后，該 channel 將會接受到信號。
啟動了子協(xié)程，函數(shù)執(zhí)行結(jié)束后，將向 channel ch 發(fā)送結(jié)束信號。
使用 select 阻塞等待 done 或 time.Done的信息，若超時，輸出timeout，若沒有超時，則輸出done。

output:

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

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

并發(fā)協(xié)程使用WaitGroup阻塞主協(xié)程保證生產(chǎn)協(xié)程和消費協(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()
}