reuse set provider logic

Signed-off-by: Vladimir Petrusevici <V.Petrusevici@maib.md>
open-feature · Oct 27, 2023 · b740910 · b740910
1 parent ad43085
commit b740910
Showing 1 changed file with 3 additions and 16 deletions.
diff --git a/src/OpenFeature/Api.cs b/src/OpenFeature/Api.cs
@@ -41,29 +41,16 @@ private Api() { }
         /// </summary>
         /// <param name="featureProvider">Implementation of <see cref="FeatureProvider"/></param>
         public void SetProvider(FeatureProvider featureProvider)
-        {
-            this._featureProviderLock.EnterWriteLock();
-            try
-            {
-                this._defaultProviderFunc = featureProvider != null ? () => featureProvider : this._defaultProviderFunc;
-            }
-            finally
-            {
-                this._featureProviderLock.ExitWriteLock();
-            }
-        }
+            => this.SetProvider(featureProvider == null ? (Func<FeatureProvider>) null : () => featureProvider);
 
         /// <summary>
         /// Sets the feature provider to given clientName
         /// </summary>
         /// <param name="clientName">Name of client</param>
         /// <param name="featureProvider">Implementation of <see cref="FeatureProvider"/></param>
         public void SetProvider(string clientName, FeatureProvider featureProvider)
-        {
-            FeatureProvider func() => featureProvider;
-            this._featureProviders.AddOrUpdate(clientName, func,
-                (key, current) => func);
-        }
+            => this.SetProvider(clientName, featureProvider == null ? (Func<FeatureProvider>)null : () => featureProvider);
+
 
         /// <summary>
         /// Sets the feature provider