This repository has been archived by the owner on Jul 31, 2023. It is now read-only.
/
span_filter.go
123 lines (102 loc) · 2.4 KB
/
span_filter.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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package tracer
import (
"fmt"
ot "github.com/opentracing/opentracing-go"
)
type SpanFilter interface {
Accept(*spanImpl) bool
}
type SpanFilterer interface {
Accept(*spanImpl) bool
}
type ThundraSpanFilterer struct {
spanFilters []SpanFilter
all bool
}
type ThundraSpanFilter struct {
DomainName string
ClassName string
OperationName string
Reverse bool
Tags ot.Tags
}
type CompositeSpanFilter struct {
spanFilters []SpanFilter
all bool
composite bool
}
func (f *CompositeSpanFilter) Accept(span *spanImpl) bool {
res := f.all
for _, sf := range f.spanFilters {
if f.all {
res = res && sf.Accept(span)
} else {
res = res || sf.Accept(span)
}
}
return res
}
func (t *ThundraSpanFilterer) Accept(span *spanImpl) bool {
res := t.all
for _, sf := range t.spanFilters {
if t.all {
res = res && sf.Accept(span)
} else {
res = res || sf.Accept(span)
}
}
return res
}
func (t *ThundraSpanFilterer) AddFilter(sf SpanFilter) {
t.spanFilters = append(t.spanFilters, sf)
}
func (t *ThundraSpanFilterer) ClearFilters() {
t.spanFilters = []SpanFilter{}
}
func (t *ThundraSpanFilter) Accept(span *spanImpl) bool {
accepted := true
if span == nil {
return accepted
}
if t.DomainName != "" {
accepted = (t.DomainName == span.raw.DomainName)
}
if accepted && t.ClassName != "" {
accepted = (t.ClassName == span.raw.ClassName)
}
if accepted && t.OperationName != "" {
accepted = (t.OperationName == span.raw.OperationName)
}
if accepted && t.Tags != nil {
for k, v := range t.Tags {
if fmt.Sprintf("%v", span.raw.GetTag(k)) != fmt.Sprintf("%v", v) {
accepted = false
break
}
}
}
if t.Reverse {
return !accepted
}
return accepted
}
// NewThundraSpanFilter creates and returns a new ThundraSpanFilter from config
func NewThundraSpanFilter(config map[string]interface{}) *ThundraSpanFilter {
spanFilter := ThundraSpanFilter{}
if domainName, ok := config["domainName"].(string); ok {
spanFilter.DomainName = domainName
}
if className, ok := config["className"].(string); ok {
spanFilter.ClassName = className
}
if operationName, ok := config["operationName"].(string); ok {
spanFilter.OperationName = operationName
}
if reverse, ok := config["reverse"].(bool); ok {
spanFilter.Reverse = reverse
}
if tags, ok := config["tags"].(map[string]interface{}); ok {
spanFilter.Tags = tags
}
return &spanFilter
}