micro/client/options.go

178 lines
3.6 KiB
Go
Raw Normal View History

package client
import (
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"
"golang.org/x/net/context"
)
type Options struct {
2016-01-04 00:14:33 +03:00
ContentType string
Broker broker.Broker
Codecs map[string]codec.NewCodec
Registry registry.Registry
Selector selector.Selector
Transport transport.Transport
Wrappers []Wrapper
Retries int
RequestTimeout time.Duration
2016-01-04 00:25:03 +03:00
DialTimeout time.Duration
// Other options for implementations of the interface
// can be stored in a context
Context context.Context
}
type CallOptions struct {
SelectOptions []selector.SelectOption
// 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
type RequestOptions struct {
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{
2016-01-04 00:14:33 +03:00
Codecs: make(map[string]codec.NewCodec),
Retries: DefaultRetries,
RequestTimeout: DefaultRequestTimeout,
2016-01-04 00:25:03 +03:00
DialTimeout: transport.DefaultDialTimeout,
2016-01-03 02:16:15 +03:00
}
for _, o := range options {
o(&opts)
}
if len(opts.ContentType) == 0 {
opts.ContentType = defaultContentType
}
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
}
}
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-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.Retries = i
}
}
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.RequestTimeout = d
}
}
2016-01-04 00:25:03 +03:00
// Transport dial timeout
func DialTimeout(d time.Duration) Option {
return func(o *Options) {
o.DialTimeout = d
}
}
2015-12-09 03:02:45 +03:00
// Call Options
2015-12-09 22:23:16 +03:00
func WithSelectOption(so selector.SelectOption) CallOption {
return func(o *CallOptions) {
o.SelectOptions = append(o.SelectOptions, so)
2015-12-09 03:02:45 +03:00
}
}
2015-12-17 23:37:35 +03:00
// Request Options
func StreamingRequest() RequestOption {
return func(o *RequestOptions) {
o.Stream = true
2015-12-17 23:37:35 +03:00
}
}