micro/server/options.go

156 lines
2.6 KiB
Go
Raw Normal View History

2015-05-27 00:39:48 +03:00
package server
import (
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"
"github.com/micro/go-micro/transport"
2015-05-27 00:39:48 +03:00
)
type options struct {
2015-11-27 03:17:36 +03:00
codecs map[string]codec.Codec
broker broker.Broker
registry registry.Registry
2015-05-27 00:39:48 +03:00
transport transport.Transport
metadata map[string]string
name string
address string
2015-11-11 21:22:04 +03:00
advertise string
2015-05-27 00:39:48 +03:00
id string
version string
2015-05-27 00:39:48 +03:00
}
func newOptions(opt ...Option) options {
2015-11-25 22:50:05 +03:00
opts := options{
2015-11-27 03:17:36 +03:00
codecs: make(map[string]codec.Codec),
2015-11-25 22:50:05 +03:00
}
2015-05-27 00:39:48 +03:00
for _, o := range opt {
o(&opts)
}
if opts.broker == nil {
opts.broker = broker.DefaultBroker
}
if opts.registry == nil {
opts.registry = registry.DefaultRegistry
}
2015-05-27 00:39:48 +03:00
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
}
2015-05-27 00:39:48 +03:00
return opts
}
func (o options) Name() string {
return o.name
}
func (o options) Id() string {
return o.name + "-" + o.id
}
func (o options) Version() string {
return o.version
}
2015-05-27 00:39:48 +03:00
func (o options) Address() string {
return o.address
}
2015-11-11 21:22:04 +03:00
func (o options) Advertise() string {
return o.advertise
}
2015-05-27 00:39:48 +03:00
func (o options) Metadata() map[string]string {
return o.metadata
}
2015-11-26 03:23:36 +03:00
// Server name
func Name(n string) Option {
return func(o *options) {
o.name = n
}
}
2015-11-26 03:23:36 +03:00
// Unique server id
func Id(id string) Option {
return func(o *options) {
o.id = id
}
}
2015-11-26 03:23:36 +03:00
// Version of the service
func Version(v string) Option {
return func(o *options) {
o.version = v
}
}
2015-11-26 03:23:36 +03:00
// Address to bind to - host:port
func Address(a string) Option {
return func(o *options) {
o.address = a
}
}
2015-11-26 03:23:36 +03:00
// The address to advertise for discovery - host:port
2015-11-11 21:22:04 +03:00
func Advertise(a string) Option {
return func(o *options) {
o.advertise = a
}
}
2015-11-26 03:23:36 +03:00
// Broker to use for pub/sub
func Broker(b broker.Broker) Option {
return func(o *options) {
o.broker = b
}
}
2015-11-26 03:23:36 +03:00
// Codec to use to encode/decode requests for a given content type
2015-11-27 03:17:36 +03:00
func Codec(contentType string, c codec.Codec) Option {
2015-11-25 22:50:05 +03:00
return func(o *options) {
2015-11-27 03:17:36 +03:00
o.codecs[contentType] = c
2015-11-25 22:50:05 +03:00
}
}
2015-11-26 03:23:36 +03:00
// Registry used for discovery
func Registry(r registry.Registry) Option {
return func(o *options) {
o.registry = r
}
}
2015-11-26 03:23:36 +03:00
// Transport mechanism for communication e.g http, rabbitmq, etc
func Transport(t transport.Transport) Option {
return func(o *options) {
o.transport = t
}
}
2015-11-26 03:23:36 +03:00
// Metadata associated with the server
func Metadata(md map[string]string) Option {
return func(o *options) {
o.metadata = md
}
}