mirror of
https://github.com/hanzoai/traces.git
synced 2026-10-30 18:20:41 +00:00
Javiandzeekay
c30a66e33d
* refactor code to make it simpler to assert if a span is excluded * added includeOnly to the config * add includeOnly to the list of policies * short circuit * docs * fmt * tune the logic to handle the absense of include filters * fix docs * rename it to include_any and add more test coverage delete .gocache * update schema * changelog * update config * fix validation tests * change method name
121 lines
2.8 KiB
Go
121 lines
2.8 KiB
Go
package spanfilter
|
|
|
|
import (
|
|
"github.com/grafana/tempo/pkg/spanfilter/config"
|
|
v1 "github.com/grafana/tempo/pkg/tempopb/resource/v1"
|
|
tracev1 "github.com/grafana/tempo/pkg/tempopb/trace/v1"
|
|
)
|
|
|
|
type SpanFilter struct {
|
|
include []*splitPolicy
|
|
includeAny []*splitPolicy
|
|
exclude []*splitPolicy
|
|
|
|
hasInclude bool
|
|
hasIncludeAny bool
|
|
hasExclude bool
|
|
}
|
|
|
|
// NewSpanFilter returns a SpanFilter that will filter spans based on the given filter policies.
|
|
func NewSpanFilter(filterPolicies []config.FilterPolicy) (*SpanFilter, error) {
|
|
sf := new(SpanFilter)
|
|
for _, policy := range filterPolicies {
|
|
err := config.ValidateFilterPolicy(policy)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
include, err := getSplitPolicy(policy.Include)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if include != nil {
|
|
sf.include = append(sf.include, include)
|
|
sf.hasInclude = true
|
|
}
|
|
|
|
includeAny, err := getSplitPolicy(policy.IncludeAny)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if includeAny != nil {
|
|
sf.includeAny = append(sf.includeAny, includeAny)
|
|
sf.hasIncludeAny = true
|
|
}
|
|
|
|
exclude, err := getSplitPolicy(policy.Exclude)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if exclude != nil {
|
|
sf.exclude = append(sf.exclude, exclude)
|
|
sf.hasExclude = true
|
|
}
|
|
}
|
|
|
|
return sf, nil
|
|
}
|
|
|
|
// ApplyFilterPolicy returns true if the span should be included in the metrics.
|
|
func (f *SpanFilter) ApplyFilterPolicy(rs *v1.Resource, span *tracev1.Span) bool {
|
|
// With no filter policies specified, all spans are included.
|
|
if !f.hasInclude && !f.hasIncludeAny && !f.hasExclude {
|
|
return true
|
|
}
|
|
|
|
if f.hasExclude && f.isExcluded(rs, span) {
|
|
return false
|
|
}
|
|
|
|
if f.hasIncludeAny && f.isIncludedAny(rs, span) {
|
|
return true
|
|
}
|
|
|
|
if f.hasInclude {
|
|
return f.isIncluded(rs, span)
|
|
}
|
|
|
|
// If we have an include_any but NO standard include, and we reached
|
|
// here, it means include_any didn't match. -> return false.
|
|
// IF NO inclusion rules exist at all -> return true.
|
|
return !f.hasIncludeAny
|
|
}
|
|
|
|
// This is different than the isIncluded. It's a VIP pass,working as an OR expression.
|
|
// if ANY policy matches the span is included
|
|
func (f *SpanFilter) isIncludedAny(rs *v1.Resource, span *tracev1.Span) bool {
|
|
for _, policy := range f.includeAny {
|
|
if policy.Match(rs, span) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (f *SpanFilter) isIncluded(rs *v1.Resource, span *tracev1.Span) bool {
|
|
for _, policy := range f.include {
|
|
if !policy.Match(rs, span) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (f *SpanFilter) isExcluded(rs *v1.Resource, span *tracev1.Span) bool {
|
|
for _, policy := range f.exclude {
|
|
if policy.Match(rs, span) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func getSplitPolicy(policy *config.PolicyMatch) (*splitPolicy, error) {
|
|
if policy == nil {
|
|
return nil, nil
|
|
}
|
|
return newSplitPolicy(policy)
|
|
}
|