/
registry.go
89 lines (78 loc) · 1.81 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
package multi
import (
"github.com/26597925/EastCloud/pkg/registry"
"github.com/26597925/EastCloud/pkg/server/api"
"golang.org/x/sync/errgroup"
)
type multiRegistry struct {
registries []registry.Registry
}
func New(registries ...registry.Registry) registry.Registry {
return &multiRegistry{
registries: registries,
}
}
func (m *multiRegistry) Register(option api.Option) error {
var eg errgroup.Group
for _, registry := range m.registries {
eg.Go(func() error {
return registry.Register(option)
})
}
return eg.Wait()
}
func (m *multiRegistry) Deregister(sv *registry.Service) error {
var eg errgroup.Group
for _, registry := range m.registries {
eg.Go(func() error {
return registry.Deregister(sv)
})
}
return eg.Wait()
}
func (m *multiRegistry) GetService(name string) (sv []*registry.Service, err error) {
var eg errgroup.Group
for _, registry := range m.registries {
eg.Go(func() error {
sv, err = registry.GetService(name)
return err
})
}
return sv, eg.Wait()
}
func (m *multiRegistry) ListServices() (sv []*registry.Service, err error) {
var eg errgroup.Group
for _, registry := range m.registries {
eg.Go(func() error {
sv, err = registry.ListServices()
return err
})
}
return sv, eg.Wait()
}
func (m *multiRegistry) Watch() (registry.Watcher, error) {
var watchers []registry.Watcher
var eg errgroup.Group
for _, registry := range m.registries {
eg.Go(func() error {
watch, err := registry.Watch()
if err == nil {
watchers = append(watchers, watch)
}
return err
})
}
return NewWatcher(watchers...), eg.Wait()
}
func (m *multiRegistry) Close() error {
var eg errgroup.Group
for _, registry := range m.registries {
eg.Go(func() error {
return registry.Close()
})
}
return eg.Wait()
}
func (m *multiRegistry) Type() string {
return "multi"
}