From ddb2776b4e344e8728ed5c4b7cedf5a0400c1d72 Mon Sep 17 00:00:00 2001
From: "ashkan.dehnavi" <ashkan.dehnavi@dotin.ir>
Date: Mon, 6 Jan 2025 10:45:48 +0330
Subject: [PATCH] fix ConfigurationExtensions

---
 .../ZookeeperConfigurationExtensions.cs              | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/DotNet.Extensions.Configuration.Zookeeper/ZookeeperConfigurationExtensions.cs b/src/DotNet.Extensions.Configuration.Zookeeper/ZookeeperConfigurationExtensions.cs
index eb73e0e..298a57a 100644
--- a/src/DotNet.Extensions.Configuration.Zookeeper/ZookeeperConfigurationExtensions.cs
+++ b/src/DotNet.Extensions.Configuration.Zookeeper/ZookeeperConfigurationExtensions.cs
@@ -17,10 +17,10 @@ public static class ZookeeperConfigurationExtensions
     /// <param name="rootPath">the zookeeper node path which you want to read it's sub node as key-value</param>
     /// <param name="timeout">zookeeper session timeout in millsecond</param>
     /// <param name="authInfo">authentication information to access keys on ZK</param>
-    public static void AddZookeeper(this IConfigurationBuilder builder,
+    public static IConfigurationBuilder AddZookeeper(this IConfigurationBuilder builder,
         string connectionString, string rootPath, int timeout, AuthData authInfo = null)
     {
-      AddZookeeper(builder, connectionString, rootPath, timeout, new List<AuthData>() { authInfo });
+      return AddZookeeper(builder, connectionString, rootPath, timeout, new List<AuthData>() { authInfo });
     }
 
     /// <summary>
@@ -31,7 +31,7 @@ public static void AddZookeeper(this IConfigurationBuilder builder,
     /// <param name="rootPath">the zookeeper node path which you want to read it's sub node as key-value</param>
     /// <param name="timeout">zookeeper session timeout in millsecond</param>
     /// <param name="authInfo">authentication information to access keys on ZK</param>
-    public static void AddZookeeper(this IConfigurationBuilder builder,
+    public static IConfigurationBuilder AddZookeeper(this IConfigurationBuilder builder,
     string connectionString, string rootPath, int timeout, List<AuthData> authInfo = null)
     {
       if (builder == null)
@@ -54,6 +54,8 @@ public static void AddZookeeper(this IConfigurationBuilder builder,
         option.SessionTimeout = timeout;
         option.AuthInfo = authInfo;
       });
+
+      return builder;
     }
 
     /// <summary>
@@ -61,7 +63,7 @@ public static void AddZookeeper(this IConfigurationBuilder builder,
     /// </summary>
     /// <param name="builder">The <see cref="IConfigurationBuilder"/> to add to.</param>
     /// <param name="config">configures the zookeeper optoin</param>
-    public static void AddZookeeper(this IConfigurationBuilder builder, Action<ZookeeperOption> config)
+    public static IConfigurationBuilder AddZookeeper(this IConfigurationBuilder builder, Action<ZookeeperOption> config)
     {
       if (builder == null)
       {
@@ -73,6 +75,8 @@ public static void AddZookeeper(this IConfigurationBuilder builder, Action<Zooke
 
       var source = new ZookeeperConfigurationSource() { Option = option };
       builder.Add(source);
+
+      return builder;
     }
   }
 }