micro/transport/options.go

94 lines
1.9 KiB
Go
Raw Normal View History

2016-01-17 03:28:57 +03:00
package transport
import (
2016-01-18 03:10:04 +03:00
"crypto/tls"
2016-01-17 03:28:57 +03:00
"time"
2016-12-06 21:56:57 +03:00
"github.com/micro/go-micro/transport/codec"
2016-01-17 03:28:57 +03:00
"golang.org/x/net/context"
)
type Options struct {
2016-03-16 01:25:32 +03:00
Addrs []string
2016-12-06 21:56:57 +03:00
Codec codec.Codec
2016-01-18 03:10:04 +03:00
Secure bool
TLSConfig *tls.Config
2016-08-01 18:31:27 +03:00
// Timeout sets the timeout for Send/Recv
Timeout time.Duration
2016-01-17 03:28:57 +03:00
// Other options for implementations of the interface
// can be stored in a context
Context context.Context
}
type DialOptions struct {
Stream bool
Timeout time.Duration
2016-01-18 03:10:04 +03:00
// 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
}
type ListenOptions struct {
// TODO: add tls options when listening
// Currently set in global options
2016-01-17 03:28:57 +03:00
// Other options for implementations of the interface
// can be stored in a context
Context context.Context
}
2016-03-16 01:25:32 +03:00
// Addrs to use for transport
func Addrs(addrs ...string) Option {
return func(o *Options) {
o.Addrs = addrs
}
}
2016-12-06 21:56:57 +03:00
// 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
}
}
2016-08-01 18:31:27 +03:00
// Timeout sets the timeout for Send/Recv execution
func Timeout(t time.Duration) Option {
2016-07-28 20:38:17 +03:00
return func(o *Options) {
2016-08-01 18:31:27 +03:00
o.Timeout = t
2016-07-28 20:38:17 +03:00
}
}
2016-01-18 03:10:04 +03:00
// 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
}
}
// Indicates whether this is a streaming connection
2016-01-17 03:28:57 +03:00
func WithStream() DialOption {
return func(o *DialOptions) {
o.Stream = true
}
}
2016-01-18 03:10:04 +03:00
// Timeout used when dialling the remote side
2016-01-17 03:28:57 +03:00
func WithTimeout(d time.Duration) DialOption {
return func(o *DialOptions) {
o.Timeout = d
}
}