如何在Go语言中使用超时
当我们不想等待某些goroutine(花费的时间超过预期)的输出时,超时起着重要的作用。需要注意的是,Go本身并不直接支持超时,但我们可以很容易地实现它。
假设我们需要从一个通道ch接收一些值,但我们不想等待超过3秒钟来接收值。如果3秒后我们才收到输出,那么我们希望丢弃它并打印不同的消息,而不是等待更长时间的输出。
示例1
让我们首先探讨一个简单的案例,其中我们从一个函数中获得输出需要较长时间。
请看下面的代码。
package main
import (
"fmt"
"time"
)
func timeConsuming() string {
time.Sleep(5 * time.Second)
return "The timeConsuming() function has stopped"
}
func main() {
currentChannel := make(chan string, 1)
go func() {
text := timeConsuming()
currentChannel <- text
}()
select {
case res := <-currentChannel:
fmt.Println(res)
}
fmt.Println("Main function exited!")
}在上面的代码中,我们有一个名为timeConsuming()的函数,它表示一个可能在较长时间或所需时间后返回特定值的函数。一个例子是网络请求,其中获取数据花费的时间太长,用户感到沮丧。
在上面代码的main函数中,我们有一个带缓冲区的通道,然后我们使用select语句等待数据的到来。因此,在上述情况下,整个代码必须等到函数timeConsuming()完成工作。
输出
如果我们使用命令go run main.go运行上面的代码,我们将得到以下输出。
The timeConsuming() function has stopped Main function exited!
示例2
现在,假设我们不想等待函数timeConsuming()完成其执行。在这种情况下,我们可以使用time包的After()函数。
语法
After()函数的语法是:
func After(d Duration) −- chan Time
After函数等待d持续时间结束后,它将返回一个通道上的当前时间。
请看下面的代码,我们使用After函数来注册超时。
package main
import (
"fmt"
"time"
)
func timeConsuming() string {
time.Sleep(5 * time.Second)
return "The timeConsuming() function has stopped"
}
func main() {
currentChannel := make(chan string, 1)
go func() {
text := timeConsuming()
currentChannel <- text
}()
select {
case res := <-currentChannel:
fmt.Println(res)
case <-time.After(3 * time.Second):
fmt.Println("Out of time :(")
}
fmt.Println("Main function exited!")
}输出
如果我们使用命令go run main.go运行上面的代码,我们将得到以下输出。
Out of time :( Main function exited!
广告
数据结构
网络
关系数据库管理系统 (RDBMS)
操作系统
Java
iOS
HTML
CSS
Android
Python
C语言编程
C++
C#
MongoDB
MySQL
Javascript
PHP