package tcping import ( "bytes" "fmt" "io" "io/ioutil" "net/http" "net/http/httptrace" "time" ) // HTTPing ... type HTTPing struct { target *Target done chan struct{} result *Result Method string } var _ Pinger = (*HTTPing)(nil) // NewHTTPing return new HTTPing func NewHTTPing(method string) *HTTPing { return &HTTPing{ done: make(chan struct{}), Method: method, } } // SetTarget ... func (ping *HTTPing) SetTarget(target *Target) { ping.target = target if ping.result == nil { ping.result = &Result{Target: target} } } // Start dns func (ping *HTTPing) Start() <-chan struct{} { go func() { t := time.NewTicker(ping.target.Interval) defer t.Stop() for { select { case <-t.C: if ping.result.Counter >= ping.target.Counter && ping.target.Counter != 0 { ping.Stop() return } duration, resp, remoteAddr, err := ping.ping() ping.result.Counter++ if err != nil { fmt.Printf("Ping %s - failed: %s\n", ping.target, err) } else { defer resp.Body.Close() length, _ := io.Copy(ioutil.Discard, resp.Body) fmt.Printf("Ping %s(%s) - %s is open - time=%s method=%s status=%d bytes=%d\n", ping.target, remoteAddr, ping.target.Protocol, duration, ping.Method, resp.StatusCode, length) if ping.result.MinDuration == 0 { ping.result.MinDuration = duration } if ping.result.MaxDuration == 0 { ping.result.MaxDuration = duration } ping.result.SuccessCounter++ if duration > ping.result.MaxDuration { ping.result.MaxDuration = duration } else if duration < ping.result.MinDuration { ping.result.MinDuration = duration } ping.result.TotalDuration += duration } case <-ping.done: return } } }() return ping.done } // Result return dns result func (ping *HTTPing) Result() *Result { return ping.result } // Stop the tcping func (ping *HTTPing) Stop() { ping.done <- struct{}{} } func (ping HTTPing) ping() (time.Duration, *http.Response, string, error) { var resp *http.Response var body io.Reader if ping.Method == "POST" { body = bytes.NewBufferString("{}") } req, err := http.NewRequest(ping.Method, ping.target.String(), body) req.Header.Set(http.CanonicalHeaderKey("User-Agent"), "tcping") if err != nil { return 0, nil, "", err } var remoteAddr string trace := &httptrace.ClientTrace{ ConnectStart: func(network, addr string) { remoteAddr = addr }, } req = req.WithContext(httptrace.WithClientTrace(req.Context(), trace)) duration, errIfce := timeIt(func() interface{} { client := http.Client{Timeout: ping.target.Timeout} resp, err = client.Do(req) return err }) if errIfce != nil { err := errIfce.(error) return 0, nil, "", err } return time.Duration(duration), resp, remoteAddr, nil }