当前位置 博文首页 > Go语言利用time.After实现超时控制的方法详解

    Go语言利用time.After实现超时控制的方法详解

    作者:格物 时间:2021-07-02 17:46

    前言

    在开始之前,对time.After使用有疑问的朋友们可以看看这篇文章:http://blog.iis7.com/article/146063.htm

    我们在Golang网络编程中,经常要遇到设置超时的需求,本文就来给大家详细介绍了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的代码。此时就已经实现了对业务代码的超时处理。

    总结

    以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对站长博客的支持。

    jsjbwy
    下一篇:没有了