这篇文章给大家分享的是有关怎么实现去超时控制的内容。小编觉得挺实用的,因此分享给大家做个参考,一起跟随小编过来看看吧。
为什么需要超时控制?
- <李>
请求时间过长,用户侧可能已经离开本页面了,服务端还在消耗资源处理,得到的结果没有意义
李> <李>过长时间的服务端处理会占用过多资源,导致并发能力下降,甚至出现不可用事故
李>去超时控制必要性
去正常都是用来写后端服务的,一般一个请求是由多个串行或并行的子任务来完成的,每个子任务可能是另外的内部请求,那么当这个请求超时的时候,我们就需要快速返回,释放占用的资源,比如goroutine,文件描述符等。
服务端常见的超时控制
- <李>
进程内的逻辑处理
李> <李>读写客户端请求,比如HTTP或者RPC请求
李> <李>调用其它服务端请求,包括调用RPC或者访问DB等
李>没有超时控制会怎样?
为了简化本文,我们以一个请求函数勤奋为例,用来做啥的不重要,顾名思义,可能处理起来比较慢。
func 勤奋(job 接口{}),error { time . sleep (time.Minute)才能 return 才能;零 } func requestWork (ctx context.Context, job 界面{}),error { ,return 勤奋(工作) }
这时客户端看到的就一直是大家熟悉的画面
& lt; img https://www.yisu.com/zixun/src=" https://gitee.com/kevwan/static/raw/master/doc/images/loading.jpg " width=" 25% ">
绝大部分用户都不会看一分钟菊花,早早弃你而去,空留了整个调用链路上一堆资源的占用,本文不究其它细节,只聚焦超时实现。
下面我们看看该怎么来实现超时,其中会有哪些坑。
第一版实现
大家可以先不往下看,自己试着想想该怎么实现这个函数的超时,第一次尝试:
func requestWork (ctx context.Context, job 界面{}),error { ,,ctx, cancel :=, context.WithTimeout (ctx, time.Second * 2) defer 才能;取消() done 才能:=,(chan 错误) go 才能;func (), { ,,,done & lt;作用;勤奋(工作) }()才能 select {才能 case 才能;err :=, & lt;完成: ,,,return 犯错 case 才能;& lt; -ctx.Done (): ,,,return ctx.Err () ,,} }
我们写个主要函数测试一下
func 主要(),{ const 才能;total =1000 var 才能;wg  sync.WaitGroup wg.Add才能(总) 你才能:=,time.Now () for 才能;小姐::=,0;,小姐:& lt;,总;,我+ +,{ ,,,go func (), { ,,,,,defer wg.Done () ,,,,,requestWork (context.Background (),“any") ,,,}() ,,} wg.Wait才能() fmt.Println才能(“运行:“,,time.Since(现在) }
跑一下试试效果
?运行超时。去
引用>
运行:2.005725931 s
超时已经生效。但这样就搞定了吗?
goroutine泄露
让我们在主函数末尾加一行代码看看执行完有多少goroutine
time . sleep (time.Minute * 2) fmt.Println (“number  of 了goroutine:“,, runtime.NumGoroutine ())睡眠2分钟是为了等待所有任务结束,然后我们打印一下当前goroutine数量。让我们执行一下看看结果
?运行超时。去
引用>
运行:2.005725931 s
了goroutine数量:1001
goroutine泄露了,让我们看看为啥会这样呢?首先,requestWork函数在2秒钟超时后就退出了,一旦requestWork函数退出,那么做通道就没有goroutine接收了,等到执行& lt;——勤奋(工作)这行代码的时候就会一直卡着写不进的去,导致每个超时的请求都会一直占用掉一个goroutine,这是一个很大的错误,等到资源耗尽的时候整个服务就失去响应了。
那么怎么解决呢?其实也很简单,只要让陈的时候把缓冲区大小设为1,如下:
done :=,使(chan 错误,,1)这样就可以让& lt;——勤奋(工作)不管在是否超时都能写入而不卡住goroutine。此时可能有人会问如果这时写入一个已经没goroutine接收的通道会不会有问题,在去里面通道不像我们常见的文件描述符一样,不是必须关闭的,只是个对象而已,关闭(频道)只是用来告诉接收者没有东西要写了,没有其它用途。
怎么实现去超时控制