迁移至Go Mongo-Driver：context.TODO()与context.Background()该如何选择？

先明确二者的核心定位：

• context.Background()：是所有上下文的根，用于明确没有上层上下文的场景，是一个“正式”的空上下文。
• context.TODO()：是临时占位符，用于不确定该用什么上下文，或者后续计划替换为带取消/超时/元数据的上下文的场景，相当于代码里的“待办标记”。

结合Mongo驱动的迁移场景，具体建议如下：

1. 程序入口（main/init 函数）：用 context.Background() #

这是程序的起点，没有任何上层传入的上下文，初始化Mongo客户端、全局连接这类操作，完全适合用Background()：

func main() {
    client, err := mongo.Connect(context.Background(), options.Client().ApplyURI("mongodb://localhost:27017"))
    if err != nil {
        log.Fatal(err)
    }
    defer client.Disconnect(context.Background())
}

2. 业务逻辑/请求处理函数：优先传上层上下文，不确定时用 TODO() #

• 如果是HTTP、gRPC这类请求驱动的场景，直接用请求自带的上下文（比如http.Request.Context()）传递给Mongo操作，这样能实现请求级别的取消、超时联动，这是最佳实践：

  func handleUserQuery(w http.ResponseWriter, r *http.Request) {
      coll := client.Database("test").Collection("users")
      var result User
      err := coll.FindOne(r.Context(), bson.M{"name": "Alice"}).Decode(&result)
      // 处理逻辑...
  }
  

• 如果当前函数没有上层传入的上下文，但后续可能会添加超时、取消逻辑（比如后续要改成支持用户取消的后台任务），先用context.TODO()占位，提醒自己后续替换；如果是确定不需要扩展的底层工具函数，也可以用Background()，但TODO()更适合标记待优化点。

3. 后台任务/定时任务：用 context.Background() #

如果是独立运行的后台任务（比如定时同步Mongo数据），没有上层请求上下文，直接用Background()作为根上下文，若需要超时或取消，可以基于它派生：

func syncMongoData() {
    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Minute)
    defer cancel()
    coll := client.Database("test").Collection("data")
    // 执行同步操作...
    _, err := coll.InsertMany(ctx, docs)
}

关键误区纠正 #

不是“除了main/init都用TODO()”，而是：

• 当你明确当前场景没有上层上下文，且不需要后续替换时，用Background()
• 当你不确定该用什么上下文，或者计划后续添加更合适的上下文（比如传入上层ctx、添加超时）时，用TODO()作为临时占位

内容的提问来源于stack exchange，提问作者Gopher_k