go.etcd.io/etcd@v3.3.27+incompatible/pkg/transport/transport.go (about) 1 // Copyright 2016 The etcd Authors 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 package transport 16 17 import ( 18 "net" 19 "net/http" 20 "strings" 21 "time" 22 ) 23 24 type unixTransport struct{ *http.Transport } 25 26 func NewTransport(info TLSInfo, dialtimeoutd time.Duration) (*http.Transport, error) { 27 cfg, err := info.ClientConfig() 28 if err != nil { 29 return nil, err 30 } 31 32 t := &http.Transport{ 33 Proxy: http.ProxyFromEnvironment, 34 Dial: (&net.Dialer{ 35 Timeout: dialtimeoutd, 36 // value taken from http.DefaultTransport 37 KeepAlive: 30 * time.Second, 38 }).Dial, 39 // value taken from http.DefaultTransport 40 TLSHandshakeTimeout: 10 * time.Second, 41 TLSClientConfig: cfg, 42 } 43 44 dialer := (&net.Dialer{ 45 Timeout: dialtimeoutd, 46 KeepAlive: 30 * time.Second, 47 }) 48 dial := func(net, addr string) (net.Conn, error) { 49 return dialer.Dial("unix", addr) 50 } 51 52 tu := &http.Transport{ 53 Proxy: http.ProxyFromEnvironment, 54 Dial: dial, 55 TLSHandshakeTimeout: 10 * time.Second, 56 TLSClientConfig: cfg, 57 } 58 ut := &unixTransport{tu} 59 60 t.RegisterProtocol("unix", ut) 61 t.RegisterProtocol("unixs", ut) 62 63 return t, nil 64 } 65 66 func (urt *unixTransport) RoundTrip(req *http.Request) (*http.Response, error) { 67 url := *req.URL 68 req.URL = &url 69 req.URL.Scheme = strings.Replace(req.URL.Scheme, "unix", "http", 1) 70 return urt.Transport.RoundTrip(req) 71 }