90 lines
2.2 KiB
Go
90 lines
2.2 KiB
Go
package metrics
|
|
|
|
import (
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/czerwonk/bird_exporter/protocol"
|
|
)
|
|
|
|
// DefaultLabelStrategy defines the labels to add to an metric and its data retrieval method
|
|
type DefaultLabelStrategy struct {
|
|
descriptionLabels bool
|
|
descriptionLabelsRegex *regexp.Regexp
|
|
}
|
|
|
|
func NewDefaultLabelStrategy(descriptionLabels bool, descriptionLabelsRegex string) *DefaultLabelStrategy {
|
|
return &DefaultLabelStrategy{
|
|
descriptionLabels: descriptionLabels,
|
|
descriptionLabelsRegex: regexp.MustCompile(descriptionLabelsRegex),
|
|
}
|
|
}
|
|
|
|
// LabelNames returns the list of label names
|
|
func (d *DefaultLabelStrategy) LabelNames(p *protocol.Protocol) []string {
|
|
res := []string{"name", "proto", "ip_version", "import_filter", "export_filter"}
|
|
if d.descriptionLabels && p.Description != "" {
|
|
res = append(res, labelKeysFromDescription(p.Description, d)...)
|
|
}
|
|
|
|
return res
|
|
}
|
|
|
|
// LabelValues returns the values for a protocol
|
|
func (d *DefaultLabelStrategy) LabelValues(p *protocol.Protocol) []string {
|
|
res := []string{p.Name, protoString(p), p.IPVersion, p.ImportFilter, p.ExportFilter}
|
|
if d.descriptionLabels && p.Description != "" {
|
|
res = append(res, labelValuesFromDescription(p.Description, d)...)
|
|
}
|
|
|
|
return res
|
|
}
|
|
|
|
func labelKeysFromDescription(desc string, d *DefaultLabelStrategy) []string {
|
|
res := []string{}
|
|
|
|
matches := d.descriptionLabelsRegex.FindAllStringSubmatch(desc, -1)
|
|
for _, submatch := range matches {
|
|
res = append(res, strings.TrimSpace(submatch[1]))
|
|
}
|
|
|
|
return res
|
|
}
|
|
|
|
func labelValuesFromDescription(desc string, d *DefaultLabelStrategy) []string {
|
|
res := []string{}
|
|
|
|
matches := d.descriptionLabelsRegex.FindAllStringSubmatch(desc, -1)
|
|
for _, submatch := range matches {
|
|
res = append(res, strings.TrimSpace(submatch[2]))
|
|
}
|
|
|
|
return res
|
|
}
|
|
|
|
func protoString(p *protocol.Protocol) string {
|
|
switch p.Proto {
|
|
case protocol.BGP:
|
|
return "BGP"
|
|
case protocol.OSPF:
|
|
if p.IPVersion == "4" {
|
|
return "OSPF"
|
|
}
|
|
return "OSPFv3"
|
|
case protocol.Static:
|
|
return "Static"
|
|
case protocol.Kernel:
|
|
return "Kernel"
|
|
case protocol.Direct:
|
|
return "Direct"
|
|
case protocol.Babel:
|
|
return "Babel"
|
|
case protocol.RPKI:
|
|
return "RPKI"
|
|
case protocol.BFD:
|
|
return "BFD"
|
|
}
|
|
|
|
return ""
|
|
}
|