2019-06-03 18:44:43 +01:00
|
|
|
package resolver
|
|
|
|
|
2020-04-09 11:03:33 +01:00
|
|
|
import (
|
2020-08-19 17:47:17 +03:00
|
|
|
"github.com/unistack-org/micro/v3/registry"
|
2020-04-09 11:03:33 +01:00
|
|
|
)
|
|
|
|
|
2020-11-03 01:08:23 +03:00
|
|
|
// Options struct
|
2020-06-26 14:28:18 +01:00
|
|
|
type Options struct {
|
|
|
|
Handler string
|
|
|
|
ServicePrefix string
|
2019-06-03 18:44:43 +01:00
|
|
|
}
|
|
|
|
|
2020-11-03 01:08:23 +03:00
|
|
|
// Option func
|
2020-06-26 14:28:18 +01:00
|
|
|
type Option func(o *Options)
|
|
|
|
|
2019-06-03 18:44:43 +01:00
|
|
|
// WithHandler sets the handler being used
|
|
|
|
func WithHandler(h string) Option {
|
|
|
|
return func(o *Options) {
|
|
|
|
o.Handler = h
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-06-26 14:28:18 +01:00
|
|
|
// WithServicePrefix sets the ServicePrefix option
|
|
|
|
func WithServicePrefix(p string) Option {
|
2019-06-03 18:44:43 +01:00
|
|
|
return func(o *Options) {
|
2020-06-26 14:28:18 +01:00
|
|
|
o.ServicePrefix = p
|
2019-06-03 18:44:43 +01:00
|
|
|
}
|
|
|
|
}
|
2020-06-26 14:28:18 +01:00
|
|
|
|
|
|
|
// NewOptions returns new initialised options
|
|
|
|
func NewOptions(opts ...Option) Options {
|
2020-11-03 01:08:23 +03:00
|
|
|
options := Options{}
|
2020-06-26 14:28:18 +01:00
|
|
|
for _, o := range opts {
|
|
|
|
o(&options)
|
|
|
|
}
|
2020-06-29 16:37:45 +01:00
|
|
|
return options
|
|
|
|
}
|
|
|
|
|
|
|
|
// ResolveOptions are used when resolving a request
|
|
|
|
type ResolveOptions struct {
|
|
|
|
Domain string
|
|
|
|
}
|
|
|
|
|
|
|
|
// ResolveOption sets an option
|
|
|
|
type ResolveOption func(*ResolveOptions)
|
|
|
|
|
|
|
|
// Domain sets the resolve Domain option
|
|
|
|
func Domain(n string) ResolveOption {
|
|
|
|
return func(o *ResolveOptions) {
|
|
|
|
o.Domain = n
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// NewResolveOptions returns new initialised resolve options
|
|
|
|
func NewResolveOptions(opts ...ResolveOption) ResolveOptions {
|
2020-11-03 01:08:23 +03:00
|
|
|
options := ResolveOptions{Domain: registry.DefaultDomain}
|
2020-06-29 16:37:45 +01:00
|
|
|
for _, o := range opts {
|
|
|
|
o(&options)
|
|
|
|
}
|
2020-06-26 14:28:18 +01:00
|
|
|
|
|
|
|
return options
|
|
|
|
}
|