我是否应该在Go中将context.Context传递到底层数据库方法？

在使用 Go 时，处理上下文 (context.Context) 对于管理资源和处理超时至关重要。在将数据库方法传递到路由器控制器时，最佳做法是在需要时传递上下文，以支持信息传递和自动取消/超时。但是，在服务启动期间等情况下，没有上下文信息时，使用 context.Background() 就足够了。传递上下文具有良好的性能，并且在需要控制并发访问时需要谨慎处理。

我在这里使用半代码只是为了表明我对代码中发生的事情的意图，而不是使问题中的事情复杂化。

我有一个 main.go 文件，它调用连接到 mongodb 数据库的方法：

mstore := store.newmongostore()

在 newmongostore 中，我有 client.connect 用于连接到数据库的上下文：

ctx, cancel := context.withtimeout(context.background(), 10*time.second)
defer cancel()

现在在 main.go 中，我通过这种方式将存储传递到我的路由器控制器文件：

routes.genericroute(router, mstore)

在 genericroute 中，我获取 mstore 并将其传递给函数处理程序：

func genericroute(router *gin.engine, mstore store.store) {
    router.post("/users", controllers.createuser(mstore))
}

现在，在 createuser 中，我再次创建一个上下文，如下所示，将文档插入 mongodb：

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

insertedId, err := repo.CreateUser(ctx, newUser{"John", "Doe"})

这里我将上下文传递给 createuser 以插入新文档。

这种编码的最佳实践是什么？从性能角度来看，哪一个更好？

正确答案

我对自己的问题找到了一个有趣的答案，因此如果未来的用户有同样的问题，我更愿意将其放在这里。

如果我将连接到 mongo 的相同上下文传递给 usercontroller 并将其进一步传递给 createuser 函数：

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

注意：不是在 newmongostore 函数中取消上下文，而是在 defer cancel() 函数中取消它。

10秒后，如果你调用post /users，你将得到context 最后期限超出，所以基本上你不能使用这个上下文做其他事情，你必须在每个createuser调用上创建新的上下文。

所以我写的很好。在我的示例中，我等待 10 秒连接到 mongo，并等待 1 秒连接插入操作上下文。

根据我的经验，context 有两个主要用例：

1. 传递信息。对于您的问题，您可能希望为每个请求生成一个 request_id 并将其传递到代码的最低部分，并记录此 request_id 以在整个代码库中进行错误跟踪。
   1. 此功能并不总是有用，例如您想要初始化 mongodb 连接，但它是在服务启动期间完成的。此时没有有意义的上下文，带有超时的 context.background 应该足够了。
   2. 请谨慎修改从 context 检索的值，如果您四处传递 context，这可能会导致并发访问。
2. 自动取消和超时。这两个功能并非凭空而来，您需要调整代码以处理来自 context 的这些信息。但大多数带有 context 参数的第三方库和标准库都可以很好地处理这两个功能（例如数据库库、http 调用库）。使用此功能，一旦 context 失效，您可以自动回收资源。
   1. 有时你会想要停止这种级联行为，例如在后台 goroutine 中写入日志，那么你需要创建一个新的 context.background() 以避免一旦上游上下文取消，这些写入就会被取消。 context.background() 还会清除信息上下文，因此有时您需要从上游上下文中提取上下文信息，并手动将它们附加到这个新上下文中。

对所有函数强制使用 context 参数有点过分了（没有必要将 context 添加到简单的 greatestcommondivisor 函数），但在任何需要的地方添加 context 参数永远不会有什么坏处。 context 具有足够好的性能，对于您的用例（http 服务器和数据库写入），它不应该对您的服务造成可见的开销。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《我是否应该在Go中将context.Context传递到底层数据库方法？》文章吧，也可关注golang学习网公众号了解相关技术文章。