Go 採用 time.After 實現逾時控制

這是一個建立於 的文章，其中的資訊可能已經有所發展或是發生改變。

情境:

假設業務中需調用服務介面A，要求逾時時間為5秒，那麼如何優雅、簡潔的實現呢？

我們可以採用select+time.After的方式，十分簡單適用的實現。

首先，我們先看time.After()源碼:

// After waits for the duration to elapse and then sends the current time// on the returned channel.// It is equivalent to NewTimer(d).C.// The underlying Timer is not recovered by the garbage collector// until the timer fires. If efficiency is a concern, use NewTimer// instead and call Timer.Stop if the timer is no longer needed.func After(d Duration) <-chan Time {    return NewTimer(d).C}

time.After()表示time.Duration長的時候後返回一條time.Time類型的通道訊息。那麼，基於這個函數，就相當於實現了定時器，且是無阻塞的。

逾時控制的代碼實現:

package mainimport (    "time"    "fmt")func main() {    ch := make(chan string)    go func() {        time.Sleep(time.Second * 2)        ch <- "result"    }()    select {    case res := <-ch:        fmt.Println(res)    case <-time.After(time.Second * 1):        fmt.Println("timeout")    }}

我們使用channel來接收協程裡的業務傳回值。

select語句阻塞等待最先返回資料的channel，當先接收到time.After的通道資料時，select則會停止阻塞並執行該case的代碼。此時就已經實現了對業務代碼的逾時處理。

原文地址: https://shockerli.net/post/go...