micro/registry/options.go

257 lines
4.6 KiB
Go
Raw Normal View History

2015-12-19 18:28:08 +00:00
package registry
import (
2018-03-03 11:53:52 +00:00
"context"
2016-01-16 23:39:47 +00:00
"crypto/tls"
2015-12-19 18:28:08 +00:00
"time"
"github.com/unistack-org/micro/v3/logger"
2015-12-19 18:28:08 +00:00
)
type Options struct {
2016-03-15 22:20:21 +00:00
Addrs []string
2016-01-16 23:39:47 +00:00
Timeout time.Duration
Secure bool
TLSConfig *tls.Config
// Logger imp
Logger logger.Logger
// Other options for implementations of the interface
// can be stored in a context
Context context.Context
2015-12-19 18:28:08 +00:00
}
func NewOptions(opts ...Option) Options {
options := Options{
Logger: logger.DefaultLogger,
Context: context.Background(),
}
for _, o := range opts {
o(&options)
}
return options
}
2016-01-26 23:32:27 +00:00
type RegisterOptions struct {
TTL time.Duration
// Other options for implementations of the interface
// can be stored in a context
Context context.Context
// Domain to register the service in
Domain string
// Attempts specify attempts for register
Attempts int
2016-01-26 23:32:27 +00:00
}
func NewRegisterOptions(opts ...RegisterOption) RegisterOptions {
options := RegisterOptions{
Domain: DefaultDomain,
Context: context.Background(),
}
for _, o := range opts {
o(&options)
}
return options
}
2018-02-19 17:12:37 +00:00
type WatchOptions struct {
// Specify a service to watch
// If blank, the watch is for all services
Service string
// Other options for implementations of the interface
// can be stored in a context
Context context.Context
// Domain to watch
Domain string
2018-02-19 17:12:37 +00:00
}
func NewWatchOptions(opts ...WatchOption) WatchOptions {
options := WatchOptions{
Domain: DefaultDomain,
Context: context.Background(),
}
for _, o := range opts {
o(&options)
}
return options
}
2020-04-14 12:32:59 +01:00
type DeregisterOptions struct {
Context context.Context
// Domain the service was registered in
Domain string
// Atempts specify max attempts for deregister
Attempts int
2020-04-14 12:32:59 +01:00
}
func NewDeregisterOptions(opts ...DeregisterOption) DeregisterOptions {
options := DeregisterOptions{
Domain: DefaultDomain,
Context: context.Background(),
}
for _, o := range opts {
o(&options)
}
return options
}
2020-04-14 12:32:59 +01:00
type GetOptions struct {
Context context.Context
// Domain to scope the request to
Domain string
2020-04-14 12:32:59 +01:00
}
func NewGetOptions(opts ...GetOption) GetOptions {
options := GetOptions{
Domain: DefaultDomain,
Context: context.Background(),
}
for _, o := range opts {
o(&options)
}
return options
}
2020-04-14 12:32:59 +01:00
type ListOptions struct {
Context context.Context
// Domain to scope the request to
Domain string
2020-04-14 12:32:59 +01:00
}
func NewListOptions(opts ...ListOption) ListOptions {
options := ListOptions{
Domain: DefaultDomain,
Context: context.Background(),
}
for _, o := range opts {
o(&options)
}
return options
}
2016-03-15 22:20:21 +00:00
// Addrs is the registry addresses to use
func Addrs(addrs ...string) Option {
return func(o *Options) {
o.Addrs = addrs
}
}
2015-12-19 18:28:08 +00:00
func Timeout(t time.Duration) Option {
return func(o *Options) {
o.Timeout = t
}
}
2016-01-16 20:25:18 +00:00
// Secure communication with the registry
func Secure(b bool) Option {
return func(o *Options) {
o.Secure = b
}
}
2016-01-16 23:39:47 +00:00
// Logger sets the logger
func Logger(l logger.Logger) Option {
return func(o *Options) {
o.Logger = l
}
}
// Context sets the context
func Context(ctx context.Context) Option {
return func(o *Options) {
o.Context = ctx
}
}
2016-01-16 23:39:47 +00:00
// Specify TLS Config
func TLSConfig(t *tls.Config) Option {
return func(o *Options) {
o.TLSConfig = t
}
}
2016-01-26 23:32:27 +00:00
func RegisterAttempts(t int) RegisterOption {
return func(o *RegisterOptions) {
o.Attempts = t
}
}
2016-01-27 12:23:18 +00:00
func RegisterTTL(t time.Duration) RegisterOption {
2016-01-26 23:32:27 +00:00
return func(o *RegisterOptions) {
o.TTL = t
}
}
2018-02-19 17:12:37 +00:00
2020-04-14 12:32:59 +01:00
func RegisterContext(ctx context.Context) RegisterOption {
return func(o *RegisterOptions) {
o.Context = ctx
}
}
func RegisterDomain(d string) RegisterOption {
return func(o *RegisterOptions) {
o.Domain = d
}
}
2018-02-19 17:12:37 +00:00
// Watch a service
func WatchService(name string) WatchOption {
return func(o *WatchOptions) {
o.Service = name
}
}
2020-04-14 12:32:59 +01:00
func WatchContext(ctx context.Context) WatchOption {
return func(o *WatchOptions) {
o.Context = ctx
}
}
func WatchDomain(d string) WatchOption {
return func(o *WatchOptions) {
o.Domain = d
}
}
func DeregisterTimeout(t int) DeregisterOption {
return func(o *DeregisterOptions) {
o.Attempts = t
}
}
2020-04-14 12:32:59 +01:00
func DeregisterContext(ctx context.Context) DeregisterOption {
return func(o *DeregisterOptions) {
o.Context = ctx
}
}
func DeregisterDomain(d string) DeregisterOption {
return func(o *DeregisterOptions) {
o.Domain = d
}
}
2020-04-14 12:32:59 +01:00
func GetContext(ctx context.Context) GetOption {
return func(o *GetOptions) {
o.Context = ctx
}
}
func GetDomain(d string) GetOption {
return func(o *GetOptions) {
o.Domain = d
}
}
2020-04-14 12:32:59 +01:00
func ListContext(ctx context.Context) ListOption {
return func(o *ListOptions) {
o.Context = ctx
}
}
func ListDomain(d string) ListOption {
return func(o *ListOptions) {
o.Domain = d
}
}