micro/client/options.go

319 lines
6.6 KiB
Go
Raw Normal View History

package client
import (
2018-03-03 14:53:52 +03:00
"context"
2016-01-04 00:14:33 +03:00
"time"
2015-11-20 19:17:33 +03:00
"github.com/micro/go-micro/broker"
2015-11-27 03:17:36 +03:00
"github.com/micro/go-micro/codec"
2015-11-20 19:17:33 +03:00
"github.com/micro/go-micro/registry"
2015-12-09 22:23:16 +03:00
"github.com/micro/go-micro/selector"
2015-11-20 19:17:33 +03:00
"github.com/micro/go-micro/transport"
)
type Options struct {
// Used to select codec
ContentType string
// Plugged interfaces
Broker broker.Broker
Codecs map[string]codec.NewCodec
Registry registry.Registry
Selector selector.Selector
Transport transport.Transport
2019-01-13 15:15:13 +03:00
// Router sets the router
Router Router
2016-06-07 02:46:14 +03:00
// Connection Pool
PoolSize int
PoolTTL time.Duration
// Middleware for client
Wrappers []Wrapper
// Default Call Options
CallOptions CallOptions
// Other options for implementations of the interface
// can be stored in a context
Context context.Context
}
type CallOptions struct {
SelectOptions []selector.SelectOption
2018-04-14 18:16:58 +03:00
// Address of remote host
Address string
2016-04-05 22:04:37 +03:00
// Backoff func
Backoff BackoffFunc
// Check if retriable func
2016-11-07 11:40:11 +03:00
Retry RetryFunc
// Transport Dial Timeout
DialTimeout time.Duration
// Number of Call attempts
Retries int
// Request/Response timeout
RequestTimeout time.Duration
2016-11-07 20:49:35 +03:00
// Middleware for low level call func
CallWrappers []CallWrapper
2016-11-07 20:49:35 +03:00
// Other options for implementations of the interface
// can be stored in a context
Context context.Context
}
type PublishOptions struct {
// Other options for implementations of the interface
// can be stored in a context
Context context.Context
2015-12-09 03:02:45 +03:00
}
2015-12-08 22:25:42 +03:00
2018-05-10 19:33:54 +03:00
type MessageOptions struct {
ContentType string
}
type RequestOptions struct {
2018-04-14 20:06:52 +03:00
ContentType string
Stream bool
2015-12-08 22:25:42 +03:00
// Other options for implementations of the interface
// can be stored in a context
Context context.Context
2015-12-17 23:37:35 +03:00
}
2016-01-03 02:16:15 +03:00
func newOptions(options ...Option) Options {
opts := Options{
Codecs: make(map[string]codec.NewCodec),
CallOptions: CallOptions{
2016-11-07 19:46:12 +03:00
Backoff: DefaultBackoff,
Retry: DefaultRetry,
Retries: DefaultRetries,
RequestTimeout: DefaultRequestTimeout,
DialTimeout: transport.DefaultDialTimeout,
},
2016-06-07 02:46:14 +03:00
PoolSize: DefaultPoolSize,
PoolTTL: DefaultPoolTTL,
2016-01-03 02:16:15 +03:00
}
for _, o := range options {
o(&opts)
}
if len(opts.ContentType) == 0 {
opts.ContentType = DefaultContentType
2016-01-03 02:16:15 +03:00
}
if opts.Broker == nil {
opts.Broker = broker.DefaultBroker
}
if opts.Registry == nil {
opts.Registry = registry.DefaultRegistry
}
if opts.Selector == nil {
opts.Selector = selector.NewSelector(
selector.Registry(opts.Registry),
)
}
if opts.Transport == nil {
opts.Transport = transport.DefaultTransport
}
return opts
}
2015-11-26 15:51:53 +03:00
// Broker to be used for pub/sub
func Broker(b broker.Broker) Option {
return func(o *Options) {
o.Broker = b
}
}
2015-11-26 15:51:53 +03:00
// Codec to be used to encode/decode requests for a given content type
func Codec(contentType string, c codec.NewCodec) Option {
return func(o *Options) {
o.Codecs[contentType] = c
2015-11-25 22:50:05 +03:00
}
}
2015-11-26 15:51:53 +03:00
// Default content type of the client
2015-11-25 22:50:05 +03:00
func ContentType(ct string) Option {
return func(o *Options) {
o.ContentType = ct
2015-11-25 22:50:05 +03:00
}
}
2016-06-07 02:46:14 +03:00
// PoolSize sets the connection pool size
func PoolSize(d int) Option {
return func(o *Options) {
o.PoolSize = d
}
}
// PoolSize sets the connection pool size
func PoolTTL(d time.Duration) Option {
return func(o *Options) {
o.PoolTTL = d
}
}
2015-11-26 15:51:53 +03:00
// Registry to find nodes for a given service
func Registry(r registry.Registry) Option {
return func(o *Options) {
o.Registry = r
}
}
2015-11-26 15:51:53 +03:00
// Transport to use for communication e.g http, rabbitmq, etc
func Transport(t transport.Transport) Option {
return func(o *Options) {
o.Transport = t
}
}
2015-11-26 23:36:42 +03:00
// Select is used to select a node to route a request to
2015-12-09 22:23:16 +03:00
func Selector(s selector.Selector) Option {
return func(o *Options) {
o.Selector = s
}
}
2015-11-26 23:36:42 +03:00
// Adds a Wrapper to a list of options passed into the client
func Wrap(w Wrapper) Option {
return func(o *Options) {
o.Wrappers = append(o.Wrappers, w)
2015-11-26 23:36:42 +03:00
}
}
2015-12-09 03:02:45 +03:00
2016-11-07 20:49:35 +03:00
// Adds a Wrapper to the list of CallFunc wrappers
2016-11-07 21:06:15 +03:00
func WrapCall(cw ...CallWrapper) Option {
2016-11-07 20:49:35 +03:00
return func(o *Options) {
o.CallOptions.CallWrappers = append(o.CallOptions.CallWrappers, cw...)
2016-11-07 20:49:35 +03:00
}
}
2016-04-05 22:04:37 +03:00
// Backoff is used to set the backoff function used
// when retrying Calls
func Backoff(fn BackoffFunc) Option {
return func(o *Options) {
o.CallOptions.Backoff = fn
}
}
2016-01-04 00:14:33 +03:00
// Number of retries when making the request.
// Should this be a Call Option?
2016-01-03 02:16:15 +03:00
func Retries(i int) Option {
return func(o *Options) {
o.CallOptions.Retries = i
2016-01-03 02:16:15 +03:00
}
}
2017-04-12 23:47:40 +03:00
// Retry sets the retry function to be used when re-trying.
func Retry(fn RetryFunc) Option {
return func(o *Options) {
o.CallOptions.Retry = fn
}
}
2016-01-04 00:14:33 +03:00
// The request timeout.
// Should this be a Call Option?
func RequestTimeout(d time.Duration) Option {
return func(o *Options) {
o.CallOptions.RequestTimeout = d
2016-01-04 00:14:33 +03:00
}
}
2016-01-04 00:25:03 +03:00
// Transport dial timeout
func DialTimeout(d time.Duration) Option {
return func(o *Options) {
o.CallOptions.DialTimeout = d
2016-01-04 00:25:03 +03:00
}
}
2015-12-09 03:02:45 +03:00
// Call Options
2018-04-14 18:16:58 +03:00
// WithAddress sets the remote address to use rather than using service discovery
func WithAddress(a string) CallOption {
return func(o *CallOptions) {
o.Address = a
}
}
2016-04-23 23:37:26 +03:00
func WithSelectOption(so ...selector.SelectOption) CallOption {
return func(o *CallOptions) {
2016-04-23 23:37:26 +03:00
o.SelectOptions = append(o.SelectOptions, so...)
2015-12-09 03:02:45 +03:00
}
}
2015-12-17 23:37:35 +03:00
// WithCallWrapper is a CallOption which adds to the existing CallFunc wrappers
func WithCallWrapper(cw ...CallWrapper) CallOption {
2016-11-07 20:49:35 +03:00
return func(o *CallOptions) {
o.CallWrappers = append(o.CallWrappers, cw...)
2016-11-07 20:49:35 +03:00
}
}
2016-04-05 22:04:37 +03:00
// WithBackoff is a CallOption which overrides that which
// set in Options.CallOptions
func WithBackoff(fn BackoffFunc) CallOption {
return func(o *CallOptions) {
o.Backoff = fn
}
}
2016-11-07 11:40:11 +03:00
// WithRetry is a CallOption which overrides that which
// set in Options.CallOptions
2016-11-07 11:40:11 +03:00
func WithRetry(fn RetryFunc) CallOption {
return func(o *CallOptions) {
2016-11-07 11:40:11 +03:00
o.Retry = fn
}
}
// WithRetries is a CallOption which overrides that which
// set in Options.CallOptions
func WithRetries(i int) CallOption {
return func(o *CallOptions) {
o.Retries = i
}
}
// WithRequestTimeout is a CallOption which overrides that which
// set in Options.CallOptions
func WithRequestTimeout(d time.Duration) CallOption {
return func(o *CallOptions) {
o.RequestTimeout = d
}
}
// WithDialTimeout is a CallOption which overrides that which
// set in Options.CallOptions
func WithDialTimeout(d time.Duration) CallOption {
return func(o *CallOptions) {
o.DialTimeout = d
}
}
2015-12-17 23:37:35 +03:00
// Request Options
2018-04-14 20:06:52 +03:00
func WithContentType(ct string) RequestOption {
return func(o *RequestOptions) {
o.ContentType = ct
}
}
2015-12-17 23:37:35 +03:00
func StreamingRequest() RequestOption {
return func(o *RequestOptions) {
o.Stream = true
2015-12-17 23:37:35 +03:00
}
}
2019-01-13 15:15:13 +03:00
// WithRouter sets the client router
func WithRouter(r Router) Option {
return func(o *Options) {
o.Router = r
}
}