From 797fe0c5db5868774c81cb2ee9d6a3811caf2c60 Mon Sep 17 00:00:00 2001
From: Mikhail Osipov <mike.osipov@gmail.com>
Date: Thu, 12 Mar 2020 19:05:18 +0300
Subject: fix stream time info

---
 pkg/server/tunnel.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'pkg')

diff --git a/pkg/server/tunnel.go b/pkg/server/tunnel.go
index 592f48c..1bb21b4 100644
--- a/pkg/server/tunnel.go
+++ b/pkg/server/tunnel.go
@@ -195,7 +195,13 @@ func (t *tunnel) delStream(s *stream) {
 }
 
 func (s *stream) info() string {
-	d := time.Since(s.since).Milliseconds()
+	var d int64
+
+	if s.until.IsZero() {
+		d = time.Since(s.since).Milliseconds()
+	} else {
+		d = s.until.Sub(s.since).Milliseconds()
+	}
 
 	return fmt.Sprintf("%.3fs %d/%d -> %d/%d",
 		float64(d)/1000.0,
-- 
cgit v1.2.3-70-g09d2