-
Notifications
You must be signed in to change notification settings - Fork 3.4k
/
drop_labels.go
86 lines (74 loc) · 1.65 KB
/
drop_labels.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package log
import (
"github.com/prometheus/prometheus/model/labels"
"github.com/grafana/loki/pkg/logqlmodel"
)
type DropLabels struct {
dropLabels []DropLabel
}
type DropLabel struct {
Matcher *labels.Matcher
Name string
}
func NewDropLabel(matcher *labels.Matcher, name string) DropLabel {
return DropLabel{
Matcher: matcher,
Name: name,
}
}
func NewDropLabels(dl []DropLabel) *DropLabels {
return &DropLabels{dropLabels: dl}
}
func (dl *DropLabels) Process(_ int64, line []byte, lbls *LabelsBuilder) ([]byte, bool) {
for _, dropLabel := range dl.dropLabels {
if dropLabel.Matcher != nil {
dropLabelMatches(dropLabel.Matcher, lbls)
continue
}
name := dropLabel.Name
dropLabelNames(name, lbls)
}
return line, true
}
func (dl *DropLabels) RequiredLabelNames() []string { return []string{} }
func isErrorLabel(name string) bool {
return name == logqlmodel.ErrorLabel
}
func isErrorDetailsLabel(name string) bool {
return name == logqlmodel.ErrorDetailsLabel
}
func dropLabelNames(name string, lbls *LabelsBuilder) {
if isErrorLabel(name) {
lbls.ResetError()
return
}
if isErrorDetailsLabel(name) {
lbls.ResetErrorDetails()
return
}
if _, ok := lbls.Get(name); ok {
lbls.Del(name)
}
}
func dropLabelMatches(matcher *labels.Matcher, lbls *LabelsBuilder) {
var value string
name := matcher.Name
if isErrorLabel(name) {
value = lbls.GetErr()
if matcher.Matches(value) {
lbls.ResetError()
}
return
}
if isErrorDetailsLabel(name) {
value = lbls.GetErrorDetails()
if matcher.Matches(value) {
lbls.ResetErrorDetails()
}
return
}
value, _ = lbls.Get(name)
if matcher.Matches(value) {
lbls.Del(name)
}
}