118 lines
4.3 KiB
Go
118 lines
4.3 KiB
Go
package config
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
type ProxyServerConfig struct {
|
|
HTTP HTTPConfig `yaml:"http"`
|
|
Metrics MetricsConfig `yaml:"metrics"`
|
|
Transport TransportConfig `yaml:"transport"`
|
|
Dial DialConfig `yaml:"dial"`
|
|
Sentry SentryConfig `yaml:"sentry"`
|
|
Cache CacheConfig `yaml:"cache"`
|
|
}
|
|
|
|
func NewDefaultProxyServerConfig() ProxyServerConfig {
|
|
return ProxyServerConfig{
|
|
HTTP: NewHTTPConfig("0.0.0.0", 8080),
|
|
Metrics: NewDefaultMetricsConfig(),
|
|
Transport: NewDefaultTransportConfig(),
|
|
Dial: NewDefaultDialConfig(),
|
|
Sentry: NewDefaultSentryConfig(),
|
|
Cache: NewDefaultCacheConfig(),
|
|
}
|
|
}
|
|
|
|
// See https://pkg.go.dev/net/http#Transport
|
|
type TransportConfig struct {
|
|
ForceAttemptHTTP2 InterpolatedBool `yaml:"forceAttemptHTTP2"`
|
|
MaxIdleConns InterpolatedInt `yaml:"maxIdleConns"`
|
|
MaxIdleConnsPerHost InterpolatedInt `yaml:"maxIdleConnsPerHost"`
|
|
MaxConnsPerHost InterpolatedInt `yaml:"maxConnsPerHost"`
|
|
IdleConnTimeout *InterpolatedDuration `yaml:"idleConnTimeout"`
|
|
TLSHandshakeTimeout *InterpolatedDuration `yaml:"tlsHandshakeTimeout"`
|
|
ExpectContinueTimeout *InterpolatedDuration `yaml:"expectContinueTimeout"`
|
|
DisableKeepAlives InterpolatedBool `yaml:"disableKeepAlives"`
|
|
DisableCompression InterpolatedBool `yaml:"disableCompression"`
|
|
ResponseHeaderTimeout *InterpolatedDuration `yaml:"responseHeaderTimeout"`
|
|
WriteBufferSize InterpolatedInt `yaml:"writeBufferSize"`
|
|
ReadBufferSize InterpolatedInt `yaml:"readBufferSize"`
|
|
MaxResponseHeaderBytes InterpolatedInt `yaml:"maxResponseHeaderBytes"`
|
|
InsecureSkipVerify InterpolatedBool `yaml:"insecureSkipVerify"`
|
|
}
|
|
|
|
func (c TransportConfig) AsTransport() *http.Transport {
|
|
httpTransport := http.DefaultTransport.(*http.Transport).Clone()
|
|
|
|
httpTransport.Proxy = http.ProxyFromEnvironment
|
|
httpTransport.ForceAttemptHTTP2 = bool(c.ForceAttemptHTTP2)
|
|
httpTransport.MaxIdleConns = int(c.MaxIdleConns)
|
|
httpTransport.MaxIdleConnsPerHost = int(c.MaxIdleConnsPerHost)
|
|
httpTransport.MaxConnsPerHost = int(c.MaxConnsPerHost)
|
|
httpTransport.IdleConnTimeout = time.Duration(*c.IdleConnTimeout)
|
|
httpTransport.TLSHandshakeTimeout = time.Duration(*c.TLSHandshakeTimeout)
|
|
httpTransport.ExpectContinueTimeout = time.Duration(*c.ExpectContinueTimeout)
|
|
httpTransport.DisableKeepAlives = bool(c.DisableKeepAlives)
|
|
httpTransport.DisableCompression = bool(c.DisableCompression)
|
|
httpTransport.ResponseHeaderTimeout = time.Duration(*c.ResponseHeaderTimeout)
|
|
httpTransport.WriteBufferSize = int(c.WriteBufferSize)
|
|
httpTransport.ReadBufferSize = int(c.ReadBufferSize)
|
|
httpTransport.MaxResponseHeaderBytes = int64(c.MaxResponseHeaderBytes)
|
|
|
|
if httpTransport.TLSClientConfig == nil {
|
|
httpTransport.TLSClientConfig = &tls.Config{}
|
|
}
|
|
httpTransport.TLSClientConfig.InsecureSkipVerify = bool(c.InsecureSkipVerify)
|
|
|
|
return httpTransport
|
|
}
|
|
|
|
func NewDefaultTransportConfig() TransportConfig {
|
|
return TransportConfig{
|
|
ForceAttemptHTTP2: true,
|
|
MaxIdleConns: 100,
|
|
MaxIdleConnsPerHost: 100,
|
|
MaxConnsPerHost: 100,
|
|
IdleConnTimeout: NewInterpolatedDuration(90 * time.Second),
|
|
TLSHandshakeTimeout: NewInterpolatedDuration(10 * time.Second),
|
|
ExpectContinueTimeout: NewInterpolatedDuration(1 * time.Second),
|
|
ResponseHeaderTimeout: NewInterpolatedDuration(10 * time.Second),
|
|
DisableCompression: false,
|
|
DisableKeepAlives: false,
|
|
ReadBufferSize: 4096,
|
|
WriteBufferSize: 4096,
|
|
MaxResponseHeaderBytes: 0,
|
|
InsecureSkipVerify: false,
|
|
}
|
|
}
|
|
|
|
// See https://pkg.go.dev/net#Dialer
|
|
type DialConfig struct {
|
|
Timeout *InterpolatedDuration `yaml:"timeout"`
|
|
KeepAlive *InterpolatedDuration `yaml:"keepAlive"`
|
|
FallbackDelay *InterpolatedDuration `yaml:"fallbackDelay"`
|
|
DualStack InterpolatedBool `yaml:"dualStack"`
|
|
}
|
|
|
|
func NewDefaultDialConfig() DialConfig {
|
|
return DialConfig{
|
|
Timeout: NewInterpolatedDuration(30 * time.Second),
|
|
KeepAlive: NewInterpolatedDuration(30 * time.Second),
|
|
FallbackDelay: NewInterpolatedDuration(300 * time.Millisecond),
|
|
DualStack: true,
|
|
}
|
|
}
|
|
|
|
type CacheConfig struct {
|
|
TTL InterpolatedDuration `yaml:"ttl"`
|
|
}
|
|
|
|
func NewDefaultCacheConfig() CacheConfig {
|
|
return CacheConfig{
|
|
TTL: *NewInterpolatedDuration(time.Second * 30),
|
|
}
|
|
}
|