/
node_registry.go
52 lines (42 loc) · 1.27 KB
/
node_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
package node
import (
"fmt"
"sort"
"strings"
"k8s.io/apimachinery/pkg/runtime"
"sigs.k8s.io/controller-runtime/pkg/client"
)
type Initializer func(c client.Client, s *runtime.Scheme) Node
type NodeRegistry interface {
Register(provider string, i Initializer)
NewNodeOfProvider(provider string, c client.Client, s *runtime.Scheme) (Node, error)
}
func NewNodeRegistry() NodeRegistry {
return &nodeRegistry{
nodeIndex: map[string]Initializer{},
}
}
// nodeRegistry implementation for fast lookup
type nodeRegistry struct {
nodeIndex map[string]Initializer
}
func (r *nodeRegistry) Register(provider string, i Initializer) {
r.nodeIndex[provider] = i
}
func (r *nodeRegistry) NewNodeOfProvider(provider string, c client.Client, s *runtime.Scheme) (Node, error) {
nodeInitializer, ok := r.nodeIndex[provider]
if !ok {
return nil, fmt.Errorf("provider %q is not supported. supported providers are %q", provider, strings.Join(r.getRegisteredProviderNodeNames(), ", "))
}
// return a new instance of the requested provider node
return nodeInitializer(c, s), nil
}
func (r *nodeRegistry) getRegisteredProviderNodeNames() []string {
var result []string
for k := range r.nodeIndex {
result = append(result, k)
}
// sort and return
sort.Strings(result)
return result
}