我目前正在将我们的代码从 global sign 包迁移到 go mongo-driver,不确定我应该在哪里使用 context.TODO()
和 context.Background()
,这真的很令人困惑,我知道它都返回非零空,所以我应该在 main 函数和 init()
函数中使用 context.Background()
?并在其他地方使用 context.TODO()
?有人可以帮忙吗?
尝试检查我应该使用哪个参数 context.TODO()
或 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.
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()
也(可能)返回错误。
关于go - context.TODO() 或 context.Background(),我更喜欢哪一个?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/74239074/