Golang contextWithTimeout请求超时控制示例

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

在 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 的例子：

pollinations

属于你的个性化媒体引擎

231

查看详情

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("响应长度: %d\n", 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请求超时控制示例的详细内容，更多请关注php中文网其它相关文章！