forked from redpanda-data/connect
-
Notifications
You must be signed in to change notification settings - Fork 0
/
registry.go
132 lines (118 loc) · 3.59 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
package docs
import (
"sync"
)
// Provider stores the component spec definitions of various component
// implementations.
type Provider interface {
GetDocs(name string, ctype Type) (ComponentSpec, bool)
}
var globalProvider = NewMappedDocsProvider()
// RegisterDocs stores the documentation spec for a component.
func RegisterDocs(spec ComponentSpec) {
globalProvider.componentLock.Lock()
switch spec.Type {
case TypeBuffer:
globalProvider.bufferMap[spec.Name] = spec
case TypeCache:
globalProvider.cacheMap[spec.Name] = spec
case TypeInput:
globalProvider.inputMap[spec.Name] = spec
case TypeMetrics:
globalProvider.metricsMap[spec.Name] = spec
case TypeOutput:
globalProvider.outputMap[spec.Name] = spec
case TypeProcessor:
globalProvider.processorMap[spec.Name] = spec
case TypeRateLimit:
globalProvider.rateLimitMap[spec.Name] = spec
case TypeTracer:
globalProvider.tracerMap[spec.Name] = spec
}
globalProvider.componentLock.Unlock()
}
// GetDocs attempts to locate a documentation spec for a component identified by
// a unique name and type combination.
func GetDocs(prov Provider, name string, ctype Type) (ComponentSpec, bool) {
refreshOldPlugins()
if prov == nil {
prov = globalProvider
}
return prov.GetDocs(name, ctype)
}
//------------------------------------------------------------------------------
// 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{},
}
}
// 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
}