golang: Add ClientFunc option for simple consumer

apache · Oct 30, 2024 · 3c3ac5b · 3c3ac5b
1 parent 33caa01
commit 3c3ac5b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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 {