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 的例子:
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
微信扫一扫
支付宝扫一扫