golang超时退出如何设置
在Go语言中,可使用context
包来设置超时退出。下面是一个示例:
package main
import (
"context"
"fmt"
"time"
)
func main() {
// 创建一个父级的context
ctx := context.Background()
// 创建一个带有超时的子级context,超时时间为3秒
ctx, cancel := context.WithTimeout(ctx, 3*time.Second)
defer cancel()
// 启动一个goroutine履行耗时操作
go func() {
// 摹拟一个耗时的操作,需要5秒
time.Sleep(5 * time.Second)
// 检查context是否是已超时
if ctx.Err() != nil {
fmt.Println("超时退出")
return
}
// 履行其他操作...
}()
// 等待context超时或被取消
select {
case <-ctx.Done():
fmt.Println(ctx.Err())
case <-time.After(10 * time.Second):
fmt.Println("正常退出")
}
}
在这个示例中,我们首先使用context.Background()
创建一个父级的context,然后使用context.WithTimeout()
创建一个带有超时的子级context,超时时间为3秒。
接着,我们在一个goroutine中履行耗时操作,这个操作需要5秒。在操作履行终了后,我们检查context是否是已超时,如果超时则打印"超时退出"并返回。
最后,我们使用select
语句来等待context超时或被取消。如果context超时,则打印超时毛病信息;如果context被取消,则打印取消毛病信息;如果超过10秒仍未超时或被取消,则打印"正常退出"。
TOP