hpz937/zurg
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Set max connections to 32

Browse Source
This commit is contained in:
Ben Sarmiento
2024-01-23 02:46:11 +01:00
parent 1640ef4c7e
commit eca9a4f49a
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/http/client.go
View File

@@ -124,6 +124,11 @@ func NewHTTPClient(token string, maxRetries int, timeoutSecs int, ensureIPv6Host
} }
} }
maxConnections := cfg.GetNumOfWorkers()
if maxConnections > 32 {
maxConnections = 32
}
if cfg.ShouldForceIPv6() { if cfg.ShouldForceIPv6() {
ipv6List, err := hosts.FetchHosts(hosts.IPV6) ipv6List, err := hosts.FetchHosts(hosts.IPV6)
if err != nil { if err != nil {
@@ -135,6 +140,8 @@ func NewHTTPClient(token string, maxRetries int, timeoutSecs int, ensureIPv6Host
} }
client.client.Transport = &http.Transport{ client.client.Transport = &http.Transport{
MaxIdleConns: maxConnections,
MaxConnsPerHost: maxConnections,
DialContext: func(ctx context.Context, network, address string) (net.Conn, error) { DialContext: func(ctx context.Context, network, address string) (net.Conn, error) {
if ipv6Address, ok := client.ipv6.Get(address); ok { if ipv6Address, ok := client.ipv6.Get(address); ok {
return dialer.Dial(network, ipv6Address) return dialer.Dial(network, ipv6Address)
@@ -159,6 +166,8 @@ func NewHTTPClient(token string, maxRetries int, timeoutSecs int, ensureIPv6Host
} }
} else { } else {
client.client.Transport = &http.Transport{ client.client.Transport = &http.Transport{
MaxIdleConns: maxConnections,
MaxConnsPerHost: maxConnections,
DialContext: func(ctx context.Context, network, address string) (net.Conn, error) { DialContext: func(ctx context.Context, network, address string) (net.Conn, error) {
return dialer.Dial(network, address) return dialer.Dial(network, address)
}, },