-
Notifications
You must be signed in to change notification settings - Fork 5
/
connfact.go
67 lines (62 loc) · 1.44 KB
/
connfact.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
package main
import (
"context"
"crypto/tls"
"golang.org/x/sync/semaphore"
"net"
"time"
)
var ZEROTIME time.Time
var EPOCH time.Time
type ConnFactory struct {
addr string
timeout time.Duration
tlsEnabled bool
tlsConfig *tls.Config
sem *semaphore.Weighted
}
func NewConnFactory(address string, timeout time.Duration, tlsEnabled bool,
certfile, keyfile string, cafile string, hostname_check bool,
tls_servername string, dialers uint, resolve_once bool) (*ConnFactory, error) {
var tlsConfig *tls.Config
host, _, err := net.SplitHostPort(address)
if err != nil {
return nil, err
}
if tlsEnabled {
cfg_servername := host
if tls_servername != "" {
cfg_servername = tls_servername
}
tlsConfig, err = makeClientTLSConfig(cfg_servername, certfile, keyfile, cafile, hostname_check)
if err != nil {
return nil, err
}
}
if resolve_once {
address, err = ProbeResolveTCP(address, timeout)
if err != nil {
return nil, err
}
}
return &ConnFactory{
addr: address,
timeout: timeout,
tlsEnabled: tlsEnabled,
tlsConfig: tlsConfig,
sem: semaphore.NewWeighted(int64(dialers)),
}, nil
}
func (f *ConnFactory) Dial(ctx context.Context) (net.Conn, error) {
var (
conn net.Conn
err error
)
var dialer net.Dialer
myctx, _ := context.WithTimeout(ctx, f.timeout)
conn, err = dialer.DialContext(myctx, "tcp", f.addr)
if f.tlsEnabled {
conn = tls.Client(conn, f.tlsConfig)
}
return conn, err
}