diff options
Diffstat (limited to 'tools/worker_table.go')
-rw-r--r-- | tools/worker_table.go | 61 |
1 files changed, 61 insertions, 0 deletions
diff --git a/tools/worker_table.go b/tools/worker_table.go new file mode 100644 index 00000000..4aeb6ae7 --- /dev/null +++ b/tools/worker_table.go @@ -0,0 +1,61 @@ +package tools + +import ( + "io" + "strconv" + "time" + + "github.com/dustin/go-humanize" + "github.com/fatih/color" + "github.com/olekukonko/tablewriter" +) + +// WorkerTable renders table with information about rr server workers. +func WorkerTable(writer io.Writer, workers []ProcessState) *tablewriter.Table { + tw := tablewriter.NewWriter(writer) + tw.SetHeader([]string{"PID", "Status", "Execs", "Memory", "Created"}) + tw.SetColMinWidth(0, 7) + tw.SetColMinWidth(1, 9) + tw.SetColMinWidth(2, 7) + tw.SetColMinWidth(3, 7) + tw.SetColMinWidth(4, 18) + + for key := range workers { + tw.Append([]string{ + strconv.Itoa(workers[key].Pid), + renderStatus(workers[key].Status), + renderJobs(workers[key].NumJobs), + humanize.Bytes(workers[key].MemoryUsage), + renderAlive(time.Unix(0, workers[key].Created)), + }) + } + + return tw +} + +func renderStatus(status string) string { + switch status { + case "inactive": + return color.YellowString("inactive") + case "ready": + return color.CyanString("ready") + case "working": + return color.GreenString("working") + case "invalid": + return color.YellowString("invalid") + case "stopped": + return color.RedString("stopped") + case "errored": + return color.RedString("errored") + } + + return status +} + +func renderJobs(number int64) string { + return humanize.Comma(number) +} + +func renderAlive(t time.Time) string { + return humanize.RelTime(t, time.Now(), "ago", "") +} |