mirrors/bird_exporter
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

reuse regex, ensure custom label names are trimmed

Browse Source
This commit is contained in:
Daniel Czerwonk 2022-02-11 07:19:04 +01:00
parent f23e0d3b3e
commit f7bc5148a4
1 changed files with 17 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
metrics/default_label_strategy.go
View File

@ -2,6 +2,7 @@ package metrics
import ( import (
"regexp" "regexp"
"strings"
"github.com/czerwonk/bird_exporter/protocol" "github.com/czerwonk/bird_exporter/protocol"
) )
@ -9,13 +10,13 @@ import (
// DefaultLabelStrategy defines the labels to add to an metric and its data retrieval method // DefaultLabelStrategy defines the labels to add to an metric and its data retrieval method
type DefaultLabelStrategy struct { type DefaultLabelStrategy struct {
descriptionLabels bool descriptionLabels bool
descriptionLabelsRegex string descriptionLabelsRegex *regexp.Regexp
} }
func NewDefaultLabelStrategy(descriptionLabels bool, descriptionLabelsRegex string) *DefaultLabelStrategy { func NewDefaultLabelStrategy(descriptionLabels bool, descriptionLabelsRegex string) *DefaultLabelStrategy {
return &DefaultLabelStrategy{ return &DefaultLabelStrategy{
descriptionLabels: descriptionLabels, descriptionLabels: descriptionLabels,
descriptionLabelsRegex: descriptionLabelsRegex, descriptionLabelsRegex: regexp.MustCompile(descriptionLabelsRegex),
} }
} }
@ -39,32 +40,26 @@ func (d *DefaultLabelStrategy) LabelValues(p *protocol.Protocol) []string {
return res return res
} }
func labelKeysFromDescription(desc string, d *DefaultLabelStrategy) (res []string) { func labelKeysFromDescription(desc string, d *DefaultLabelStrategy) []string {
reAllStringSubmatch := labelFindAllStringSubmatch(desc, d) res := []string{}
for _, submatch := range reAllStringSubmatch {
res = append(res, submatch[1]) matches := d.descriptionLabelsRegex.FindAllStringSubmatch(desc, -1)
for _, submatch := range matches {
res = append(res, strings.TrimSpace(submatch[1]))
} }
return return res
} }
func labelValuesFromDescription(desc string, d *DefaultLabelStrategy) (res []string) { func labelValuesFromDescription(desc string, d *DefaultLabelStrategy) []string {
reAllStringSubmatch := labelFindAllStringSubmatch(desc, d) res := []string{}
for _, submatch := range reAllStringSubmatch {
res = append(res, submatch[2]) matches := d.descriptionLabelsRegex.FindAllStringSubmatch(desc, -1)
for _, submatch := range matches {
res = append(res, strings.TrimSpace(submatch[2]))
} }
return return res
}
func labelFindAllStringSubmatch(desc string, d *DefaultLabelStrategy) (result [][]string) {
// Regex pattern captures "key: value" pair from the content.
pattern := regexp.MustCompile(d.descriptionLabelsRegex)
result = pattern.FindAllStringSubmatch(desc, -1)
return
} }
func protoString(p *protocol.Protocol) string { func protoString(p *protocol.Protocol) string {