func createPingFactory( config *Config, hostname string, port uint16, tls *transport.TLSConfig, request *http.Request, body []byte, validator RespCheck, ) func(*net.IPAddr) monitors.TaskRunner { fields := common.MapStr{ "scheme": request.URL.Scheme, "port": port, "url": request.URL.String(), } timeout := config.Timeout isTLS := request.URL.Scheme == "https" checkRedirect := makeCheckRedirect(config.MaxRedirects) return monitors.MakePingIPFactory(fields, func(ip *net.IPAddr) (common.MapStr, error) { addr := net.JoinHostPort(ip.String(), strconv.Itoa(int(port))) d := &dialchain.DialerChain{ Net: dialchain.ConstAddrDialer("tcp_connect_rtt", addr, timeout), } if isTLS { d.AddLayer(dialchain.TLSLayer("tls_handshake_rtt", tls, timeout)) } measures := common.MapStr{} dialer, err := d.BuildWithMeasures(measures) if err != nil { return nil, err } var httpStart, httpEnd time.Time client := &http.Client{ CheckRedirect: checkRedirect, Timeout: timeout, Transport: &SimpleTransport{ Dialer: dialer, OnStartWrite: func() { httpStart = time.Now() }, OnStartRead: func() { httpEnd = time.Now() }, }, } event, err := execPing(client, request, body, timeout, validator) if event == nil { event = measures } else { event.Update(measures) } if !httpEnd.IsZero() { event["http_rtt"] = look.RTT(httpEnd.Sub(httpStart)) } return event, err }) }
func buildDialerChain( scheme string, tls *transport.TLSConfig, config *Config, ) (*dialchain.DialerChain, error) { d := &dialchain.DialerChain{ Net: dialchain.TCPDialer("tcp_connect_rtt", config.Timeout), } if config.Socks5.URL != "" { d.AddLayer(dialchain.SOCKS5Layer("socks5_connect_rtt", &config.Socks5)) } if isTLSAddr(scheme) { d.AddLayer(dialchain.TLSLayer("tls_handshake_rtt", tls, config.Timeout)) } if err := d.TestBuild(); err != nil { return nil, err } return d, nil }