-
Notifications
You must be signed in to change notification settings - Fork 3
/
transport.go
90 lines (73 loc) · 2.16 KB
/
transport.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package http
import (
"crypto/tls"
"net/http"
"time"
kilntypes "github.com/kilnfi/go-utils/common/types"
kilntls "github.com/kilnfi/go-utils/crypto/tls"
kilnnet "github.com/kilnfi/go-utils/net"
"golang.org/x/net/http2"
)
// TransportConfig options to configure communication between Traefik and the servers
type TransportConfig struct {
Dialer *kilnnet.DialerConfig
IdleConnTimeout *kilntypes.Duration
ResponseHeaderTimeout *kilntypes.Duration
ExpectContinueTimeout *kilntypes.Duration
MaxIdleConnsPerHost int
MaxConnsPerHost int
DisableKeepAlives bool
DisableCompression bool
EnableHTTP2 bool
TLS *kilntls.Config
}
func (cfg *TransportConfig) SetDefault() *TransportConfig {
if cfg.Dialer == nil {
cfg.Dialer = new(kilnnet.DialerConfig)
}
cfg.Dialer.SetDefault()
if cfg.IdleConnTimeout == nil {
cfg.IdleConnTimeout = &kilntypes.Duration{Duration: 90 * time.Second}
}
if cfg.ResponseHeaderTimeout == nil {
cfg.ResponseHeaderTimeout = &kilntypes.Duration{Duration: 0}
}
if cfg.ExpectContinueTimeout == nil {
cfg.ExpectContinueTimeout = &kilntypes.Duration{Duration: time.Second}
}
return cfg
}
// NewTransport creates a http.Transport
func NewTransport(cfg *TransportConfig) (*http.Transport, error) {
// Create dialer
dlr := kilnnet.NewDialer(cfg.Dialer)
transport := &http.Transport{
Proxy: http.ProxyFromEnvironment,
DialContext: dlr.DialContext,
DisableKeepAlives: cfg.DisableKeepAlives,
DisableCompression: cfg.DisableCompression,
MaxIdleConnsPerHost: cfg.MaxIdleConnsPerHost,
MaxConnsPerHost: cfg.MaxConnsPerHost,
IdleConnTimeout: cfg.IdleConnTimeout.Duration,
ResponseHeaderTimeout: cfg.ResponseHeaderTimeout.Duration,
ExpectContinueTimeout: cfg.ExpectContinueTimeout.Duration,
}
if cfg.TLS != nil {
tlsCfg, err := cfg.TLS.ToTLSConfig()
if err != nil {
return nil, err
}
tlsDlr := tls.Dialer{
NetDialer: dlr,
Config: tlsCfg,
}
transport.DialTLSContext = tlsDlr.DialContext
}
if cfg.EnableHTTP2 {
err := http2.ConfigureTransport(transport)
if err != nil {
return nil, err
}
}
return transport, nil
}