我有响应 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
}
最佳答案
首先，我强烈建议您查看 Context blog post熟悉上下文，除了阅读 context documentation本身。
解决您的具体问题:
如果用户取消了他们的任务，您如何取消数据库查询？
要使这项工作正常进行，您需要检查几件事情:
- 确保您的数据库驱动程序(如果您使用的是
database/sql)支持上下文取消。 - 确保您使用的是所有可用方法的上下文变体(例如
db.QueryContext而不是db.Query)。 - 确保从 HTTP 请求到数据库调用，您正在通过堆栈传递上下文(或上下文的衍生物)。
我需要将 ctx 传递给 F1() 吗？
根据上面的第 3 点，是的:您需要通过所有中间调用传递上下文，以“连接”数据库调用与请求上下文。
如何在 F1() 返回后立即返回？
问题中的代码会连续调用 F1，而不是同时调用取消/超时 select。
如果您想对数据库调用应用特定的截止日期，请使用 context.WithTimeout限制需要多长时间。否则，您无需执行任何特殊操作:只需使用您的上下文调用 F1，其余的将为您完成，不需要 select。
关于http - 取消 Http 处理程序请求，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/57650645/