-
Notifications
You must be signed in to change notification settings - Fork 2
/
dialer.go
63 lines (54 loc) · 1.63 KB
/
dialer.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
package net
import (
"net"
"time"
"github.com/prometheus/client_golang/prometheus"
"github.com/go-gophers/gophers/config"
"github.com/go-gophers/gophers/utils/log"
)
// shared Prometheus metrics for all Dial calls
var (
mDials = prometheus.NewCounterVec(prometheus.CounterOpts{
Namespace: "gophers",
Subsystem: "net",
Name: "dials",
Help: "Dial calls",
}, []string{"network"})
mDialResults = prometheus.NewCounterVec(prometheus.CounterOpts{
Namespace: "gophers",
Subsystem: "net",
Name: "dial_results",
Help: "Dial results",
}, []string{"network", "error"})
)
func init() {
prometheus.MustRegister(mDials, mDialResults)
}
// Dial wraps net.Dial with DNS lookup cache and Prometheus metrics.
func Dial(network, addr string) (net.Conn, error) {
if network != "tcp" {
panic("only tcp is supported for now")
}
host, port, err := net.SplitHostPort(addr)
if err != nil {
log.Printf("net.SplitHostPort(%q): %s", addr, err)
return nil, err
}
host, err = lookupHost(host)
if err != nil {
log.Printf("lookupHost(%q): %s", host, err)
return nil, err
}
addr = net.JoinHostPort(host, port)
mDials.WithLabelValues(network).Inc()
log.Debugf("gophers/net.Dial(%q, %q)", network, addr)
start := time.Now()
conn, err := net.DialTimeout(network, addr, config.Default.DialTimeout)
mDialResults.WithLabelValues(network, errorLabelValue(err)).Inc()
if err == nil {
log.Debugf("gophers/net.Dial(%q, %q): connection established (in %s)", network, addr, time.Now().Sub(start))
} else {
log.Printf("gophers/net.Dial(%q, %q): %s (in %s)", network, addr, err, time.Now().Sub(start))
}
return &Conn{conn}, err
}