超时取消

假设我们希望HTTP请求在给定时间内完成,超时自动取消。

首先定义超时上下文,设定时间返回取消函数(一旦超时用于清理资源)。调用取消函数取消后续操作,删除子上下文对父的引用。

还可以通过特定时间进行设定:

完整实例如下:

超时输出结果:

2022/12/27 14:36:00 Get "https://www.baidu.com": context deadline exceeded

我们可以调大超时时间,则能正常看到输出结果。

取消后续操作

有时请求被取消后,需要阻止系统继续做后续比必要的工作。请看下面用户发起的http请求,应用程序接收请求后查询数据库并返回查询结果:

正常流程如下:

但如果客户端取消了请求,如果没有取消,应用服务和数据库仍然继续工作,然后结果却不能反馈给客户端。理想状况为所有下游过程停止,如图所示:

考虑有两个相关操作的情况,“相关”的意思是如果一个失败了,另一个即使完成也没有意义了。如果已经知道前一个操作失败了,则希望取消所有相关的操作。请看示例:

由于我们设置操作2执行时间较长,而操作1很快就报错,因此输出结果为操作2被取消:

halted operation2

上下文传值

我们可以实用上下文变量在不同协程中传递值。

假设一个操作需要调用函数多次,其中用于标识的公共ID需要被 日志记录,请看示例:

这里在main函数中创建上下文,并采用键值对方式存储id值,从而后续函数调用时可以从上下文中获取该值。如图所示:

使用context变量在不同操作中传递信息非常有用,主要原因包括:

  • 线程安全: 一旦设置了上下文键,就不能修改它的值,可以使用context.WithValue方法可以设置新的值
  • 通用方法: 在Go的官方库和应用程序中大量使用上下文传递数据,我们当然最好也使用这种模式