convertnhcb: use CutSuffix instead of regex replace for histogram name

This is much quicker.

Signed-off-by: György Krajcsovits <gyorgy.krajcsovits@grafana.com>
This commit is contained in:
György Krajcsovits 2024-10-14 16:38:56 +02:00
parent 522149a2ae
commit 78de9bd10f

View File

@ -19,30 +19,10 @@ import (
"sort" "sort"
"strings" "strings"
"github.com/grafana/regexp"
"github.com/prometheus/prometheus/model/histogram" "github.com/prometheus/prometheus/model/histogram"
"github.com/prometheus/prometheus/model/labels" "github.com/prometheus/prometheus/model/labels"
) )
var histogramNameSuffixReplacements = []struct {
pattern *regexp.Regexp
repl string
}{
{
pattern: regexp.MustCompile(`_bucket$`),
repl: "",
},
{
pattern: regexp.MustCompile(`_sum$`),
repl: "",
},
{
pattern: regexp.MustCompile(`_count$`),
repl: "",
},
}
// TempHistogram is used to collect information about classic histogram // TempHistogram is used to collect information about classic histogram
// samples incrementally before creating a histogram.Histogram or // samples incrementally before creating a histogram.Histogram or
// histogram.FloatHistogram based on the values collected. // histogram.FloatHistogram based on the values collected.
@ -176,9 +156,18 @@ func GetHistogramMetricBase(m labels.Labels, suffix string) labels.Labels {
Labels() Labels()
} }
// GetHistogramMetricBaseName removes the suffixes _bucket, _sum, _count from
// the metric name. We specifically do not remove the _created suffix as that
// should be removed by the caller.
func GetHistogramMetricBaseName(s string) string { func GetHistogramMetricBaseName(s string) string {
for _, rep := range histogramNameSuffixReplacements { if r, ok := strings.CutSuffix(s, "_bucket"); ok {
s = rep.pattern.ReplaceAllString(s, rep.repl) return r
}
if r, ok := strings.CutSuffix(s, "_sum"); ok {
return r
}
if r, ok := strings.CutSuffix(s, "_count"); ok {
return r
} }
return s return s
} }