micro/api/router/options.go

76 lines
1.4 KiB
Go
Raw Normal View History

2019-06-03 20:44:43 +03:00
package router
import (
"context"
"go.unistack.org/micro/v3/api/resolver"
"go.unistack.org/micro/v3/api/resolver/vpath"
"go.unistack.org/micro/v3/logger"
"go.unistack.org/micro/v3/register"
2019-06-03 20:44:43 +03:00
)
// Options holds the options for api router
2019-06-03 20:44:43 +03:00
type Options struct {
// Register for service lookup
Register register.Register
// Resolver to use
2020-04-08 17:39:01 +03:00
Resolver resolver.Resolver
// Logger micro logger
Logger logger.Logger
// Context is for external options
Context context.Context
// Handler name
Handler string
2019-06-03 20:44:43 +03:00
}
// Option func signature
2019-06-03 20:44:43 +03:00
type Option func(o *Options)
// NewOptions returns options struct filled by opts
2019-06-03 20:44:43 +03:00
func NewOptions(opts ...Option) Options {
options := Options{
Context: context.Background(),
Handler: "meta",
2019-06-03 20:44:43 +03:00
}
for _, o := range opts {
o(&options)
}
2020-04-09 12:34:21 +03:00
if options.Resolver == nil {
options.Resolver = vpath.NewResolver(
resolver.WithHandler(options.Handler),
)
}
2019-06-03 20:44:43 +03:00
return options
}
// WithContext sets the context
func WithContext(ctx context.Context) Option {
return func(o *Options) {
o.Context = ctx
}
}
// WithHandler sets the handler
2019-06-03 20:44:43 +03:00
func WithHandler(h string) Option {
return func(o *Options) {
o.Handler = h
}
}
// WithRegister sets the register
func WithRegister(r register.Register) Option {
2019-06-03 20:44:43 +03:00
return func(o *Options) {
o.Register = r
2019-06-03 20:44:43 +03:00
}
}
// WithResolver sets the resolver
2019-06-03 20:44:43 +03:00
func WithResolver(r resolver.Resolver) Option {
return func(o *Options) {
o.Resolver = r
}
}