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/golang-select-time-implement-timeout/