update zk client

Signed-off-by: Rudro-25 <[email protected]>
kubedb · Aug 9, 2024 · 0e44a89 · 0e44a89
1 parent f17f15a
commit 0e44a89
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/zookeeper/kubedb_client_builder.go b/zookeeper/kubedb_client_builder.go
@@ -35,6 +35,7 @@ const (
 type KubeDBClientBuilder struct {
 	kc      client.Client
 	db      *dbapi.ZooKeeper
+	ctx     context.Context
 	podName string
 	url     string
 }
@@ -56,6 +57,11 @@ func (o *KubeDBClientBuilder) WithURL(url string) *KubeDBClientBuilder {
 	return o
 }
 
+func (o *KubeDBClientBuilder) WithContext(ctx context.Context) *KubeDBClientBuilder {
+	o.ctx = ctx
+	return o
+}
+
 func (o *KubeDBClientBuilder) GetZooKeeperClient(ctx context.Context) (*Client, error) {
 	var err error
 	if o.podName != "" {