许多使用去的人,都会用到它的上下文库。大多数使用上下文进行下游操作,比如发出HTTP调用,或者从数据库获取数据,或者在协程中执行异步操作。最常见的用法是传递可由所有下游操作使用的公共数据。然而,一个不太为人所知,但非常有用的上下文特性是,它能够在中途取消或停止一个操作。
本篇文章将解释我们如何利用上下文库的取消特性,并通过一些模式和最佳实践来使用取消,使你的程序更快,更健壮。
简而言之,我们需要取消,以防止我们的系统做不不需要的工作。
考虑HTTP服务器对数据库的调用的常见情况,并将查询的数据返回给客户端:
时间图,如果一切都很完美,就会是这样的:
但是,如果客户端取消了中间的请求,会发生什么呢?例如,如果客户端关闭了他们的浏览器,这可能会发生。如果没有取消,应用服务器和数据库将继续执行它们的工作,即使工作的结果将被浪费:
理想情况下,如果我们知道进程(在本例中是HTTP请求)停止了,我们希望流程的所有下游组件停止工作:
现在我们知道了为什么需要取消,让我们来看看如何实现它。因为“取消”的事件与交易或正在执行的操作高度相关,所以它与上下文捆绑在一起是很自然的。
取消的有两个方面,你可能想要实现:
上下文类型提供了完成()方法,每当上下文收到取消事件时,它都会返回接收空结构{}类型的通道。监听取消事件就像等待& lt; -ctx.done()一样简单。
例如,让我们考虑一个HTTP服务器,它需要两秒钟来处理一个事件。如果在此之前请求被取消,我们希望立即返回:
函数main () {//创建一个HTTP服务器,听>
{func operation1 (ctx context.Context)错误//假设这个操作由于某种原因失败//我们使用时间。睡眠来模拟一个资源密集型操作
时间。睡眠(100 * time.Millisecond)
返回errors.New(“失败”)
}
func operation2 (ctx context.Context) {//我们使用一个类似的模式到HTTP服务器//我们看到在前面的例子
选择{
& lt; -时间。后(500 * time.Millisecond):
fmt.Println(“完成”)
& lt; -ctx.Done ():
fmt。Println(“停止operation2”)
}
}
函数main () {//创建一个新的上下文
ctx:=context.Background ()//创建一个新的上下文,取消功能//从原始上下文
ctx、取消:=context.WithCancel (ctx)//运行两个操作:>//背景3秒后将会被取消//如果需要取消之前,“取消”功能//使用,就像之前
ctx、取消:=上下文。3 * time.Second WithTimeout (ctx)//上下文将被取消>
函数main () {//创建一个新的上下文//截止日期的100毫秒
ctx:=context.Background ()
ctx _=上下文。100 * time.Millisecond WithTimeout (ctx)//请求,调用谷歌主页
点播,_:=http.NewRequest (http。MethodGet,“http://google.com”, nil)//联系我们刚刚创建的可删除的上下文的请求
要求=req.WithContext (ctx)//创建一个新的HTTP客户端和执行请求
客户:=,http.Client {}
res,犯错:=client.Do(要求)//如果请求失败,日志发送到STDOUT
如果犯错!=nil {
fmt。Println(“请求失败:犯错)
返回
}//打印statuscode如果请求成功
fmt。Println(“接收到的响应状态码:“res.StatusCode)
}
根据谷歌主页对你的请求的响应速度,你将收到:
以前接收到的响应状态码:200
或者
请求失败:http://google.com:上下文期限超过了
你可以使用超时来实现上述两个结果。
尽管走的上下文取消是一个通用的工具,但是在继续之前,有一些事情是你应该记住的。其中最重要的一点是,上下文只能被取消一次。
