155 lines
2.9 KiB
Go
155 lines
2.9 KiB
Go
package server
|
|
|
|
import (
|
|
"github.com/micro/go-micro/broker"
|
|
"github.com/micro/go-micro/codec"
|
|
"github.com/micro/go-micro/registry"
|
|
"github.com/micro/go-micro/transport"
|
|
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
type Options struct {
|
|
Codecs map[string]codec.NewCodec
|
|
Broker broker.Broker
|
|
Registry registry.Registry
|
|
Transport transport.Transport
|
|
Metadata map[string]string
|
|
Name string
|
|
Address string
|
|
Advertise string
|
|
Id string
|
|
Version string
|
|
HdlrWrappers []HandlerWrapper
|
|
SubWrappers []SubscriberWrapper
|
|
|
|
// Other options for implementations of the interface
|
|
// can be stored in a context
|
|
Context context.Context
|
|
}
|
|
|
|
func newOptions(opt ...Option) Options {
|
|
opts := Options{
|
|
Codecs: make(map[string]codec.NewCodec),
|
|
Metadata: map[string]string{},
|
|
}
|
|
|
|
for _, o := range opt {
|
|
o(&opts)
|
|
}
|
|
|
|
if opts.Broker == nil {
|
|
opts.Broker = broker.DefaultBroker
|
|
}
|
|
|
|
if opts.Registry == nil {
|
|
opts.Registry = registry.DefaultRegistry
|
|
}
|
|
|
|
if opts.Transport == nil {
|
|
opts.Transport = transport.DefaultTransport
|
|
}
|
|
|
|
if len(opts.Address) == 0 {
|
|
opts.Address = DefaultAddress
|
|
}
|
|
|
|
if len(opts.Name) == 0 {
|
|
opts.Name = DefaultName
|
|
}
|
|
|
|
if len(opts.Id) == 0 {
|
|
opts.Id = DefaultId
|
|
}
|
|
|
|
if len(opts.Version) == 0 {
|
|
opts.Version = DefaultVersion
|
|
}
|
|
|
|
return opts
|
|
}
|
|
|
|
// Server name
|
|
func Name(n string) Option {
|
|
return func(o *Options) {
|
|
o.Name = n
|
|
}
|
|
}
|
|
|
|
// Unique server id
|
|
func Id(id string) Option {
|
|
return func(o *Options) {
|
|
o.Id = id
|
|
}
|
|
}
|
|
|
|
// Version of the service
|
|
func Version(v string) Option {
|
|
return func(o *Options) {
|
|
o.Version = v
|
|
}
|
|
}
|
|
|
|
// Address to bind to - host:port
|
|
func Address(a string) Option {
|
|
return func(o *Options) {
|
|
o.Address = a
|
|
}
|
|
}
|
|
|
|
// The address to advertise for discovery - host:port
|
|
func Advertise(a string) Option {
|
|
return func(o *Options) {
|
|
o.Advertise = a
|
|
}
|
|
}
|
|
|
|
// Broker to use for pub/sub
|
|
func Broker(b broker.Broker) Option {
|
|
return func(o *Options) {
|
|
o.Broker = b
|
|
}
|
|
}
|
|
|
|
// Codec to use 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
|
|
}
|
|
}
|
|
|
|
// Registry used for discovery
|
|
func Registry(r registry.Registry) Option {
|
|
return func(o *Options) {
|
|
o.Registry = r
|
|
}
|
|
}
|
|
|
|
// Transport mechanism for communication e.g http, rabbitmq, etc
|
|
func Transport(t transport.Transport) Option {
|
|
return func(o *Options) {
|
|
o.Transport = t
|
|
}
|
|
}
|
|
|
|
// Metadata associated with the server
|
|
func Metadata(md map[string]string) Option {
|
|
return func(o *Options) {
|
|
o.Metadata = md
|
|
}
|
|
}
|
|
|
|
// Adds a handler Wrapper to a list of options passed into the server
|
|
func WrapHandler(w HandlerWrapper) Option {
|
|
return func(o *Options) {
|
|
o.HdlrWrappers = append(o.HdlrWrappers, w)
|
|
}
|
|
}
|
|
|
|
// Adds a subscriber Wrapper to a list of options passed into the server
|
|
func WrapSubscriber(w SubscriberWrapper) Option {
|
|
return func(o *Options) {
|
|
o.SubWrappers = append(o.SubWrappers, w)
|
|
}
|
|
}
|