feat(protocol): allow override of dial func

This commit is contained in:
2024-08-05 18:10:19 +02:00
parent b976bde363
commit 6842d4d88a
16 changed files with 158 additions and 18 deletions

View File

@ -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
}
}