micro/register/options.go

303 lines
6.4 KiB
Go
Raw Normal View History

package register
2015-12-19 21:28:08 +03:00
import (
2018-03-03 14:53:52 +03:00
"context"
2016-01-17 02:39:47 +03:00
"crypto/tls"
2015-12-19 21:28:08 +03:00
"time"
"go.unistack.org/micro/v4/logger"
"go.unistack.org/micro/v4/meter"
"go.unistack.org/micro/v4/tracer"
2015-12-19 21:28:08 +03:00
)
// Options holds options for register
2015-12-19 21:28:08 +03:00
type Options struct {
// Tracer used for tracing
Tracer tracer.Tracer
// Context holds external options
Context context.Context
// Logged used for logging
Logger logger.Logger
// Meter used for metrics
Meter meter.Meter
// TLSConfig holds tls.TLSConfig options
TLSConfig *tls.Config
// Name holds the name of register
Name string
// Addrs specifies register addrs
Addrs []string
// Timeout specifies timeout
Timeout time.Duration
2015-12-19 21:28:08 +03:00
}
// NewOptions returns options that filled by opts
func NewOptions(opts ...Option) Options {
options := Options{
Logger: logger.DefaultLogger,
Meter: meter.DefaultMeter,
Tracer: tracer.DefaultTracer,
Context: context.Background(),
}
for _, o := range opts {
o(&options)
}
return options
}
// RegisterOptions holds options for register method
type RegisterOptions struct { // nolint: golint,revive
Context context.Context
Domain string
TTL time.Duration
Attempts int
2016-01-27 02:32:27 +03:00
}
// NewRegisterOptions returns register options struct filled by opts
func NewRegisterOptions(opts ...RegisterOption) RegisterOptions {
options := RegisterOptions{
Domain: DefaultDomain,
Context: context.Background(),
}
for _, o := range opts {
o(&options)
}
return options
}
// WatchOptions holds watch options
2018-02-19 20:12:37 +03: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 20:12:37 +03:00
}
// NewWatchOptions returns watch options filled by opts
func NewWatchOptions(opts ...WatchOption) WatchOptions {
options := WatchOptions{
Domain: DefaultDomain,
Context: context.Background(),
}
for _, o := range opts {
o(&options)
}
return options
}
// DeregisterOptions holds options for deregister method
2020-04-14 14:32:59 +03: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 14:32:59 +03:00
}
// NewDeregisterOptions returns options for deregister filled by opts
func NewDeregisterOptions(opts ...DeregisterOption) DeregisterOptions {
options := DeregisterOptions{
Domain: DefaultDomain,
Context: context.Background(),
}
for _, o := range opts {
o(&options)
}
return options
}
// LookupOptions holds lookup options
type LookupOptions struct {
2020-04-14 14:32:59 +03:00
Context context.Context
// Domain to scope the request to
Domain string
2020-04-14 14:32:59 +03:00
}
// NewLookupOptions returns lookup options filled by opts
func NewLookupOptions(opts ...LookupOption) LookupOptions {
options := LookupOptions{
Domain: DefaultDomain,
Context: context.Background(),
}
for _, o := range opts {
o(&options)
}
return options
}
// ListOptions holds the list options for list method
2020-04-14 14:32:59 +03:00
type ListOptions struct {
Context context.Context
// Domain to scope the request to
Domain string
2020-04-14 14:32:59 +03:00
}
// NewListOptions returns list options filled by opts
func NewListOptions(opts ...ListOption) ListOptions {
options := ListOptions{
Domain: DefaultDomain,
Context: context.Background(),
}
for _, o := range opts {
o(&options)
}
return options
}
// Addrs is the register addresses to use
2016-03-16 01:20:21 +03:00
func Addrs(addrs ...string) Option {
return func(o *Options) {
o.Addrs = addrs
}
}
// Timeout sets the timeout
2015-12-19 21:28:08 +03:00
func Timeout(t time.Duration) Option {
return func(o *Options) {
o.Timeout = t
}
}
2016-01-16 23:25:18 +03:00
// Logger sets the logger
func Logger(l logger.Logger) Option {
2016-01-16 23:25:18 +03:00
return func(o *Options) {
o.Logger = l
2016-01-16 23:25:18 +03:00
}
}
2016-01-17 02:39:47 +03:00
// Meter sets the meter
func Meter(m meter.Meter) Option {
return func(o *Options) {
o.Meter = m
}
}
// Tracer sets the tracer
func Tracer(t tracer.Tracer) Option {
return func(o *Options) {
o.Tracer = t
}
}
// Context sets the context
func Context(ctx context.Context) Option {
return func(o *Options) {
o.Context = ctx
}
}
// TLSConfig Specify TLS Config
2016-01-17 02:39:47 +03:00
func TLSConfig(t *tls.Config) Option {
return func(o *Options) {
o.TLSConfig = t
}
}
2016-01-27 02:32:27 +03:00
// RegisterAttempts specifies register atempts count
func RegisterAttempts(t int) RegisterOption { // nolint: golint,revive
return func(o *RegisterOptions) {
o.Attempts = t
}
}
// RegisterTTL specifies register ttl
func RegisterTTL(t time.Duration) RegisterOption { // nolint: golint,revive
2016-01-27 02:32:27 +03:00
return func(o *RegisterOptions) {
o.TTL = t
}
}
2018-02-19 20:12:37 +03:00
// RegisterContext sets the register context
func RegisterContext(ctx context.Context) RegisterOption { // nolint: golint,revive
2020-04-14 14:32:59 +03:00
return func(o *RegisterOptions) {
o.Context = ctx
}
}
// RegisterDomain secifies register domain
func RegisterDomain(d string) RegisterOption { // nolint: golint,revive
return func(o *RegisterOptions) {
o.Domain = d
}
}
// WatchService name
2018-02-19 20:12:37 +03:00
func WatchService(name string) WatchOption {
return func(o *WatchOptions) {
o.Service = name
}
}
2020-04-14 14:32:59 +03:00
// WatchContext sets the context for watch method
2020-04-14 14:32:59 +03:00
func WatchContext(ctx context.Context) WatchOption {
return func(o *WatchOptions) {
o.Context = ctx
}
}
// WatchDomain sets the domain for watch
func WatchDomain(d string) WatchOption {
return func(o *WatchOptions) {
o.Domain = d
}
}
// DeregisterAttempts specifies deregister atempts count
func DeregisterAttempts(t int) DeregisterOption {
return func(o *DeregisterOptions) {
o.Attempts = t
}
}
// DeregisterContext sets the context for deregister method
2020-04-14 14:32:59 +03:00
func DeregisterContext(ctx context.Context) DeregisterOption {
return func(o *DeregisterOptions) {
o.Context = ctx
}
}
// DeregisterDomain specifies deregister domain
func DeregisterDomain(d string) DeregisterOption {
return func(o *DeregisterOptions) {
o.Domain = d
}
}
// LookupContext sets the context for lookup method
func LookupContext(ctx context.Context) LookupOption {
return func(o *LookupOptions) {
2020-04-14 14:32:59 +03:00
o.Context = ctx
}
}
// LookupDomain sets the domain for lookup
func LookupDomain(d string) LookupOption {
return func(o *LookupOptions) {
o.Domain = d
}
}
// ListContext specifies context for list method
2020-04-14 14:32:59 +03:00
func ListContext(ctx context.Context) ListOption {
return func(o *ListOptions) {
o.Context = ctx
}
}
// ListDomain sets the domain for list method
func ListDomain(d string) ListOption {
return func(o *ListOptions) {
o.Domain = d
}
}
// Name sets the name
func Name(n string) Option {
return func(o *Options) {
o.Name = n
}
}