package transport import ( "context" "crypto/tls" "time" "github.com/unistack-org/micro/v3/codec" "github.com/unistack-org/micro/v3/logger" ) type Options struct { // 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 // Secure tells the transport to secure the connection. // In the case TLSConfig is not specified best effort self-signed // certs should be used Secure bool // 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 // 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, 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 } } // 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 } } // Use secure communication. If TLSConfig is not specified we // use InsecureSkipVerify and generate a self signed cert func Secure(b bool) Option { return func(o *Options) { o.Secure = b } } // 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 } }