了解 Golang 上下文超时

Posted 2023-03-29

【中文标题】了解 Golang 上下文超时

【英文标题】：Understanding Golang Context timeout

【发布时间】：2019-05-09 15:47:28

【问题描述】：

我无法理解如何检查上下文是否超过了超时设置的截止日期，或者我是否应该检查？

这是来自 mongo-go-driver 的 sn-p：

client, err := NewClient("mongodb://foo:bar@localhost:27017")
if err != nil  return err 
ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
defer cancel()
err = client.Connect(ctx)
if err != nil  return err 

阅读此代码，我如何知道上下文是否超过了截止日期？从我天真的理解（或不理解）来看，err = client.Connect(ctx) 行会给我错误，包括超过期限（如果超过），因此我认为我什至不需要明确检查？

但是，当我浏览互联网以更好地了解上下文的工作原理时，我遇到了一些使用 select case 明确检查上下文的用法，如下所示（来自http://p.agnihotry.com/post/understanding_the_context_package_in_golang/ 的代码 sn-p）：

//Use a select statement to exit out if context expires
  select 
  case <-ctx.Done():
    fmt.Println("sleepRandomContext: Time to return")
  case sleeptime := <-sleeptimeChan:
    //This case is selected when processing finishes before the context is cancelled
    fmt.Println("Slept for ", sleeptime, "ms")
  

我应该明确地检查它吗？如果不是，我应该什么时候使用显式检查？

【参考方案1】：

问题第二部分中的select 代码是Connect 方法中的代码可能是什么样子。它正在检查ctx.Done() 是否准备好发送。如果是，则上下文被取消，因为发生超时，或者因为调用了cancel()。

错误是值。就这样对待他们。如果了解错误原因对您很重要（网络故障？意外数据？超时？），那么您应该进行检查并采取相应措施。如果无论原因如何，从错误中恢复都是相同的，那么检查错误就没有那么重要了。

【讨论】：

你能确认我理解你在说什么吗？因此，如果我使用像mongo-go-driver 这样的外包包，并且它们的功能包含上下文，我可以放心地假设它们会为我处理上下文？意思是在某个地方，如果超过超时，这个函数只会返回一个错误（可能会说“超过最后期限”）。最重要的是，我通常只担心 err != nil?

@Leesa 关闭，但不完全。您最终也会得到 cancel 函数，您可以调用该函数，并且驱动程序应该但不是必须遵守。但是查看代码，您至少可以在一个地方看到他们确实检查了取消/超时上下文。

【参考方案2】：

context 最适合用于管理事物的生命周期。因此，如果您正在做某事（例如，处理文件），那么您应该检查上下文以确保该进程尚未被处理。

有两种方法可以检查context 是否已被取消或超时：

Context.Done() - 这将返回一个在上下文被取消时将关闭的通道。 Context.Err() - 如果上下文被取消，这将返回错误。

选择最适合您的需求。如果您使用频道做事，那么在select 中使用context.Done() 效果很好。如果您在读取io.Reader 时使用for 循环，那么检查context.Err() 会更容易。

话虽如此，如果你开始的东西也需要一个上下文，你应该总是使用你已经拥有的上下文，如果你想在你被取消的情况下取消那个新的“东西”。