在官方文档中有提到 net/http
是协程安全的,应该复用。
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。
想了想解决的办法:
- 每个请求新建一个 Client?
- 把更改参数和请求一起加锁锁起来?
请问这种情况有靠谱的解决方法吗?
目前已经用 context.Context 解决了这个问题,实现了并发的情况下给单个请求指定 Proxy 代理、超时、重定向控制,以下是代码实现方式:
https://github.com/wnanbei/di...
https://github.com/wnanbei/di...
有此需求的朋友可以考虑直接使用我写的这个 http 客户端库 direwolf:
https://github.com/wnanbei/di...