71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
|
package retrieval
|
||
|
|
|
||
|
|
import (
|
||
|
|
"regexp"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
clientmodel "github.com/prometheus/client_golang/model"
|
||
|
|
|
||
|
|
"github.com/prometheus/prometheus/config"
|
||
|
|
pb "github.com/prometheus/prometheus/config/generated"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Relabel returns a relabeled copy of the given label set. The relabel configurations
|
||
|
|
// are applied in order of input.
|
||
|
|
// If a label set is dropped, nil is returned.
|
||
|
|
func Relabel(labels clientmodel.LabelSet, cfgs ...*config.RelabelConfig) (clientmodel.LabelSet, error) {
|
||
|
|
out := clientmodel.LabelSet{}
|
||
|
|
for ln, lv := range labels {
|
||
|
|
out[ln] = lv
|
||
|
|
}
|
||
|
|
var err error
|
||
|
|
for _, cfg := range cfgs {
|
||
|
|
if out, err = relabel(out, cfg); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if out == nil {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return out, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func relabel(labels clientmodel.LabelSet, cfg *config.RelabelConfig) (clientmodel.LabelSet, error) {
|
||
|
|
pat, err := regexp.Compile(cfg.GetRegex())
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
values := make([]string, 0, len(cfg.GetSourceLabel()))
|
||
|
|
for _, name := range cfg.GetSourceLabel() {
|
||
|
|
values = append(values, string(labels[clientmodel.LabelName(name)]))
|
||
|
|
}
|
||
|
|
val := strings.Join(values, cfg.GetSeparator())
|
||
|
|
|
||
|
|
switch cfg.GetAction() {
|
||
|
|
case pb.RelabelConfig_DROP:
|
||
|
|
if pat.MatchString(val) {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
case pb.RelabelConfig_KEEP:
|
||
|
|
if !pat.MatchString(val) {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
case pb.RelabelConfig_REPLACE:
|
||
|
|
// If there is no match no replacement must take place.
|
||
|
|
if !pat.MatchString(val) {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
res := pat.ReplaceAllString(val, cfg.GetReplacement())
|
||
|
|
ln := clientmodel.LabelName(cfg.GetTargetLabel())
|
||
|
|
if res == "" {
|
||
|
|
delete(labels, ln)
|
||
|
|
} else {
|
||
|
|
labels[ln] = clientmodel.LabelValue(res)
|
||
|
|
}
|
||
|
|
default:
|
||
|
|
panic("retrieval.relabel: unknown relabel action type")
|
||
|
|
}
|
||
|
|
return labels, nil
|
||
|
|
}
|