go - context.TODO() 或 context.Background()，我更喜欢哪一个？

Question

我目前正在将我们的代码从 global sign 包迁移到 go mongo-driver，不确定我应该在哪里使用 context.TODO() 和 context.Background()，这真的很令人困惑，我知道它都返回非零空，所以我应该在 main 函数和 init() 函数中使用 context.Background() ？并在其他地方使用 context.TODO() ？有人可以帮忙吗？

尝试检查我应该使用哪个参数 context.TODO() 或 context.Background()。

Answer 1

检查他们的文档:

context.Background() :

func Background() Context

Background returns a non-nil, empty Context. It is never canceled, has no values, and has no deadline. It is typically used by the main function, initialization, and tests, and as the top-level Context for incoming requests.

context.TODO() :

func TODO() Context

TODO returns a non-nil, empty Context. Code should use context.TODO when it's unclear which Context to use or it is not yet available (because the surrounding function has not yet been extended to accept a Context parameter).

根据文档，当您需要上下文但您(还)没有并且不知道使用什么时，请使用 context.TODO()。这正是您的情况，正确使用 context.TODO()记录。此外，静态分析工具和 IDE 可能会支持发现这些上下文，并在稍后向您发出警告以解决该问题。

另请注意，如果您在必须使用 mongo 驱动程序时确实有上下文，请考虑使用该上下文，或从中派生一个新上下文。

例如，如果您正在编写一个需要查询某些文档的 HTTP 处理程序，则 HTTP 请求 http.Request已经有一个可以使用 Request.Context() 访问的上下文.当您调用需要上下文的其他 API 函数时，这是一个主要的候选者。

为什么，你可能会问？因为当 HTTP 客户端放弃/中止请求时请求的上下文被取消，这意味着无论你做什么，客户端都不会收到它，所以例如如果 HTTP 处理程序只是提供信息，你也可以中止生成它，节省一些资源。因此，如果在您执行 MongoDB 查询时请求上下文被取消，您也可以取消查询，因为您无论如何都不需要(不会处理)结果。取消传递给查询执行的上下文(例如 Collection.Find())是告诉 MongoDB 服务器尽可能取消查询的方法(因为您不需要结果)。

因此在这种情况下使用请求上下文可以在 HTTP 服务器和 MongoDB 服务器上为您节省一些 CPU 时间和内存。

另一个例子:假设您必须在 10 秒内生成 HTTP 响应(可能是平台限制)，在此期间您必须执行 MongoDB 操作，并且必须对结果进行后处理，这需要 4 秒。在这种情况下，如果 MongoDB 操作花费的时间超过 6 秒，您将无法完成后处理以适应 10 秒的限制。因此，如果 MongoDB 操作未在 6 秒内完成，您不妨取消它。

解决这个问题的一个简单方法是派生一个超时为 6 秒的上下文:

ctx, cancel := context.WithTimeout(r.Context(), 6 * time.Second)
defer cancel()

// ctx automatically times out after 6 seconds
curs, err := c.Find(ctx, bson.M{"some": "filter"})

在这个例子中，ctx 将在 6 秒后超时，所以如果 c.Find() 操作没有完成，传递的 ctx 将发出可以取消的信号(在这种情况下将返回错误)。由于 ctx 派生自 r.Context()，如果请求的上下文被提前取消(例如 HTTP 客户端中止)，ctx 也会被取消。

现在假设您不是在编写 HTTP 处理程序，而是在编写一些其他代码(独立应用程序、后台工作程序等)，您可能没有上下文，但您有自己的标准，不想再等了超过 30 秒的查询。这是一个主要示例，您可以使用 30 秒超时从 context.Background() 派生上下文:

ctx, cancel := context.WithTimeout(context.Background(), 30 * time.Second)
defer cancel()

// ctx automatically times out after 30 seconds
curs, err := c.Find(ctx, bson.M{"some": "filter"})

如果查询在 30 秒内完成，一切都很好，您可以使用结果。否则，上下文将在 30 秒后取消，因此 c.Find() 也(可能)返回错误。