mirror of
https://github.com/crowdsecurity/crowdsec.git
synced 2025-05-13 21:05:52 +02:00
8d96ddd48e
* fix empty line between metrics tables * refact metrics tables: use go-pretty api directly * lint
42 lines
1.3 KiB
Go
42 lines
1.3 KiB
Go
package climetrics
|
||
|
||
import (
|
||
"io"
|
||
|
||
"github.com/jedib0t/go-pretty/v6/table"
|
||
log "github.com/sirupsen/logrus"
|
||
|
||
"github.com/crowdsecurity/crowdsec/cmd/crowdsec-cli/cstable"
|
||
)
|
||
|
||
type statBucket map[string]map[string]int
|
||
|
||
func (s statBucket) Description() (string, string) {
|
||
return "Scenario Metrics",
|
||
`Measure events in different scenarios. Current count is the number of buckets during metrics collection. ` +
|
||
`Overflows are past event-producing buckets, while Expired are the ones that didn’t receive enough events to Overflow.`
|
||
}
|
||
|
||
func (s statBucket) Process(bucket, metric string, val int) {
|
||
if _, ok := s[bucket]; !ok {
|
||
s[bucket] = make(map[string]int)
|
||
}
|
||
|
||
s[bucket][metric] += val
|
||
}
|
||
|
||
func (s statBucket) Table(out io.Writer, wantColor string, noUnit bool, showEmpty bool) {
|
||
t := cstable.New(out, wantColor).Writer
|
||
t.AppendHeader(table.Row{"Scenario", "Current Count", "Overflows", "Instantiated", "Poured", "Expired"})
|
||
|
||
keys := []string{"curr_count", "overflow", "instantiation", "pour", "underflow"}
|
||
|
||
if numRows, err := metricsToTable(t, s, keys, noUnit); err != nil {
|
||
log.Warningf("while collecting scenario stats: %s", err)
|
||
} else if numRows > 0 || showEmpty {
|
||
title, _ := s.Description()
|
||
io.WriteString(out, title + ":\n")
|
||
io.WriteString(out, t.Render() + "\n")
|
||
io.WriteString(out, "\n")
|
||
}
|
||
}
|