2016-12-14 15:41:48 +00:00
|
|
|
// Package registry is an interface for service discovery
|
2015-01-13 23:31:27 +00:00
|
|
|
package registry
|
|
|
|
|
2016-02-25 12:42:31 +00:00
|
|
|
import (
|
|
|
|
"errors"
|
|
|
|
)
|
|
|
|
|
2016-01-30 21:13:34 +00:00
|
|
|
// The registry provides an interface for service discovery
|
|
|
|
// and an abstraction over varying implementations
|
|
|
|
// {consul, etcd, zookeeper, ...}
|
2015-01-13 23:31:27 +00:00
|
|
|
type Registry interface {
|
2016-01-26 23:32:27 +00:00
|
|
|
Register(*Service, ...RegisterOption) error
|
2015-05-25 22:14:28 +01:00
|
|
|
Deregister(*Service) error
|
2015-11-08 01:48:48 +00:00
|
|
|
GetService(string) ([]*Service, error)
|
2015-05-25 22:14:28 +01:00
|
|
|
ListServices() ([]*Service, error)
|
2015-06-01 18:55:27 +01:00
|
|
|
Watch() (Watcher, error)
|
2015-12-19 21:56:14 +00:00
|
|
|
String() string
|
2015-06-01 18:55:27 +01:00
|
|
|
}
|
|
|
|
|
2015-12-19 18:28:08 +00:00
|
|
|
type Option func(*Options)
|
2015-05-16 00:34:02 +01:00
|
|
|
|
2016-01-26 23:32:27 +00:00
|
|
|
type RegisterOption func(*RegisterOptions)
|
|
|
|
|
2015-01-13 23:31:27 +00:00
|
|
|
var (
|
2016-03-15 22:20:21 +00:00
|
|
|
DefaultRegistry = newConsulRegistry()
|
2016-02-25 12:42:31 +00:00
|
|
|
|
|
|
|
ErrNotFound = errors.New("not found")
|
2015-01-13 23:31:27 +00:00
|
|
|
)
|
|
|
|
|
2016-03-15 22:20:21 +00:00
|
|
|
func NewRegistry(opts ...Option) Registry {
|
|
|
|
return newConsulRegistry(opts...)
|
2015-05-23 20:04:16 +01:00
|
|
|
}
|
|
|
|
|
2016-01-30 21:13:34 +00:00
|
|
|
// Register a service node. Additionally supply options such as TTL.
|
2016-01-26 23:32:27 +00:00
|
|
|
func Register(s *Service, opts ...RegisterOption) error {
|
|
|
|
return DefaultRegistry.Register(s, opts...)
|
2015-01-13 23:31:27 +00:00
|
|
|
}
|
|
|
|
|
2016-01-30 21:13:34 +00:00
|
|
|
// Deregister a service node
|
2015-05-25 22:14:28 +01:00
|
|
|
func Deregister(s *Service) error {
|
2015-01-13 23:31:27 +00:00
|
|
|
return DefaultRegistry.Deregister(s)
|
|
|
|
}
|
|
|
|
|
2016-01-30 21:13:34 +00:00
|
|
|
// Retrieve a service. A slice is returned since we separate Name/Version.
|
2015-11-08 01:48:48 +00:00
|
|
|
func GetService(name string) ([]*Service, error) {
|
2015-01-13 23:31:27 +00:00
|
|
|
return DefaultRegistry.GetService(name)
|
|
|
|
}
|
2015-04-03 00:52:49 +02:00
|
|
|
|
2016-01-30 21:13:34 +00:00
|
|
|
// List the services. Only returns service names
|
2015-05-25 22:14:28 +01:00
|
|
|
func ListServices() ([]*Service, error) {
|
2015-04-03 00:52:49 +02:00
|
|
|
return DefaultRegistry.ListServices()
|
|
|
|
}
|
2015-12-05 01:12:29 +00:00
|
|
|
|
2016-01-30 21:13:34 +00:00
|
|
|
// Watch returns a watcher which allows you to track updates to the registry.
|
2015-12-05 01:12:29 +00:00
|
|
|
func Watch() (Watcher, error) {
|
|
|
|
return DefaultRegistry.Watch()
|
|
|
|
}
|
2015-12-19 21:56:14 +00:00
|
|
|
|
|
|
|
func String() string {
|
|
|
|
return DefaultRegistry.String()
|
|
|
|
}
|