Merge pull request #206 from darren-west/master
Added Options() to registry interface
This commit is contained in:
		| @@ -17,7 +17,7 @@ import ( | ||||
| type consulRegistry struct { | ||||
| 	Address string | ||||
| 	Client  *consul.Client | ||||
| 	Options Options | ||||
| 	opts    Options | ||||
|  | ||||
| 	sync.Mutex | ||||
| 	register map[string]uint64 | ||||
| @@ -94,7 +94,7 @@ func newConsulRegistry(opts ...Option) Registry { | ||||
| 	cr := &consulRegistry{ | ||||
| 		Address:  config.Address, | ||||
| 		Client:   client, | ||||
| 		Options:  options, | ||||
| 		opts:     options, | ||||
| 		register: make(map[string]uint64), | ||||
| 	} | ||||
|  | ||||
| @@ -286,3 +286,7 @@ func (c *consulRegistry) Watch(opts ...WatchOption) (Watcher, error) { | ||||
| func (c *consulRegistry) String() string { | ||||
| 	return "consul" | ||||
| } | ||||
|  | ||||
| func (c *consulRegistry) Options() Options { | ||||
| 	return c.opts | ||||
| } | ||||
|   | ||||
| @@ -322,6 +322,10 @@ func (m *mdnsRegistry) String() string { | ||||
| 	return "mdns" | ||||
| } | ||||
|  | ||||
| func (m *mdnsRegistry) Options() registry.Options { | ||||
| 	return m.opts | ||||
| } | ||||
|  | ||||
| func NewRegistry(opts ...registry.Option) registry.Registry { | ||||
| 	return newRegistry(opts...) | ||||
| } | ||||
|   | ||||
| @@ -99,6 +99,10 @@ func (m *mockRegistry) String() string { | ||||
| 	return "mock" | ||||
| } | ||||
|  | ||||
| func (m *mockRegistry) Options() registry.Options { | ||||
| 	return registry.Options{} | ||||
| } | ||||
|  | ||||
| func NewRegistry() registry.Registry { | ||||
| 	m := &mockRegistry{Services: make(map[string][]*registry.Service)} | ||||
| 	m.init() | ||||
|   | ||||
| @@ -15,6 +15,7 @@ type Registry interface { | ||||
| 	ListServices() ([]*Service, error) | ||||
| 	Watch(...WatchOption) (Watcher, error) | ||||
| 	String() string | ||||
| 	Options() Options | ||||
| } | ||||
|  | ||||
| type Option func(*Options) | ||||
|   | ||||
		Reference in New Issue
	
	Block a user