feat: crono

main.go

@@ -0,0 +1,84 @@
+package crono
+
+import (
+	"fmt"
+	"os"
+	"time"
+
+	"github.com/jedib0t/go-pretty/v6/table"
+)
+
+type Timing struct {
+	Msg     string
+	Elapsed time.Duration
+}
+
+type Crono struct {
+	Begin   time.Time
+	Cursor  time.Time
+	Timings []Timing
+}
+
+// new cronograph with autostart.
+func New() *Crono {
+	c := new(Crono)
+	c.Timings = make([]Timing, 0)
+	c.Begin = time.Now()
+	c.Cursor = c.Begin
+	return c
+}
+
+// restart the crono
+// keeps the begin time.
+func (c *Crono) Restart() {
+	c.Cursor = time.Now()
+}
+
+// push a mark.
+func (c *Crono) Mark(msg string) {
+	c.Timings = append(c.Timings, Timing{msg, c.GetElapsed()})
+}
+
+// push a mark and restart the cursor time.
+func (c *Crono) MarkAndRestart(msg string) {
+	c.Mark(msg)
+	c.Restart()
+}
+
+// time in seconds since last start
+// it doesn't restart the crono.
+func (c *Crono) GetElapsed() time.Duration {
+	return time.Since(c.Cursor)
+}
+
+// total time sice start.
+func (c *Crono) Total() time.Duration {
+	return time.Since(c.Begin)
+}
+
+func (c *Crono) Table() {
+	t := table.NewWriter()
+	t.SetOutputMirror(os.Stdout)
+	t.AppendHeader(table.Row{"#", "Description", "Elapsed"})
+	for i, e := range c.Timings {
+		t.AppendRow([]interface{}{i, e.Msg, durationToString(e.Elapsed)})
+	}
+	t.AppendSeparator()
+	t.AppendFooter(table.Row{"", "Total", durationToString(c.Total())})
+	t.Render()
+}
+
+func durationToString(t time.Duration) string {
+	timeString := fmt.Sprintf("%dms", t.Milliseconds())
+	if t.Seconds() > 3 {
+		timeString += " !!!"
+	} else if t.Seconds() > 2 {
+		timeString += " .!!"
+	} else if t.Seconds() > 1 {
+		timeString += " ..!"
+	} else {
+		timeString += " ..."
+	}
+
+	return timeString
+}