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
package networkdial
import (
"crypto/tls"
"github.com/gorilla/websocket"
)
func NewWebSocketDialer(hosts Resolver, tlsConfig *tls.Config) (*websocket.Dialer, error) {
return NewWebSocketDialerWithOptions(DialOptions{Hosts: hosts, TLSConfig: tlsConfig})
}
func NewWebSocketDialerWithOptions(opts DialOptions) (*websocket.Dialer, error) {
d, err := NewDialerWithOptions(opts)
if err != nil {
return nil, err
}
return &websocket.Dialer{
NetDialContext: d.DialContext,
TLSClientConfig: d.TLSConfig(),
HandshakeTimeout: DefaultHandshakeTimeout,
}, nil
}
func NewNetworkWebSocketDialer(hosts Resolver, caBundlePEM []byte) (*websocket.Dialer, error) {
cfg, err := TLSConfigFromCA(caBundlePEM, "")
if err != nil {
return nil, err
}
return NewWebSocketDialerWithOptions(DialOptions{Hosts: hosts, TLSConfig: cfg})
}
