diff --git a/golang/simple_consumer_options.go b/golang/simple_consumer_options.go index 253b3c3db..3db93bf15 100644 --- a/golang/simple_consumer_options.go +++ b/golang/simple_consumer_options.go @@ -85,6 +85,14 @@ func newFuncSimpleConsumerOption(f func(*simpleConsumerOptions)) *funcSimpleCons } } +// WithClientFuncForSimpleConsumer returns a consumerOption that sets ClientFunc for consumer. +// Default is nameserver.New. +func WithClientFuncForSimpleConsumer(f NewClientFunc) SimpleConsumerOption { + return newFuncSimpleConsumerOption(func(o *simpleConsumerOptions) { + o.clientFunc = f + }) +} + // WithTag returns a consumerOption that sets tag for consumer. // Note: Default it uses *. func WithSubscriptionExpressions(subscriptionExpressions map[string]*FilterExpression) SimpleConsumerOption {