micro/registry/mock/mock.go

134 lines
2.5 KiB
Go
Raw Normal View History

2019-01-09 11:31:23 +03:00
// Package mock provides a mock registry for testing
2015-12-09 22:23:16 +03:00
package mock
import (
2019-01-09 09:24:12 +03:00
"sync"
2015-12-09 22:23:16 +03:00
"github.com/micro/go-micro/registry"
)
2016-03-18 01:54:18 +03:00
type mockRegistry struct {
2019-01-09 09:24:12 +03:00
sync.RWMutex
2016-02-26 03:09:06 +03:00
Services map[string][]*registry.Service
}
2015-12-09 22:23:16 +03:00
2016-03-14 22:15:42 +03:00
var (
mockData = map[string][]*registry.Service{
"foo": []*registry.Service{
{
Name: "foo",
Version: "1.0.0",
Nodes: []*registry.Node{
{
Id: "foo-1.0.0-123",
Address: "localhost",
Port: 9999,
},
{
Id: "foo-1.0.0-321",
Address: "localhost",
Port: 9999,
},
2015-12-09 22:23:16 +03:00
},
},
2016-03-14 22:15:42 +03:00
{
Name: "foo",
Version: "1.0.1",
Nodes: []*registry.Node{
{
Id: "foo-1.0.1-321",
Address: "localhost",
Port: 6666,
},
2015-12-09 22:23:16 +03:00
},
},
2016-03-14 22:15:42 +03:00
{
Name: "foo",
Version: "1.0.3",
Nodes: []*registry.Node{
{
Id: "foo-1.0.3-345",
Address: "localhost",
Port: 8888,
},
2015-12-09 22:23:16 +03:00
},
},
},
2016-02-26 03:09:06 +03:00
}
2016-03-14 22:15:42 +03:00
)
2016-03-18 01:54:18 +03:00
func (m *mockRegistry) init() {
2019-01-09 09:24:12 +03:00
m.Lock()
defer m.Unlock()
2016-03-14 22:15:42 +03:00
// add some mock data
m.Services = mockData
2016-02-26 03:09:06 +03:00
}
2016-03-18 01:54:18 +03:00
func (m *mockRegistry) GetService(service string) ([]*registry.Service, error) {
2019-01-09 09:24:12 +03:00
m.Lock()
defer m.Unlock()
2016-02-26 03:09:06 +03:00
s, ok := m.Services[service]
if !ok || len(s) == 0 {
2016-02-26 03:09:06 +03:00
return nil, registry.ErrNotFound
}
return s, nil
2015-12-09 22:23:16 +03:00
}
2016-03-18 01:54:18 +03:00
func (m *mockRegistry) ListServices() ([]*registry.Service, error) {
2019-01-09 09:24:12 +03:00
m.Lock()
defer m.Unlock()
2016-02-26 03:09:06 +03:00
var services []*registry.Service
for _, service := range m.Services {
services = append(services, service...)
}
return services, nil
2015-12-09 22:23:16 +03:00
}
2016-03-18 01:54:18 +03:00
func (m *mockRegistry) Register(s *registry.Service, opts ...registry.RegisterOption) error {
2019-01-09 09:24:12 +03:00
m.Lock()
defer m.Unlock()
2016-03-14 22:15:42 +03:00
services := addServices(m.Services[s.Name], []*registry.Service{s})
m.Services[s.Name] = services
2015-12-09 22:23:16 +03:00
return nil
}
2016-03-18 01:54:18 +03:00
func (m *mockRegistry) Deregister(s *registry.Service) error {
2019-01-09 09:24:12 +03:00
m.Lock()
defer m.Unlock()
2016-03-14 22:15:42 +03:00
services := delServices(m.Services[s.Name], []*registry.Service{s})
m.Services[s.Name] = services
2015-12-09 22:23:16 +03:00
return nil
}
2018-02-19 20:12:37 +03:00
func (m *mockRegistry) Watch(opts ...registry.WatchOption) (registry.Watcher, error) {
var wopts registry.WatchOptions
for _, o := range opts {
o(&wopts)
}
return &mockWatcher{exit: make(chan bool), opts: wopts}, nil
2015-12-09 22:23:16 +03:00
}
2016-03-18 01:54:18 +03:00
func (m *mockRegistry) String() string {
2015-12-20 00:56:14 +03:00
return "mock"
}
func (m *mockRegistry) Init(opts ...registry.Option) error {
return nil
}
2017-09-28 13:16:56 +03:00
func (m *mockRegistry) Options() registry.Options {
return registry.Options{}
}
func NewRegistry(opts ...registry.Options) registry.Registry {
2016-03-18 01:54:18 +03:00
m := &mockRegistry{Services: make(map[string][]*registry.Service)}
2016-02-26 03:09:06 +03:00
m.init()
return m
2015-12-09 22:23:16 +03:00
}