micro/config/cmd/options.go

167 lines
3.6 KiB
Go
Raw Normal View History

2016-01-01 04:16:21 +03:00
package cmd
import (
2018-03-03 14:53:52 +03:00
"context"
"github.com/micro/go-micro/v2/broker"
"github.com/micro/go-micro/v2/client"
"github.com/micro/go-micro/v2/client/selector"
"github.com/micro/go-micro/v2/debug/trace"
"github.com/micro/go-micro/v2/registry"
"github.com/micro/go-micro/v2/runtime"
"github.com/micro/go-micro/v2/server"
"github.com/micro/go-micro/v2/store"
"github.com/micro/go-micro/v2/transport"
2016-01-01 04:16:21 +03:00
)
type Options struct {
2016-01-02 03:38:57 +03:00
// For the Command Line itself
2016-01-01 04:16:21 +03:00
Name string
Description string
Version string
2016-01-02 03:38:57 +03:00
// We need pointers to things so we can swap them out if needed.
Broker *broker.Broker
Registry *registry.Registry
Selector *selector.Selector
Transport *transport.Transport
Client *client.Client
Server *server.Server
Runtime *runtime.Runtime
2020-01-06 20:44:32 +03:00
Store *store.Store
2020-01-29 18:45:11 +03:00
Tracer *trace.Tracer
2016-01-02 03:38:57 +03:00
2016-03-16 01:12:28 +03:00
Brokers map[string]func(...broker.Option) broker.Broker
Clients map[string]func(...client.Option) client.Client
2016-03-16 01:20:21 +03:00
Registries map[string]func(...registry.Option) registry.Registry
2016-01-01 04:16:21 +03:00
Selectors map[string]func(...selector.Option) selector.Selector
Servers map[string]func(...server.Option) server.Server
2016-03-16 01:25:32 +03:00
Transports map[string]func(...transport.Option) transport.Transport
Runtimes map[string]func(...runtime.Option) runtime.Runtime
2020-01-06 20:44:32 +03:00
Stores map[string]func(...store.Option) store.Store
2020-01-29 18:45:11 +03:00
Tracers map[string]func(...trace.Option) trace.Tracer
// Other options for implementations of the interface
// can be stored in a context
Context context.Context
2016-01-01 04:16:21 +03:00
}
2016-01-02 03:38:57 +03:00
// Command line Name
2016-01-01 04:16:21 +03:00
func Name(n string) Option {
return func(o *Options) {
o.Name = n
}
}
2016-01-02 03:38:57 +03:00
// Command line Description
2016-01-01 04:16:21 +03:00
func Description(d string) Option {
return func(o *Options) {
o.Description = d
}
}
2016-01-02 03:38:57 +03:00
// Command line Version
2016-01-01 04:16:21 +03:00
func Version(v string) Option {
return func(o *Options) {
o.Version = v
}
}
2016-01-02 03:38:57 +03:00
func Broker(b *broker.Broker) Option {
return func(o *Options) {
o.Broker = b
}
}
func Selector(s *selector.Selector) Option {
return func(o *Options) {
o.Selector = s
}
}
func Registry(r *registry.Registry) Option {
return func(o *Options) {
o.Registry = r
}
}
func Transport(t *transport.Transport) Option {
return func(o *Options) {
o.Transport = t
}
}
func Client(c *client.Client) Option {
return func(o *Options) {
o.Client = c
}
}
func Server(s *server.Server) Option {
return func(o *Options) {
o.Server = s
}
}
2020-01-29 18:45:11 +03:00
func Tracer(t *trace.Tracer) Option {
return func(o *Options) {
o.Tracer = t
}
}
2016-01-02 03:38:57 +03:00
// New broker func
2016-03-16 01:12:28 +03:00
func NewBroker(name string, b func(...broker.Option) broker.Broker) Option {
2016-01-01 04:16:21 +03:00
return func(o *Options) {
o.Brokers[name] = b
}
}
// New client func
func NewClient(name string, b func(...client.Option) client.Client) Option {
return func(o *Options) {
o.Clients[name] = b
}
}
2016-01-02 03:38:57 +03:00
// New registry func
2016-03-16 01:20:21 +03:00
func NewRegistry(name string, r func(...registry.Option) registry.Registry) Option {
2016-01-01 04:16:21 +03:00
return func(o *Options) {
o.Registries[name] = r
}
}
2016-01-02 03:38:57 +03:00
// New selector func
func NewSelector(name string, s func(...selector.Option) selector.Selector) Option {
2016-01-01 04:16:21 +03:00
return func(o *Options) {
o.Selectors[name] = s
}
}
// New server func
func NewServer(name string, s func(...server.Option) server.Server) Option {
return func(o *Options) {
o.Servers[name] = s
}
}
2016-01-02 03:38:57 +03:00
// New transport func
2016-03-16 01:25:32 +03:00
func NewTransport(name string, t func(...transport.Option) transport.Transport) Option {
2016-01-01 04:16:21 +03:00
return func(o *Options) {
o.Transports[name] = t
}
}
// New runtime func
func NewRuntime(name string, r func(...runtime.Option) runtime.Runtime) Option {
return func(o *Options) {
o.Runtimes[name] = r
}
}
2020-01-29 18:45:11 +03:00
// New tracer func
func NewTracer(name string, t func(...trace.Option) trace.Tracer) Option {
return func(o *Options) {
o.Tracers[name] = t
}
}