package transport import ( "context" "crypto/tls" "time" "github.com/unistack-org/micro/v3/codec" "github.com/unistack-org/micro/v3/logger" "github.com/unistack-org/micro/v3/meter" "github.com/unistack-org/micro/v3/tracer" ) // Options struct holds the transport options type Options struct { Name string // Addrs is the list of intermediary addresses to connect to Addrs []string // Codec is the codec interface to use where headers are not supported // by the transport and the entire payload must be encoded Codec codec.Codec // TLSConfig to secure the connection. The assumption is that this // is mTLS keypair TLSConfig *tls.Config // Timeout sets the timeout for Send/Recv Timeout time.Duration // Logger sets the logger Logger logger.Logger // Meter sets the meter Meter meter.Meter // Tracer sets the tracer Tracer tracer.Tracer // Other options for implementations of the interface // can be stored in a context Context context.Context } // NewOptions returns new options func NewOptions(opts ...Option) Options { options := Options{ Logger: logger.DefaultLogger, Meter: meter.DefaultMeter, Tracer: tracer.DefaultTracer, Context: context.Background(), } for _, o := range opts { o(&options) } return options } // DialOptions struct type DialOptions struct { // Tells the transport this is a streaming connection with // multiple calls to send/recv and that send may not even be called Stream bool // Timeout for dialing Timeout time.Duration // TODO: add tls options when dialling // Currently set in global options // Other options for implementations of the interface // can be stored in a context Context context.Context } // NewDialOptions returns new DialOptions func NewDialOptions(opts ...DialOption) DialOptions { options := DialOptions{ Timeout: DefaultDialTimeout, Context: context.Background(), } for _, o := range opts { o(&options) } return options } // ListenOptions struct type ListenOptions struct { // TODO: add tls options when listening // Currently set in global options // Other options for implementations of the interface // can be stored in a context Context context.Context } // NewListenOptions returns new ListenOptions func NewListenOptions(opts ...ListenOption) ListenOptions { options := ListenOptions{ Context: context.Background(), } for _, o := range opts { o(&options) } return options } // Addrs to use for transport func Addrs(addrs ...string) Option { return func(o *Options) { o.Addrs = addrs } } // Logger sets the logger func Logger(l logger.Logger) Option { return func(o *Options) { o.Logger = l } } // Meter sets the meter func Meter(m meter.Meter) Option { return func(o *Options) { o.Meter = m } } // Context sets the context func Context(ctx context.Context) Option { return func(o *Options) { o.Context = ctx } } // Codec sets the codec used for encoding where the transport // does not support message headers func Codec(c codec.Codec) Option { return func(o *Options) { o.Codec = c } } // Timeout sets the timeout for Send/Recv execution func Timeout(t time.Duration) Option { return func(o *Options) { o.Timeout = t } } // TLSConfig to be used for the transport. func TLSConfig(t *tls.Config) Option { return func(o *Options) { o.TLSConfig = t } } // WithStream indicates whether this is a streaming connection func WithStream() DialOption { return func(o *DialOptions) { o.Stream = true } } // WithTimeout used when dialling the remote side func WithTimeout(d time.Duration) DialOption { return func(o *DialOptions) { o.Timeout = d } } // Tracer to be used for tracing func Tracer(t tracer.Tracer) Option { return func(o *Options) { o.Tracer = t } } // Name sets the name func Name(n string) Option { return func(o *Options) { o.Name = n } }