-
Notifications
You must be signed in to change notification settings - Fork 229
/
latency.go
56 lines (48 loc) · 1.24 KB
/
latency.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
package trace
import (
"context"
"net/http/httptrace"
"time"
"k8s.io/klog/v2"
)
type LatencyInfo struct {
DNS time.Duration
Connect time.Duration
DNSStart time.Time
ConnectStart time.Time
}
func (r *LatencyInfo) dnsStart() {
r.DNSStart = time.Now()
}
func (r *LatencyInfo) connectStart(addr string) {
if r.ConnectStart.IsZero() {
r.ConnectStart = time.Now()
}
}
func (r *LatencyInfo) dnsDone() {
r.DNS = time.Since(r.DNSStart)
}
func (r *LatencyInfo) connectDone(addr string) {
r.Connect = time.Since(r.ConnectStart)
}
func WithLatencyInfoCapture(ctx context.Context) (context.Context, *LatencyInfo) {
trace := &LatencyInfo{}
return httptrace.WithClientTrace(ctx, &httptrace.ClientTrace{
DNSStart: func(info httptrace.DNSStartInfo) {
trace.dnsStart()
klog.V(5).Infof("DNSStart: %s\n", info.Host)
},
DNSDone: func(info httptrace.DNSDoneInfo) {
trace.dnsDone()
klog.V(5).Infof("DNSDone: %v\n", info)
},
ConnectStart: func(network, addr string) {
trace.connectStart(addr)
klog.V(5).Infof("ConnectStart: %s %s\n", network, addr)
},
ConnectDone: func(network, addr string, err error) {
trace.connectDone(addr)
klog.V(5).Infof("ConnectDone: %s,%s,%v\n", network, addr, err)
},
}), trace
}