Go设置http请求超时的方法实现

 更新时间:2024年08月11日 10:11:05   作者:Grassto  
这篇文章主要介绍了Go设置http请求超时的方法实现,最近接手了一个老项目进行维护,发现其中有个关于 http 请求的方法设置的 timeout 没有生效,很奇怪,一开始查看代码并没有发现什么可疑点,后查看了源码,打断点调试才发现问题所在,这里简单记录复盘一下

背景

最近接手了一个老项目进行维护,发现其中有个关于 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请求超时的资料请关注脚本之家其它相关文章!

相关文章

  • 详解Golang如何优雅的终止一个服务

    详解Golang如何优雅的终止一个服务

    后端服务通常会需要创建子协程来进行相应的作业,但进程接受到终止信号或正常结束时,并没有判断或等待子协程执行结束,下面这篇文章主要给大家介绍了关于Golang如何优雅的终止一个服务的相关资料,需要的朋友可以参考下
    2022-03-03
  • 使用Gin框架返回JSON、XML和HTML数据

    使用Gin框架返回JSON、XML和HTML数据

    Gin是一个高性能的Go语言Web框架,它不仅提供了简洁的API,还支持快速的路由和中间件处理,在Web开发中,返回JSON、XML和HTML数据是非常常见的需求,本文将介绍如何使用Gin框架来返回这三种类型的数据,需要的朋友可以参考下
    2024-08-08
  • Golang库插件注册加载机制的问题

    Golang库插件注册加载机制的问题

    这篇文章主要介绍了Golang库插件注册加载机制,这里说的插件并不是指的golang原生的可以在buildmode中加载指定so文件的那种加载机制,需要的朋友可以参考下
    2022-03-03
  • 详解Go语言如何解析带注释的json

    详解Go语言如何解析带注释的json

    标准的json格式是不带注释,但是有时候为了方便理解json中各字段的含义,需要支持带注释的json,这篇文章主要介绍了Go语言解析带注释json的相关方法,希望对大家有所帮助
    2024-03-03
  • Golang实现web文件共享服务的示例代码

    Golang实现web文件共享服务的示例代码

    这篇文章主要介绍了Golang实现web文件共享服务的示例代码,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2018-10-10
  • Golang中map缩容的实现

    Golang中map缩容的实现

    本文主要介绍了Go语言中map的扩缩容机制,包括grow和hashGrow方法的处理,具有一定的参考价值,感兴趣的可以了解一下
    2025-03-03
  • Golang服务中context超时处理的方法详解

    Golang服务中context超时处理的方法详解

    在Go语言中,Context是一个非常重要的概念,它存在于一个完整的业务生命周期内,Context类型是一个接口类型,在实际应用中,我们可以使用Context包来传递请求的元数据,本文将给大家介绍Golang服务中context超时处理的方法和超时原因,需要的朋友可以参考下
    2023-05-05
  • Go语言中高效地处理集合的方法小结

    Go语言中高效地处理集合的方法小结

    在 Go 语言中,处理集合(如切片、映射等)时,可以通过多种方式提高效率,本文为大家整理了一些常见的方法,有需要的小伙伴可以了解一下
    2025-02-02
  • go xorm存库处理null值问题

    go xorm存库处理null值问题

    这篇文章主要介绍了go xorm存库处理null值问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2023-12-12
  • Golang中如何对MySQL进行操作详解

    Golang中如何对MySQL进行操作详解

    这篇文章主要给大家介绍了关于在Golang中如何对MySQL进行操作的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者使用Golang具有一定的参考学习价值,需要的朋友们下面来一起学习学习吧
    2019-03-03

最新评论