我要评论背景
最近接手了一个老项目进行维护，发现其中有个关于 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请求超时的资料请关注代码网其它相关文章！
发表评论