取消 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文档本身之外，我强烈建议您查看Context 博客文章以熟悉context 。

要解决您的具体问题：

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

为了使这项工作正常进行，您需要检查以下几点：

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

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

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

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

您在问题中的代码将F1与您的取消/超时select一起调用，而不是同时调用。

如果您想对数据库调用应用特定的截止日期，请使用context.WithTimeout来限制它需要多长时间。 否则，您不需要做任何特别的事情：只需使用您的上下文调用F1 ，其余的事情都会为您完成，不需要select 。