Golang contextWithTimeout请求超时控制示例

context.WithTimeout用于设置操作超时,防止程序长时间阻塞;2. 示例中通过context.WithTimeout控制模拟耗时操作的执行时间,超时后自动取消。

golang contextwithtimeout请求超时控制示例

在 Go 语言中,context.WithTimeout 是控制请求执行时间的常用方式,尤其适用于网络请求、数据库查询等可能长时间阻塞的操作。当操作超过指定时间仍未完成时,会自动触发超时,防止程序卡死。

基本用法:使用 context.WithTimeout 设置超时

以下是一个简单的示例,展示如何使用 context.WithTimeout 控制一个模拟耗时操作的执行时间:

package main

立即学习“go语言免费学习笔记(深入)”;

import (
  “context”
  “fmt”
  “time”
)

func slowOperation(ctx context.Context) {
  select {
  case     fmt.Println(“操作成功完成”)
  case     fmt.Println(“操作被取消:”, ctx.Err())
  }
}

func main() {
  // 设置 2 秒超时
  ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
  defer cancel()

  fmt.Println(“开始执行…”)
  slowOperation(ctx)
  fmt.Println(“主函数结束”)
}

输出结果:
开始执行…
操作被取消: context deadline exceeded
主函数结束

说明:虽然 slowOperation 需要 3 秒完成,但上下文只给了 2 秒,因此触发超时,ctx.Done() 被触发,返回错误 context deadline exceeded

实际场景:HTTP 请求超时控制

在网络请求中使用超时控制更为常见。以下是使用 http.Get 并结合 context.WithTimeout 的例子:

package main

import (
  “context”
  “fmt”
  “io”
  “net/http”
  “time”
)

func fetch(ctx context.Context, url string) {
  req, err := http.NewRequestWithContext(ctx, “GET”, url, nil)
  if err != nil {
    fmt.Println(“创建请求失败:”, err)
    return
  }

  resp, err := http.DefaultClient.Do(req)
  if err != nil {
    fmt.Println(“请求失败:”, err)
    return
  }
  defer resp.Body.Close()

  body, _ := io.ReadAll(resp.Body)
  fmt.Printf(“响应长度: %dn”, len(body))
}

func main() {
  ctx, cancel := context.WithTimeout(context.Background(), 3 * time.Second)
  defer cancel()

  fmt.Println(“开始请求…”)
  fetch(ctx, “https://httpbin.org/delay/5”) // 延迟 5 秒返回
  fmt.Println(“请求结束”)
}

输出:
开始请求…
请求失败: Get “https://httpbin.org/delay/5”: context deadline exceeded
请求结束

说明:目标 URL 会延迟 5 秒返回,但我们设置了 3 秒超时,因此请求在完成前被取消。

关键点总结

context.WithTimeout 返回一个带有自动取消功能的上下文和一个 cancel 函数。即使未显式调用 cancel,在超时后也会自动释放资源,但仍建议始终调用 defer cancel() 以确保及时清理。

超时时间从创建上下文时开始计算所有传递该 context 的函数都能感知到超时信号HTTP 请求需通过 http.NewRequestWithContext 绑定 context子 goroutine 中使用相同 context 可实现统一超时控制

基本上就这些。合理使用 context.WithTimeout 能有效提升服务稳定性。

以上就是Golang contextWithTimeout请求超时控制示例的详细内容,更多请关注创想鸟其它相关文章!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1409363.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2025年12月16日 02:20:25
下一篇 2025年12月16日 02:20:42

相关推荐

发表回复

登录后才能评论
关注微信