feat(protocol): allow override of dial func
This commit is contained in:
@ -16,6 +16,7 @@ type Options struct {
|
||||
FallbackProtocol protocol.Identifier
|
||||
AvailableTimeout time.Duration
|
||||
Logger logger.Logger
|
||||
Dial protocol.DialFunc
|
||||
}
|
||||
|
||||
type OptionFunc func(opts *Options)
|
||||
@ -27,6 +28,7 @@ func NewOptions(funcs ...OptionFunc) *Options {
|
||||
Protocols: protocol.DefaultRegistry(),
|
||||
AvailableTimeout: 5 * time.Second,
|
||||
Logger: slog.Default(),
|
||||
Dial: protocol.DefaultDialFunc,
|
||||
}
|
||||
|
||||
for _, fn := range funcs {
|
||||
@ -65,3 +67,9 @@ func WithAvailableTimeout(timeout time.Duration) OptionFunc {
|
||||
opts.AvailableTimeout = timeout
|
||||
}
|
||||
}
|
||||
|
||||
func WithDial(dial protocol.DialFunc) OptionFunc {
|
||||
return func(opts *Options) {
|
||||
opts.Dial = dial
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user