mirror of
https://github.com/yusing/godoxy.git
synced 2026-02-18 08:27:43 +01:00
6528fb0a8d
Add context parameter to TCP/UDP stream health checks and client constructors for proper cancellation and deadline propagation. Switch from encoding/json to sonic for faster JSON unmarshaling. Standardize HTTP client timeouts to 5 seconds across agent pool and health check.
56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package agentpool
|
|
|
|
import (
|
|
"net"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/valyala/fasthttp"
|
|
"github.com/yusing/godoxy/agent/pkg/agent"
|
|
)
|
|
|
|
type Agent struct {
|
|
*agent.AgentConfig
|
|
|
|
httpClient *http.Client
|
|
fasthttpHcClient *fasthttp.Client
|
|
}
|
|
|
|
func newAgent(cfg *agent.AgentConfig) *Agent {
|
|
transport := cfg.Transport()
|
|
transport.MaxIdleConns = 100
|
|
transport.MaxIdleConnsPerHost = 100
|
|
transport.ReadBufferSize = 16384
|
|
transport.WriteBufferSize = 16384
|
|
|
|
return &Agent{
|
|
AgentConfig: cfg,
|
|
httpClient: &http.Client{
|
|
Transport: transport,
|
|
Timeout: 5 * time.Second,
|
|
},
|
|
fasthttpHcClient: &fasthttp.Client{
|
|
DialTimeout: func(addr string, timeout time.Duration) (net.Conn, error) {
|
|
if addr != agent.AgentHost+":443" {
|
|
return nil, &net.AddrError{Err: "invalid address", Addr: addr}
|
|
}
|
|
return net.DialTimeout("tcp", cfg.Addr, timeout)
|
|
},
|
|
TLSConfig: cfg.TLSConfig(),
|
|
ReadTimeout: 5 * time.Second,
|
|
WriteTimeout: 3 * time.Second,
|
|
DisableHeaderNamesNormalizing: true,
|
|
DisablePathNormalizing: true,
|
|
NoDefaultUserAgentHeader: true,
|
|
ReadBufferSize: 1024,
|
|
WriteBufferSize: 1024,
|
|
},
|
|
}
|
|
}
|
|
|
|
func (agent *Agent) HTTPClient() *http.Client {
|
|
return &http.Client{
|
|
Transport: agent.Transport(),
|
|
}
|
|
}
|