Go net/http Client 的某些参数不是并发安全的?

2019-11-25 16:16:13 +08:00
 wnanbei

在官方文档中有提到 net/http client 是协程安全的,应该复用。

The Client's Transport typically has internal state (cached TCP connections), so Clients should be reused instead of created as needed. Clients are safe for concurrent use by multiple goroutines.

但使用 Client 发起请求时,有一部分请求的设置是以函数或字段的方式放在 Client 的参数中的。例如 Proxy 代理、重定向检查、超时设置。要设置的话必须像以下这般设置:

func RedirectFunc(req *http.Request, via []*http.Request) error {
	if len(via) > 5 {
		err := &RedirectError{r}
		return WrapErr(err, "RedirectError")
	}
	return nil
}

client.CheckRedirect = RedirectFunc  // 设置重定向

这样的话就会造成一个问题,在并发的过程中如果要更改重定向次数的话,就会有并发安全问题,设置 Proxy 代理和超时时间也有这个问题。

比如在这样一个假设情况中,我现在有 10000 个请求需要并发,每个请求需要设置不同的特定 Proxy 代理。那么这时候使用全局的 Client,在每个协程中更改 client.CheckRedirect 函数,然后发起请求,显然会有并发问题,发起请求时使用的并不一定是指定的那个 Proxy。

想了想解决的办法:

  1. 每个请求新建一个 Client ?
  2. 把更改参数和请求一起加锁锁起来?

请问这种情况有靠谱的解决方法吗?

6674 次点击
所在节点    Go 编程语言
27 条回复
tsl0922
2019-11-25 16:22:40 +08:00
大兄弟,人家说的是让你复用 Transport,划重点 。
wnanbei
2019-11-25 16:28:27 +08:00
@tsl0922 说的是 `so Clients should be reused`,而且设置代理在 transport 里,一样的也是这个问题。
tsl0922
2019-11-25 16:33:13 +08:00
抱歉,没仔细看题。我仔细看了文档,感觉这里的复用应该不是只并发的复用,Client 应该是有状态的。GitHub 上 golang 的官方 issue 里也有人提到关于复用的另一个问题:

https://github.com/golang/go/issues/26095
janxin
2019-11-25 16:36:05 +08:00
你对并发安全理解有问题吧,这个明显你修改的话是在请求过程中还是使用你指定的 Func,但是你每个都修改,client 只会使用你最后指定的参数,这样看上去是不是没问题了?

如果是单纯 proxy,你有个简单方案是按照 proxy 生成对应的 client,然后使用就行了
wnanbei
2019-11-25 16:43:13 +08:00
@tsl0922 嗯嗯,我也知道官方文档的复用大部分意思指的是复用连接。

但代理、超时、重定向这些设置,只有 Client 和 Transport 这种级别可以设置,在 Request 这个级别无法指定到特定的请求上。现在又需要复用 Client,但在并发的时候更改 Client 里的这些设置又不是并发安全的,就很头疼。
wnanbei
2019-11-25 16:47:47 +08:00
@janxin 目前就是需要把 超时、代理、重定向 这些设置到单个 Request 上,但 net/http 不提供,就只能设置在 Client 上。
并发的时候改 Client 就会有问题,因为 Client 需要复用,是全局的。
wnanbei
2019-11-25 16:49:37 +08:00
@janxin 在当前协程改了 func 后,发起请求,不一定用的就是刚刚改的这个 func。因为在这个过程中这个 func 很可能又被其他协程改掉了
tsl0922
2019-11-25 17:28:35 +08:00
go 文档里说的复用应该是指复用 tcp 连接( keep-alive ),你这种换代理的方式很显然要重新连接。你想要的只是复用 Client 这一个结构体,还是老老实实不同的 proxy 创建不同的 Client 对象吧,应该不会多出多少开销。
wnanbei
2019-11-25 17:54:47 +08:00
@tsl0922 有时候其实换代理的量很大,而且还有超时,重定向等等。哎,脑阔疼
chennqqi
2019-11-25 17:59:21 +08:00
Client 基本可以复用,request 不能复用,是这个意思?
gaara
2019-11-25 18:15:45 +08:00
session = client
session.Timeout=
session.Redirect=
chennqqi
2019-11-25 18:25:28 +08:00
@wnanbei 我看了 github issue 原文是如果请求的 body 被正确关闭,并被读取就可以重用吧
darrh00
2019-11-25 18:34:15 +08:00
CheckRedirect 这个属性为什么需要修改呢?

这个函数的签名提供的两个 req *http.Request, via []*http.Request 参数还不够做业务判断吗?

我的意思是,client 创建完成了,在生命周期内保持 CheckRedirect 不变化, 所有需要处理重定向而需要作区分对待的统一在 CheckRedirect 函数里实现不行吗?
zhujinliang
2019-11-25 18:44:29 +08:00
同意 @darrh00,楼主用法错了,client 设置完以后就不要修改了
wnanbei
2019-11-26 10:10:19 +08:00
@chennqqi 不是,是 Request 里根本没有设置重定向,超时,代理这些的地方,只能去 client 里设置。并发的时候要改这些设置就只能改 client,并发的时候改 client 就有问题。
wnanbei
2019-11-26 10:11:17 +08:00
@darrh00 因为重定向次数是写死在这个 CheckRedirect 里的,有一些请求需要限制的重定向次数可能是不同的,所以需要改。
wnanbei
2019-11-26 10:12:09 +08:00
@zhujinliang 是 Request 里根本没有设置重定向,超时,代理这些的地方,只能去 client 里设置。并发的时候要改这些设置就只能改 client。不然就只有建新的 Client。
icexin
2019-11-26 11:12:55 +08:00
@wnanbei 可以试试使用 NewRequestWithContext 在 Request 里面附加上下文信息,在 CheckRedirect 或者 Proxy 函数里面从 Request.Context 拿到 Context 来切换对应的策略。
wnanbei
2019-11-26 13:45:00 +08:00
@icexin 好像可行,我去研究一下,非常感谢!!!
reus
2019-11-26 15:19:00 +08:00
一般这些都只设一次的

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/622953

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX