util: add TimeProfile

ichiaohsu · Jun 16, 2021 · 2d520df · 2d520df
1 parent cce9e77
commit 2d520df
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/pkg/util/profile.go b/pkg/util/profile.go
@@ -0,0 +1,22 @@
+package util
+
+import "time"
+
+type TimeProfile struct {
+	StartTime, EndTime time.Time
+	Duration           time.Duration
+}
+
+func StartTimeProfile() TimeProfile {
+	return TimeProfile{StartTime: time.Now()}
+}
+
+func (p *TimeProfile) TilNow() time.Duration {
+	return time.Now().Sub(p.StartTime)
+}
+
+func (p *TimeProfile) Stop() time.Duration {
+	p.EndTime = time.Now()
+	p.Duration = p.EndTime.Sub(p.StartTime)
+	return p.Duration
+}