From 65e7761f06c9355235009ea8d00cf5c1f93a2f3b Mon Sep 17 00:00:00 2001
From: Denys Kashkovskyi <dannie.k@me.com>
Date: Wed, 13 Apr 2022 22:41:40 +0000
Subject: [PATCH] Transport with metrics

---
 core/transport.go | 57 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)
 create mode 100644 core/transport.go

diff --git a/core/transport.go b/core/transport.go
new file mode 100644
index 0000000..4ffed7d
--- /dev/null
+++ b/core/transport.go
@@ -0,0 +1,57 @@
+package core
+
+import (
+	"net"
+	"net/http"
+	"time"
+)
+
+type Transport struct {
+	rtp       http.RoundTripper
+	dialer    *net.Dialer
+	connStart time.Time
+	connEnd   time.Time
+	reqStart  time.Time
+	reqEnd    time.Time
+}
+
+func NewTransport(timeout time.Duration) *Transport {
+	tr := &Transport{
+		dialer: &net.Dialer{
+			Timeout:   timeout,
+			KeepAlive: timeout,
+		},
+	}
+	tr.rtp = &http.Transport{
+		Proxy:               http.ProxyFromEnvironment,
+		Dial:                tr.Dial,
+		TLSHandshakeTimeout: timeout,
+	}
+	return tr
+}
+
+func (tr *Transport) RoundTrip(r *http.Request) (*http.Response, error) {
+	tr.reqStart = time.Now()
+	resp, err := tr.rtp.RoundTrip(r)
+	tr.reqEnd = time.Now()
+	return resp, err
+}
+
+func (tr *Transport) Dial(network, addr string) (net.Conn, error) {
+	tr.connStart = time.Now()
+	cn, err := tr.dialer.Dial(network, addr)
+	tr.connEnd = time.Now()
+	return cn, err
+}
+
+func (tr *Transport) ReqDuration() time.Duration {
+	return tr.Duration() - tr.ConnDuration()
+}
+
+func (tr *Transport) ConnDuration() time.Duration {
+	return tr.connEnd.Sub(tr.connStart)
+}
+
+func (tr *Transport) Duration() time.Duration {
+	return tr.reqEnd.Sub(tr.reqStart)
+}