forked from redpanda-data/connect
-
Notifications
You must be signed in to change notification settings - Fork 0
/
tracers.go
99 lines (83 loc) · 2.95 KB
/
tracers.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
package bundle
import (
"fmt"
"sort"
"go.opentelemetry.io/otel/trace"
"github.com/benthosdev/benthos/v4/internal/component"
"github.com/benthosdev/benthos/v4/internal/component/tracer"
"github.com/benthosdev/benthos/v4/internal/docs"
)
// AllTracers is a set containing every single tracer that has been imported.
var AllTracers = &TracerSet{
specs: map[string]tracerSpec{},
}
//------------------------------------------------------------------------------
// TracersAdd adds a new tracers exporter to this environment by providing a
// constructor and documentation.
func (e *Environment) TracersAdd(constructor TracerConstructor, spec docs.ComponentSpec) error {
return e.tracers.Add(constructor, spec)
}
// TracersInit attempts to initialise a tracers exporter from a config.
func (e *Environment) TracersInit(conf tracer.Config, nm NewManagement) (trace.TracerProvider, error) {
return e.tracers.Init(conf, nm)
}
// TracersDocs returns a slice of tracers exporter specs.
func (e *Environment) TracersDocs() []docs.ComponentSpec {
return e.tracers.Docs()
}
//------------------------------------------------------------------------------
// TracerConstructor constructs an tracer component.
type TracerConstructor func(tracer.Config, NewManagement) (trace.TracerProvider, error)
type tracerSpec struct {
constructor TracerConstructor
spec docs.ComponentSpec
}
// TracerSet contains an explicit set of tracers available to a Benthos service.
type TracerSet struct {
specs map[string]tracerSpec
}
// Add a new tracer to this set by providing a spec (name, documentation, and
// constructor).
func (s *TracerSet) Add(constructor TracerConstructor, spec docs.ComponentSpec) error {
if !nameRegexp.MatchString(spec.Name) {
return fmt.Errorf("component name '%v' does not match the required regular expression /%v/", spec.Name, nameRegexpRaw)
}
if s.specs == nil {
s.specs = map[string]tracerSpec{}
}
spec.Type = docs.TypeTracer
s.specs[spec.Name] = tracerSpec{
constructor: constructor,
spec: spec,
}
docs.DeprecatedProvider.RegisterDocs(spec)
return nil
}
// Init attempts to initialise an tracer from a config.
func (s *TracerSet) Init(conf tracer.Config, nm NewManagement) (trace.TracerProvider, error) {
spec, exists := s.specs[conf.Type]
if !exists {
return nil, component.ErrInvalidType("tracer", conf.Type)
}
return spec.constructor(conf, nm)
}
// Docs returns a slice of tracer specs, which document each method.
func (s *TracerSet) Docs() []docs.ComponentSpec {
var docs []docs.ComponentSpec
for _, v := range s.specs {
docs = append(docs, v.spec)
}
sort.Slice(docs, func(i, j int) bool {
return docs[i].Name < docs[j].Name
})
return docs
}
// DocsFor returns the documentation for a given component name, returns a
// boolean indicating whether the component name exists.
func (s *TracerSet) DocsFor(name string) (docs.ComponentSpec, bool) {
c, ok := s.specs[name]
if !ok {
return docs.ComponentSpec{}, false
}
return c.spec, true
}