前言
在开始之前，对time.After使用有疑问的朋友们可以看看这篇文章：http://www.zzvips.com/article/67050.html
我们在Golang网络编程中，经常要遇到设置超时的需求，本文就来给大家详细介绍了Go语言利用time.After实现超时控制的相关内容，下面话不多说了，来一起看看详细的介绍吧。
场景:
假设业务中需调用服务接口A，要求超时时间为5秒，那么如何优雅、简洁的实现呢？
我们可以采用select+time.After的方式，十分简单适用的实现。
首先，我们先看time.After()源码:
|
1
2
3
4
5
6
7
8
9
|
// 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类型的通道消息。那么，基于这个函数，就相当于实现了定时器，且是无阻塞的。
超时控制的代码实现:
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
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/golang-select-time-implement-timeout/
