-
Notifications
You must be signed in to change notification settings - Fork 250
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
10 changed files
with
547 additions
and
88 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,111 @@ | ||
| package relabel | ||
|
|
||
| // TODO(thampiotr): write comment | ||
| // TODO(thampiotr): port the test | ||
|
|
||
| import ( | ||
| "crypto/md5" | ||
| "encoding/binary" | ||
| "fmt" | ||
| "strings" | ||
|
|
||
| "github.com/prometheus/common/model" | ||
| ) | ||
|
|
||
| type LabelBuilder interface { | ||
| Get(label string) string | ||
| Range(f func(label string, value string)) | ||
| Set(label string, val string) | ||
| Del(ns ...string) | ||
| } | ||
|
|
||
| // ProcessBuilder is like Process, but the caller passes a labels.Builder | ||
| // containing the initial set of labels, which is mutated by the rules. | ||
| func ProcessBuilder(lb LabelBuilder, cfgs ...*Config) (keep bool) { | ||
| for _, cfg := range cfgs { | ||
| keep = doRelabel(cfg, lb) | ||
| if !keep { | ||
| return false | ||
| } | ||
| } | ||
| return true | ||
| } | ||
|
|
||
| func doRelabel(cfg *Config, lb LabelBuilder) (keep bool) { | ||
| var va [16]string | ||
| values := va[:0] | ||
| if len(cfg.SourceLabels) > cap(values) { | ||
| values = make([]string, 0, len(cfg.SourceLabels)) | ||
| } | ||
| for _, ln := range cfg.SourceLabels { | ||
| values = append(values, lb.Get(string(ln))) | ||
| } | ||
| val := strings.Join(values, cfg.Separator) | ||
|
|
||
| switch cfg.Action { | ||
| case Drop: | ||
| if cfg.Regex.MatchString(val) { | ||
| return false | ||
| } | ||
| case Keep: | ||
| if !cfg.Regex.MatchString(val) { | ||
| return false | ||
| } | ||
| case DropEqual: | ||
| if lb.Get(cfg.TargetLabel) == val { | ||
| return false | ||
| } | ||
| case KeepEqual: | ||
| if lb.Get(cfg.TargetLabel) != val { | ||
| return false | ||
| } | ||
| case Replace: | ||
| indexes := cfg.Regex.FindStringSubmatchIndex(val) | ||
| // If there is no match no replacement must take place. | ||
| if indexes == nil { | ||
| break | ||
| } | ||
| target := model.LabelName(cfg.Regex.ExpandString([]byte{}, cfg.TargetLabel, val, indexes)) | ||
| if !target.IsValid() { | ||
| break | ||
| } | ||
| res := cfg.Regex.ExpandString([]byte{}, cfg.Replacement, val, indexes) | ||
| if len(res) == 0 { | ||
| lb.Del(string(target)) | ||
| break | ||
| } | ||
| lb.Set(string(target), string(res)) | ||
| case Lowercase: | ||
| lb.Set(cfg.TargetLabel, strings.ToLower(val)) | ||
| case Uppercase: | ||
| lb.Set(cfg.TargetLabel, strings.ToUpper(val)) | ||
| case HashMod: | ||
| hash := md5.Sum([]byte(val)) | ||
| // Use only the last 8 bytes of the hash to give the same result as earlier versions of this code. | ||
| mod := binary.BigEndian.Uint64(hash[8:]) % cfg.Modulus | ||
| lb.Set(cfg.TargetLabel, fmt.Sprintf("%d", mod)) | ||
| case LabelMap: | ||
| lb.Range(func(name, value string) { | ||
| if cfg.Regex.MatchString(name) { | ||
| res := cfg.Regex.ReplaceAllString(name, cfg.Replacement) | ||
| lb.Set(res, value) | ||
| } | ||
| }) | ||
| case LabelDrop: | ||
| lb.Range(func(name, value string) { | ||
| if cfg.Regex.MatchString(name) { | ||
| lb.Del(name) | ||
| } | ||
| }) | ||
| case LabelKeep: | ||
| lb.Range(func(name, value string) { | ||
| if !cfg.Regex.MatchString(name) { | ||
| lb.Del(value) | ||
| } | ||
| }) | ||
| default: | ||
| panic(fmt.Errorf("relabel: unknown relabel action type %q", cfg.Action)) | ||
| } | ||
|
|
||
| return true | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
12 changes: 12 additions & 0 deletions
12
internal/component/discovery/benchmark_data/05_typical_pipeline_labels_wrappers.txt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,12 @@ | ||
| goos: darwin | ||
| goarch: arm64 | ||
| pkg: github.com/grafana/alloy/internal/component/discovery | ||
| cpu: Apple M2 | ||
| Benchmark_Targets_TypicalPipeline-8 33 34000016 ns/op 20898478 B/op 160937 allocs/op | ||
| Benchmark_Targets_TypicalPipeline-8 32 36584542 ns/op 20826227 B/op 160935 allocs/op | ||
| Benchmark_Targets_TypicalPipeline-8 30 35797154 ns/op 20827766 B/op 160935 allocs/op | ||
| Benchmark_Targets_TypicalPipeline-8 33 34531398 ns/op 20899688 B/op 160941 allocs/op | ||
| Benchmark_Targets_TypicalPipeline-8 32 35422260 ns/op 20827490 B/op 160933 allocs/op | ||
| Benchmark_Targets_TypicalPipeline-8 33 34420867 ns/op 20897838 B/op 160939 allocs/op | ||
| PASS | ||
| ok github.com/grafana/alloy/internal/component/discovery 8.385s |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.