85 lines
1.6 KiB
Go
85 lines
1.6 KiB
Go
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
|
|
}
|