Working on this data setup logic

Core/Configurations/ConfigurationManager/KeyValueConfigurationBuilder.cs

@@ -2,55 +2,102 @@ using Newtonsoft.Json;
 using Newtonsoft.Json.Linq;
 using Microsoft.Extensions.Configuration;
 
-namespace Configuration.Core;
-    public class KeyValueConfigurationBuilder
+namespace Configuration.Core
 {
-    private readonly IConfigurationRepository _repository;
-    private readonly JObject _rootObject = new();
-    private ConfigurationReloadToken _reloadToken = new();
-    private IConfiguration _configuration;
-
-    public KeyValueConfigurationBuilder(IConfigurationRepository repository)
+    public class KeyValueConfigurationBuilder
     {
-        _repository = repository;
-    }
+        private readonly IConfigurationRepository _repository;
+        private readonly JObject _rootObject = new();
+        private ConfigurationReloadToken _reloadToken = new();
+        private IConfiguration _configuration;
+        private readonly object _configurationLock = new();
 
-    public async Task LoadConfiguration()
-    {
-        var configurations = await _repository.GetActiveConfigurations();
-        foreach (var config in configurations)
+        public KeyValueConfigurationBuilder(IConfigurationRepository repository)
         {
-            AddKeyValue(config.Key, config.Value);
+            _repository = repository ?? throw new ArgumentNullException(nameof(repository));
         }
-        OnReload();
-    }
 
-    public void AddKeyValue(string key, string jsonValue)
-    {
-        var valueObject = JsonConvert.DeserializeObject<JObject>(jsonValue);
-        var parts = key.Split(':');
-
-        JObject current = _rootObject;
-        for (int i = 0; i < parts.Length - 1; i++)
+        /// <summary>
+        /// Loads configurations from the repository and builds the configuration tree.
+        /// </summary>
+        public async Task LoadConfiguration()
         {
-            var part = parts[i];
-            if (!current.ContainsKey(part))
+            try
             {
-                current[part] = new JObject();
+                var configurations = await _repository.GetActiveConfigurations();
+                foreach (var config in configurations)
+                {
+                    AddKeyValue(config.Key, config.Value);
+                }
+                OnReload();
+            }
+            catch (Exception ex)
+            {
+                // Log the exception or handle it as needed
+                throw new InvalidOperationException("Failed to load configurations.", ex);
             }
-            current = (JObject)current[part];
         }
 
-        current[parts[^1]] = valueObject;
-    }
+        /// <summary>
+        /// Adds a key-value pair to the configuration tree.
+        /// </summary>
+        /// <param name="key">The key to add.</param>
+        /// <param name="jsonValue">The JSON value to add.</param>
+        public void AddKeyValue(string key, string jsonValue)
+        {
+            if (string.IsNullOrEmpty(key))
+                throw new ArgumentNullException(nameof(key));
+            if (string.IsNullOrEmpty(jsonValue))
+                throw new ArgumentNullException(nameof(jsonValue));
 
-    private void OnReload()
-    {
-        var previousToken = Interlocked.Exchange(ref _reloadToken, new ConfigurationReloadToken());
-        previousToken.OnReload();
-        _configuration = null;
-    }
+            try
+            {
+                var valueObject = JsonConvert.DeserializeObject<JObject>(jsonValue);
+                var parts = key.Split(':');
 
-    public IConfiguration Build() =>
-        _configuration ??= new JsonConfiguration(_rootObject, _reloadToken);
+                JObject current = _rootObject;
+                for (int i = 0; i < parts.Length - 1; i++)
+                {
+                    var part = parts[i];
+                    if (!current.ContainsKey(part))
+                    {
+                        current[part] = new JObject();
+                    }
+                    current = (JObject)current[part];
+                }
+
+                current[parts[^1]] = valueObject;
+            }
+            catch (JsonException ex)
+            {
+                throw new ArgumentException("Invalid JSON value.", nameof(jsonValue), ex);
+            }
+        }
+
+        private void OnReload()
+        {
+            var previousToken = Interlocked.Exchange(ref _reloadToken, new ConfigurationReloadToken());
+            previousToken.OnReload();
+            _configuration = null; // Reset the configuration to force a rebuild
+        }
+
+        /// <summary>
+        /// Builds the configuration instance.
+        /// </summary>
+        /// <returns>The built <see cref="IConfiguration"/> instance.</returns>
+        public IConfiguration Build()
+        {
+            if (_configuration == null)
+            {
+                lock (_configurationLock)
+                {
+                    if (_configuration == null)
+                    {
+                        _configuration = new JsonConfiguration(_rootObject, _reloadToken);
+                    }
+                }
+            }
+            return _configuration;
+        }
+    }
 }