Go context 对HTTP请求的影响

1-programmer2024-12-20 18:03

一、起因

Go语言中contex是控制协程行为的利器，既然是利器，那么用得好可以取得非常好的效果，用的不好，反手就可能伤了自己。为能使用好它，就需要了解其特性，以及在各种场景下的用法。最近遇到的一个问题，HTTP请求发给Server后，对应的context取消掉之后，请求被Server正常处理了，但客户端因为Context被取消，认为请求失败了。以此为契机研究下为什么会出现这种现象，即研究清楚，HTTP请求中的Context是被如何使用的。

二、HTTP标准库中的context

由于发送和接收本身是一个异常过程，猜测是在某个环节发送完数据后，等待Server端响应。

还是带着以下问题来看。

2.1 context如何包含在HTTP请求中的？

标准库中的http.Request类型，有WithContext的方法，这个方法为一个Request包含了一个Context实例。

2.2 HTTP请求中的Context是如何生效的？

在roundTrip中(net/http/transport.go)，如下逻辑。

复制代码

for {
	select {
	case <-ctx.Done():
		req.closeBody()
		return nil, ctx.Err()
	default:
	}
	// 发起连接，执行HTTP请求。
}

正是在等待Server端响应，同时也在判断Request的Context是否取消了。

三、小结

当Request中的Context有设置时，这个Context会影响到HTTP的结果，如果你不期望Context取消时影响到具体的HTTP请求，那么就不应该设置这个参数。默认不设置时，Context的取消不会终止掉具体的HTTP请求。

具体来说，如果你是在一个协程中等待一个HTTP请求的完成，那么会有如下现象：

未设置Context时，默认情况，协程对应的Context取消时，对应的协程不会终止，直到请求完成。
设置Context时，协程的Context取消时，HTTP请求终止，协程退出。所以就可能出现，请求被Server正常处理，但Client看到了失败。