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 main

import (
    "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...