1
0
mirror of https://github.com/go-micro/go-micro.git synced 2024-12-18 08:26:38 +02:00
go-micro/selector/options.go

71 lines
1.4 KiB
Go
Raw Normal View History

2015-12-09 21:23:16 +02:00
package selector
import (
2018-03-03 13:53:52 +02:00
"context"
"go-micro.dev/v4/logger"
2021-10-12 13:55:53 +02:00
"go-micro.dev/v4/registry"
2015-12-09 21:23:16 +02:00
)
type Options struct {
Registry registry.Registry
2016-05-03 23:06:19 +02:00
Strategy Strategy
// Other options for implementations of the interface
// can be stored in a context
Context context.Context
// Logger is the underline logger
Logger logger.Logger
2015-12-09 21:23:16 +02:00
}
type SelectOptions struct {
2016-05-03 23:06:19 +02:00
Filters []Filter
Strategy Strategy
// Other options for implementations of the interface
// can be stored in a context
Context context.Context
2015-12-09 21:23:16 +02:00
}
// Option used to initialise the selector
type Option func(*Options)
// SelectOption used when making a select call
type SelectOption func(*SelectOptions)
// Registry sets the registry used by the selector
func Registry(r registry.Registry) Option {
return func(o *Options) {
o.Registry = r
}
}
2016-05-03 23:06:19 +02:00
// SetStrategy sets the default strategy for the selector
func SetStrategy(fn Strategy) Option {
return func(o *Options) {
o.Strategy = fn
}
}
2016-04-23 21:15:01 +02:00
// WithFilter adds a filter function to the list of filters
2015-12-09 21:23:16 +02:00
// used during the Select call.
2016-04-23 22:37:26 +02:00
func WithFilter(fn ...Filter) SelectOption {
2015-12-09 21:23:16 +02:00
return func(o *SelectOptions) {
2016-04-23 22:37:26 +02:00
o.Filters = append(o.Filters, fn...)
2015-12-09 21:23:16 +02:00
}
}
2016-05-03 23:06:19 +02:00
// Strategy sets the selector strategy
func WithStrategy(fn Strategy) SelectOption {
return func(o *SelectOptions) {
o.Strategy = fn
}
}
// WithLogger sets the underline logger
func WithLogger(l logger.Logger) Option {
return func(o *Options) {
o.Logger = l
}
}