sampler-fork/component/gauge/gauge.go

130 lines
2.7 KiB
Go
Raw Normal View History

2019-02-26 04:36:23 +00:00
package gauge
import (
"fmt"
2019-03-14 03:01:44 +00:00
ui "github.com/gizak/termui/v3"
2019-03-13 03:19:19 +00:00
"github.com/sqshq/sampler/component"
"github.com/sqshq/sampler/config"
2019-02-26 04:36:23 +00:00
"github.com/sqshq/sampler/console"
"github.com/sqshq/sampler/data"
"image"
"math"
"strconv"
)
const (
MinValueLabel = "min"
MaxValueLabel = "max"
CurValueLabel = "cur"
)
type Gauge struct {
2019-03-16 23:59:28 +00:00
*ui.Block
*data.Consumer
alert *data.Alert
2019-02-26 04:36:23 +00:00
minValue float64
maxValue float64
curValue float64
color ui.Color
scale int
palette console.Palette
2019-02-26 04:36:23 +00:00
}
func NewGauge(c config.GaugeConfig, palette console.Palette) *Gauge {
2019-03-08 04:04:46 +00:00
gauge := Gauge{
Block: component.NewBlock(c.Title, true, palette),
2019-03-16 23:59:28 +00:00
Consumer: data.NewConsumer(),
scale: *c.Scale,
color: *c.Color,
palette: palette,
2019-03-08 04:04:46 +00:00
}
go func() {
for {
select {
case sample := <-gauge.SampleChannel:
gauge.ConsumeSample(sample)
2019-03-16 23:59:28 +00:00
case alert := <-gauge.AlertChannel:
gauge.alert = alert
}
}
}()
2019-03-08 04:04:46 +00:00
return &gauge
}
2019-03-16 23:59:28 +00:00
func (g *Gauge) ConsumeSample(sample *data.Sample) {
2019-02-26 04:36:23 +00:00
float, err := strconv.ParseFloat(sample.Value, 64)
if err != nil {
2019-03-16 23:59:28 +00:00
g.AlertChannel <- &data.Alert{
Title: "FAILED TO PARSE NUMBER",
Text: err.Error(),
Color: sample.Color,
}
return
2019-02-26 04:36:23 +00:00
}
switch sample.Label {
case MinValueLabel:
g.minValue = float
break
case MaxValueLabel:
g.maxValue = float
break
case CurValueLabel:
g.curValue = float
break
}
}
2019-03-16 23:59:28 +00:00
func (g *Gauge) Draw(buffer *ui.Buffer) {
2019-02-26 04:36:23 +00:00
2019-03-16 23:59:28 +00:00
g.Block.Draw(buffer)
2019-02-26 04:36:23 +00:00
percent := 0.0
2019-02-27 04:23:56 +00:00
if g.curValue != 0 && g.maxValue != g.minValue {
2019-02-26 04:36:23 +00:00
percent = (100 * g.curValue) / (g.maxValue - g.minValue)
}
label := fmt.Sprintf("%v%% (%v)", formatValue(percent, g.scale), g.curValue)
// draw bar
2019-02-26 04:36:23 +00:00
barWidth := int((percent / 100) * float64(g.Inner.Dx()))
if barWidth == 0 {
barWidth = 1
} else if barWidth > g.Dx()-2 {
barWidth = g.Dx() - 2
}
2019-03-16 23:59:28 +00:00
buffer.Fill(
2019-02-26 04:36:23 +00:00
ui.NewCell(console.SymbolVerticalBar, ui.NewStyle(g.color)),
image.Rect(g.Inner.Min.X+1, g.Inner.Min.Y, g.Inner.Min.X+barWidth, g.Inner.Max.Y),
)
// draw label
2019-02-26 04:36:23 +00:00
labelXCoordinate := g.Inner.Min.X + (g.Inner.Dx() / 2) - int(float64(len(label))/2)
labelYCoordinate := g.Inner.Min.Y + ((g.Inner.Dy() - 1) / 2)
if labelYCoordinate < g.Inner.Max.Y {
for i, char := range label {
style := ui.NewStyle(g.palette.BaseColor)
2019-02-26 04:36:23 +00:00
if labelXCoordinate+i+1 <= g.Inner.Min.X+barWidth {
style = ui.NewStyle(g.palette.BaseColor, ui.ColorClear)
2019-02-26 04:36:23 +00:00
}
2019-03-16 23:59:28 +00:00
buffer.SetCell(ui.NewCell(char, style), image.Pt(labelXCoordinate+i, labelYCoordinate))
2019-02-26 04:36:23 +00:00
}
}
2019-03-16 23:59:28 +00:00
component.RenderAlert(g.alert, g.Rectangle, buffer)
2019-02-26 04:36:23 +00:00
}
// TODO extract to utils
func formatValue(value float64, scale int) string {
if math.Abs(value) == math.MaxFloat64 {
return "Inf"
} else {
format := "%." + strconv.Itoa(scale) + "f"
return fmt.Sprintf(format, value)
}
}