micro/client/options.go

83 lines
1.7 KiB
Go
Raw Normal View History

package client
import (
2015-11-20 16:17:33 +00:00
"github.com/micro/go-micro/broker"
2015-11-27 00:17:36 +00:00
"github.com/micro/go-micro/codec"
2015-11-20 16:17:33 +00:00
"github.com/micro/go-micro/registry"
2015-12-09 19:23:16 +00:00
"github.com/micro/go-micro/selector"
2015-11-20 16:17:33 +00:00
"github.com/micro/go-micro/transport"
)
type options struct {
2015-11-25 19:50:05 +00:00
contentType string
broker broker.Broker
codecs map[string]codec.NewCodec
2015-11-25 19:50:05 +00:00
registry registry.Registry
2015-12-09 19:23:16 +00:00
selector selector.Selector
2015-11-25 19:50:05 +00:00
transport transport.Transport
2015-11-26 20:36:42 +00:00
wrappers []Wrapper
}
2015-12-09 00:02:45 +00:00
type callOptions struct {
2015-12-09 19:23:16 +00:00
selectOptions []selector.SelectOption
2015-12-09 00:02:45 +00:00
}
2015-12-08 19:25:42 +00:00
type publishOptions struct{}
2015-11-26 12:51:53 +00:00
// Broker to be used for pub/sub
func Broker(b broker.Broker) Option {
return func(o *options) {
o.broker = b
}
}
2015-11-26 12:51:53 +00:00
// Codec to be used to encode/decode requests for a given content type
func Codec(contentType string, c codec.NewCodec) Option {
2015-11-25 19:50:05 +00:00
return func(o *options) {
2015-11-27 00:17:36 +00:00
o.codecs[contentType] = c
2015-11-25 19:50:05 +00:00
}
}
2015-11-26 12:51:53 +00:00
// Default content type of the client
2015-11-25 19:50:05 +00:00
func ContentType(ct string) Option {
return func(o *options) {
o.contentType = ct
}
}
2015-11-26 12:51:53 +00: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 12:51:53 +00: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 20:36:42 +00:00
// Select is used to select a node to route a request to
2015-12-09 19:23:16 +00:00
func Selector(s selector.Selector) Option {
return func(o *options) {
o.selector = s
}
}
2015-11-26 20:36:42 +00: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-12-09 00:02:45 +00:00
// Call Options
2015-12-09 19:23:16 +00:00
func WithSelectOption(so selector.SelectOption) CallOption {
2015-12-09 00:02:45 +00:00
return func(o *callOptions) {
o.selectOptions = append(o.selectOptions, so)
}
}