forked from redpanda-data/connect
-
Notifications
You must be signed in to change notification settings - Fork 0
/
registry.go
142 lines (128 loc) · 3.85 KB
/
registry.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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
package docs
import (
"sync"
)
// Provider stores the component spec definitions of various component
// implementations.
type Provider interface {
GetDocs(name string, ctype Type) (ComponentSpec, bool)
}
// DeprecatedProvider is a globally declared docs provider that enables the old
// config parse style to work with dynamic plugins. Eventually we can eliminate
// all the UnmarshalYAML methods on config structs and remove this as well.
var DeprecatedProvider = NewMappedDocsProvider()
//------------------------------------------------------------------------------
// MappedDocsProvider stores component documentation in maps, protected by a
// mutex, allowing safe concurrent use.
type MappedDocsProvider struct {
bufferMap map[string]ComponentSpec
cacheMap map[string]ComponentSpec
inputMap map[string]ComponentSpec
metricsMap map[string]ComponentSpec
outputMap map[string]ComponentSpec
processorMap map[string]ComponentSpec
rateLimitMap map[string]ComponentSpec
tracerMap map[string]ComponentSpec
componentLock sync.Mutex
}
// NewMappedDocsProvider creates a new (empty) provider of component docs.
func NewMappedDocsProvider() *MappedDocsProvider {
return &MappedDocsProvider{
bufferMap: map[string]ComponentSpec{},
cacheMap: map[string]ComponentSpec{},
inputMap: map[string]ComponentSpec{},
metricsMap: map[string]ComponentSpec{},
outputMap: map[string]ComponentSpec{},
processorMap: map[string]ComponentSpec{},
rateLimitMap: map[string]ComponentSpec{},
tracerMap: map[string]ComponentSpec{},
}
}
// Clone returns a copied version of the provider that can be modified
// independently.
func (m *MappedDocsProvider) Clone() *MappedDocsProvider {
newM := &MappedDocsProvider{
bufferMap: map[string]ComponentSpec{},
cacheMap: map[string]ComponentSpec{},
inputMap: map[string]ComponentSpec{},
metricsMap: map[string]ComponentSpec{},
outputMap: map[string]ComponentSpec{},
processorMap: map[string]ComponentSpec{},
rateLimitMap: map[string]ComponentSpec{},
tracerMap: map[string]ComponentSpec{},
}
for k, v := range m.bufferMap {
newM.bufferMap[k] = v
}
for k, v := range m.cacheMap {
newM.cacheMap[k] = v
}
for k, v := range m.inputMap {
newM.inputMap[k] = v
}
for k, v := range m.metricsMap {
newM.metricsMap[k] = v
}
for k, v := range m.outputMap {
newM.outputMap[k] = v
}
for k, v := range m.processorMap {
newM.processorMap[k] = v
}
for k, v := range m.rateLimitMap {
newM.rateLimitMap[k] = v
}
for k, v := range m.tracerMap {
newM.tracerMap[k] = v
}
return newM
}
// RegisterDocs adds the documentation of a component implementation.
func (m *MappedDocsProvider) RegisterDocs(spec ComponentSpec) {
m.componentLock.Lock()
defer m.componentLock.Unlock()
switch spec.Type {
case TypeBuffer:
m.bufferMap[spec.Name] = spec
case TypeCache:
m.cacheMap[spec.Name] = spec
case TypeInput:
m.inputMap[spec.Name] = spec
case TypeMetrics:
m.metricsMap[spec.Name] = spec
case TypeOutput:
m.outputMap[spec.Name] = spec
case TypeProcessor:
m.processorMap[spec.Name] = spec
case TypeRateLimit:
m.rateLimitMap[spec.Name] = spec
case TypeTracer:
m.tracerMap[spec.Name] = spec
}
}
// GetDocs attempts to obtain component implementation docs.
func (m *MappedDocsProvider) GetDocs(name string, ctype Type) (ComponentSpec, bool) {
m.componentLock.Lock()
defer m.componentLock.Unlock()
var spec ComponentSpec
var ok bool
switch ctype {
case TypeBuffer:
spec, ok = m.bufferMap[name]
case TypeCache:
spec, ok = m.cacheMap[name]
case TypeInput:
spec, ok = m.inputMap[name]
case TypeMetrics:
spec, ok = m.metricsMap[name]
case TypeOutput:
spec, ok = m.outputMap[name]
case TypeProcessor:
spec, ok = m.processorMap[name]
case TypeRateLimit:
spec, ok = m.rateLimitMap[name]
case TypeTracer:
spec, ok = m.tracerMap[name]
}
return spec, ok
}