首页 > 后端开发 > Golang > 正文

Golang使用context控制请求超时示例

P粉602998670
发布: 2025-09-17 14:50:01
原创
362人浏览过
使用Context控制请求超时可避免资源浪费和系统阻塞。通过context.WithTimeout创建带超时的Context,传递给HTTP请求,若超时则自动取消,释放资源并提升系统稳定性。

golang使用context控制请求超时示例

使用context控制请求超时,核心在于利用

context.WithTimeout
登录后复制
context.WithDeadline
登录后复制
函数创建带有超时或截止时间的context,并将其传递给需要进行超时控制的操作。这样,如果操作在指定时间内未完成,context会自动取消,从而终止操作。

package main

import (
    "context"
    "fmt"
    "net/http"
    "time"
)

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

    req, err := http.NewRequestWithContext(ctx, "GET", "https://www.example.com", nil)
    if err != nil {
        fmt.Println("Error creating request:", err)
        return
    }

    client := &http.Client{}
    resp, err := client.Do(req)

    if err != nil {
        fmt.Println("Error making request:", err)
        return
    }
    defer resp.Body.Close()

    fmt.Println("Response Status:", resp.Status)
}
登录后复制

为什么需要使用Context控制请求超时?

资源浪费!如果一个请求因为网络问题或者服务器压力过大而长时间得不到响应,那么程序会一直等待,消耗系统资源。使用超时控制可以避免这种情况,及时释放资源。另外,超时机制可以提高系统的稳定性和用户体验,避免因为个别请求的阻塞而影响整体性能。

如何在不同的Golang HTTP客户端中使用Context超时控制?

芦笋演示
芦笋演示

一键出成片的录屏演示软件,专为制作产品演示、教学课程和使用教程而设计。

芦笋演示 34
查看详情 芦笋演示

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

除了标准库

net/http
登录后复制
,你可能还会用到一些第三方HTTP客户端库。 它们的使用方法大同小异,关键在于将context传递给发起请求的函数。

  • net/http
    登录后复制
    (标准库):
    像上面的示例一样,使用
    http.NewRequestWithContext
    登录后复制
    创建请求,然后使用
    client.Do
    登录后复制
    发起请求。
  • go-resty
    登录后复制
    :
    go-resty
    登录后复制
    库提供了更友好的API,可以使用
    R().SetContext(ctx).Get(url)
    登录后复制
    方法来设置context。
  • fasthttp
    登录后复制
    :
    fasthttp
    登录后复制
    需要手动检查context的
    Done()
    登录后复制
    channel 来判断是否超时,稍微复杂一些。
// go-resty 示例
package main

import (
    "context"
    "fmt"
    "time"

    "github.com/go-resty/resty/v2"
)

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

    client := resty.New()

    resp, err := client.R().
        SetContext(ctx).
        Get("https://www.example.com")

    if err != nil {
        fmt.Println("Error making request:", err)
        return
    }

    fmt.Println("Response Status:", resp.Status())
}
登录后复制

Context超时控制的常见错误和陷阱有哪些?

  • 忘记调用
    cancel()
    登录后复制
    :
    使用
    context.WithTimeout
    登录后复制
    context.WithDeadline
    登录后复制
    创建的context,必须调用
    cancel()
    登录后复制
    函数来释放资源,否则可能会造成资源泄漏。
  • 超时时间设置过短: 如果超时时间设置得太短,可能会导致正常的请求也被中断。需要根据实际情况合理设置超时时间。
  • 没有正确处理超时错误: 当context超时时,
    client.Do
    登录后复制
    会返回一个错误。需要检查这个错误,并进行相应的处理,例如重试或者返回错误给用户。 常见的错误类型是
    context.DeadlineExceeded
    登录后复制
  • goroutine中使用context,但未正确传递或取消: 如果在goroutine中使用context,确保将context正确传递给goroutine,并且在不再需要时取消它。
// 示例:正确处理超时错误
package main

import (
    "context"
    "fmt"
    "net/http"
    "time"
)

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

    req, err := http.NewRequestWithContext(ctx, "GET", "https://www.example.com", nil)
    if err != nil {
        fmt.Println("Error creating request:", err)
        return
    }

    client := &http.Client{}
    resp, err := client.Do(req)

    if err != nil {
        if err == context.DeadlineExceeded {
            fmt.Println("Request timed out!")
        } else {
            fmt.Println("Error making request:", err)
        }
        return
    }
    defer resp.Body.Close()

    fmt.Println("Response Status:", resp.Status)
}
登录后复制

以上就是Golang使用context控制请求超时示例的详细内容,更多请关注php中文网其它相关文章!

最佳 Windows 性能的顶级免费优化软件
最佳 Windows 性能的顶级免费优化软件

每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。

下载
来源:php中文网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新 English
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号