http - 取消 Http 处理程序请求

Question

我有响应 https 请求的处理程序。在处理程序中，我调用了一个函数 F1()，它执行一些应用程序逻辑并连接到 mysql 数据库并执行查询。我想知道如果客户端取消请求，我如何使用 golang 上下文包来取消 Db 查询。我需要将 ctx 传递给 F1() 吗？此外，即使 F1() 在不到 4 秒内返回，我现在拥有的代码也将花费 4 秒。如何在 F1() 返回后立即返回？

func handler(w http.ResponseWriter, r *http.Request) {

  ctx:= r.context()

  F1()


 select {

     case <-ctx.Done():

     case <- time.After( 4*time.Second):

   }


   w.WriteHeader(http.statusOk)
  return 

 }

Answer 1

首先，我强烈建议您查看 Context blog post熟悉上下文，除了阅读 context documentation本身。

解决您的具体问题:

如果用户取消了他们的任务，您如何取消数据库查询？

要使这项工作正常进行，您需要检查几件事情:

1. 确保您的数据库驱动程序(如果您使用的是 database/sql)支持上下文取消。
2. 确保您使用的是所有可用方法的上下文变体(例如 db.QueryContext 而不是 db.Query)。
3. 确保从 HTTP 请求到数据库调用，您正在通过堆栈传递上下文(或上下文的衍生物)。

我需要将 ctx 传递给 F1() 吗？

根据上面的第 3 点，是的:您需要通过所有中间调用传递上下文，以“连接”数据库调用与请求上下文。

如何在 F1() 返回后立即返回？

问题中的代码会连续调用 F1，而不是同时调用取消/超时 select。

如果您想对数据库调用应用特定的截止日期，请使用 context.WithTimeout限制需要多长时间。否则，您无需执行任何特殊操作:只需使用您的上下文调用 F1，其余的将为您完成，不需要 select。