2016-01-17 02:28:57 +02:00
|
|
|
package transport
|
|
|
|
|
|
|
|
import (
|
2018-03-03 13:53:52 +02:00
|
|
|
"context"
|
2016-01-18 02:10:04 +02:00
|
|
|
"crypto/tls"
|
2022-07-30 11:56:14 +02:00
|
|
|
"net"
|
2016-01-17 02:28:57 +02:00
|
|
|
"time"
|
|
|
|
|
2021-10-12 13:55:53 +02:00
|
|
|
"go-micro.dev/v4/codec"
|
2016-01-17 02:28:57 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
type Options struct {
|
2019-07-07 13:23:03 +02:00
|
|
|
// 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.Marshaler
|
|
|
|
// 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
|
2016-01-18 02:10:04 +02:00
|
|
|
TLSConfig *tls.Config
|
2016-08-01 17:31:27 +02:00
|
|
|
// Timeout sets the timeout for Send/Recv
|
|
|
|
Timeout time.Duration
|
2016-01-17 02:28:57 +02:00
|
|
|
// Other options for implementations of the interface
|
|
|
|
// can be stored in a context
|
|
|
|
Context context.Context
|
|
|
|
}
|
|
|
|
|
|
|
|
type DialOptions struct {
|
2019-07-07 13:23:03 +02:00
|
|
|
// 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
|
2016-01-17 02:28:57 +02:00
|
|
|
Timeout time.Duration
|
|
|
|
|
2016-01-18 02:10:04 +02: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 02:28:57 +02:00
|
|
|
// Other options for implementations of the interface
|
|
|
|
// can be stored in a context
|
|
|
|
Context context.Context
|
|
|
|
}
|
|
|
|
|
2016-03-16 00:25:32 +02:00
|
|
|
// Addrs to use for transport
|
|
|
|
func Addrs(addrs ...string) Option {
|
|
|
|
return func(o *Options) {
|
|
|
|
o.Addrs = addrs
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-12-06 20:56:57 +02:00
|
|
|
// Codec sets the codec used for encoding where the transport
|
|
|
|
// does not support message headers
|
2019-01-10 11:42:02 +02:00
|
|
|
func Codec(c codec.Marshaler) Option {
|
2016-12-06 20:56:57 +02:00
|
|
|
return func(o *Options) {
|
|
|
|
o.Codec = c
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-08-01 17:31:27 +02:00
|
|
|
// Timeout sets the timeout for Send/Recv execution
|
|
|
|
func Timeout(t time.Duration) Option {
|
2016-07-28 19:38:17 +02:00
|
|
|
return func(o *Options) {
|
2016-08-01 17:31:27 +02:00
|
|
|
o.Timeout = t
|
2016-07-28 19:38:17 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-01-18 02:10:04 +02: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 02:28:57 +02:00
|
|
|
func WithStream() DialOption {
|
|
|
|
return func(o *DialOptions) {
|
|
|
|
o.Stream = true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-01-18 02:10:04 +02:00
|
|
|
// Timeout used when dialling the remote side
|
2016-01-17 02:28:57 +02:00
|
|
|
func WithTimeout(d time.Duration) DialOption {
|
|
|
|
return func(o *DialOptions) {
|
|
|
|
o.Timeout = d
|
|
|
|
}
|
|
|
|
}
|
2022-07-30 11:56:14 +02:00
|
|
|
|
|
|
|
// NetListener Set net.Listener for httpTransport
|
|
|
|
func NetListener(customListener net.Listener) ListenOption {
|
|
|
|
return func(o *ListenOptions) {
|
|
|
|
if customListener != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
if o.Context == nil {
|
|
|
|
o.Context = context.TODO()
|
|
|
|
}
|
|
|
|
o.Context = context.WithValue(o.Context, netListener{}, customListener)
|
|
|
|
}
|
|
|
|
}
|