Go设置http请求超时的方法实现
背景
最近接手了一个老项目进行维护，发现其中有个关于 http 请求的方法设置的 timeout 没有生效，很奇怪！
一开始查看代码并没有发现什么可疑点，后查看了源码，打断点调试才发现问题所在，这里简单记录复盘一下。
说明：本篇的源码的 go 版本是 1.20.2 。
问题
示例代码
package main
import (
"context"
"fmt"
"net/http"
"time"
)
func main() {
req, err := http.NewRequest(http.MethodGet, "https://www.baidu.com", nil)
if err != nil {
panic(err)
}
ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond)
defer cancel()
req.WithContext(ctx)
resp, err := http.DefaultClient.Do(req)
if err != nil {
panic(err)
}
//resp.Write(os.Stdout)
fmt.Println("end: ", resp.StatusCode)
}
程序正常跑完并输出了，但是预期的是 http.DefaultClient.Do(req) 这里会直接报错，难道请求 1ms 就结束了？？Why？？？
大家可以自己看下这段代码哪里有问题。
先说解决，其实就是 req.WithContext(ctx) 生成的是一个新的 http.Request 对象，上述的问题代码中并没有将其赋值给当前的 http.Request。大意了，没有闪。
req = req.WithContext(ctx)
WithContext 方法的源码如下（net/http/request.go 356）
func (r *Request) WithContext(ctx context.Context) *Request {
if ctx == nil {
panic("nil context")
}
r2 := new(Request)
*r2 = *r
r2.ctx = ctx
return r2
}
请求超时设置
翻了下源码，看了下超时设置的方式，http 设置超时主要有两种方式：
- http.Client
c := http.Client{
Timeout: time.Minute,
}
c.Do(req)
- http.Request 设置 context 超时
ctx, cancel := context.WithTimeout(context.Background(), time.Minute) defer cancel() req = req.WithContext(ctx)
在 Client 上指定 Timeout 会作用于通过该 Client 发起的所有请求，而 Request 设置 Context，仅针对这一次请求。使用的时候需要注意自己的场景。
设置 Tcp 连接阶段的超时可以这样：
client := http.Client{
Transport: &http.Transport{
Dial: (&net.Dialer{
Timeout: 2 * time.Second, // tcp 连接时设置的连接超时
Deadline: time.Now().Add(3 * time.Second), // 超时强制关闭
}).Dial,
TLSHandshakeTimeout: 2 * time.Second, //https 握手超时
},
Timeout: 5 * time.Second,
}
可以设置 Transport 中的 Dial。
总结
平常自己使用 http 发送请求设置超时，都是直接给 http.Client 对象设置 Timeout 属性，很少使用这种对单个 Request 设置超时的。
其实还是个熟练度问题，平常源码读的比较少。有空读读源码不仅可以在使用代码的时候更得心应手，也能够学习借鉴源码的代码设计实现。对自己平常经常需要使用的库，还是建议都过一遍源码，很不错的一个打发空闲时间的方式。
以上就是Go设置http请求超时的方法实现的详细内容，更多关于Go设置http请求超时的资料请关注脚本之家其它相关文章！
相关文章
这篇文章主要介绍了为什么 Go 语言 struct 要使用 tags,在本文中，我们将探讨为什么 Go 语言中需要使用 struct tags，以及 struct tags 的使用场景和优势，需要的朋友可以参考下2023-03-03
这篇文章主要带大家一起探索 Go 中原子操作的概念,了解为什么它们是重要的,以及如何有效地使用它们,文中的示例代码讲解详细,需要的可以了解下2023-11-11
anko是一个可以让 Go 项目支持脚本语言的小工具,换句话说,就是可以给 Go 项目加点“脚本魔法”,下面就跟随小编一起来学习一下他的具体使用吧2024-11-11
这篇文章主要为大家详细介绍了Golang实现断点续传、复制文件功能，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-07-07
当存储或传输 大整数（int64） 时,往往会出现精度丢失的问题,本文通过一个示例来详细分析原因,并给出解决方案,文中的示例代码讲解详细,感兴趣的小伙伴可以了解下2026-01-01
这篇文章主要给大家爱介绍了关于go语言学习之包和变量的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用go语言具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2018-06-06
这篇文章主要为大家介绍了Golang 流水线设计模式实践示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-12-12
关于golang中map使用的几点注意事项总结(强烈推荐!)
map是一种无序的基于key-value的数据结构,Go语言中的map是引用类型,必须初始化才能使用,下面这篇文章主要给大家介绍了关于golang中map使用的几点注意事项,需要的朋友可以参考下2023-01-01
这篇文章主要介绍了golang模板template自定义函数用法,结合实例形式分析了Go语言模板自定义函数的基本定义与使用方法,需要的朋友可以参考下2016-07-07
为了更加便捷地遍历这些数据类型，Go语言引入了for...range语句，本文将以数组遍历为起点，逐步介绍for...range语句在不同数据类型中的应用，希望对大家有所帮助2023-06-06
最新评论