Adding transaction propagator to the Api.

Signed-off-by: André Silva <2493377+askpt@users.noreply.github.com>

Author: askpt

src/OpenFeature/Api.cs

@@ -22,6 +22,8 @@ public sealed class Api : IEventBus
         private EventExecutor _eventExecutor = new EventExecutor();
         private ProviderRepository _repository = new ProviderRepository();
         private readonly ConcurrentStack<Hook> _hooks = new ConcurrentStack<Hook>();
+        private ITransactionContextPropagator? _transactionContextPropagator;
+        private object _transactionContextPropagatorLock = new();
 
         /// The reader/writer locks are not disposed because the singleton instance should never be disposed.
         private readonly ReaderWriterLockSlim _evaluationContextLock = new ReaderWriterLockSlim();
@@ -46,7 +48,9 @@ private Api() { }
         public async Task SetProviderAsync(FeatureProvider featureProvider)
         {
             this._eventExecutor.RegisterDefaultFeatureProvider(featureProvider);
-            await this._repository.SetProviderAsync(featureProvider, this.GetContext(), this.AfterInitialization, this.AfterError).ConfigureAwait(false);
+            await this._repository
+                .SetProviderAsync(featureProvider, this.GetContext(), this.AfterInitialization, this.AfterError)
+                .ConfigureAwait(false);
         }
 
         /// <summary>
@@ -61,8 +65,11 @@ public async Task SetProviderAsync(string domain, FeatureProvider featureProvide
             {
                 throw new ArgumentNullException(nameof(domain));
             }
+
             this._eventExecutor.RegisterClientFeatureProvider(domain, featureProvider);
-            await this._repository.SetProviderAsync(domain, featureProvider, this.GetContext(), this.AfterInitialization, this.AfterError).ConfigureAwait(false);
+            await this._repository
+                .SetProviderAsync(domain, featureProvider, this.GetContext(), this.AfterInitialization, this.AfterError)
+                .ConfigureAwait(false);
         }
 
         /// <summary>
@@ -85,7 +92,6 @@ public FeatureProvider GetProvider()
         /// Gets the feature provider with given domain
         /// </summary>
         /// <param name="domain">An identifier which logically binds clients with providers</param>
-
         /// <returns>A provider associated with the given domain, if domain is empty or doesn't
         /// have a corresponding provider the default provider will be returned</returns>
         public FeatureProvider GetProvider(string domain)
@@ -109,7 +115,6 @@ public FeatureProvider GetProvider(string domain)
         /// assigned to it the default provider will be returned
         /// </summary>
         /// <param name="domain">An identifier which logically binds clients with providers</param>
-
         /// <returns>Metadata assigned to provider</returns>
         public Metadata? GetProviderMetadata(string domain) => this.GetProvider(domain).GetMetadata();
 
@@ -218,22 +223,62 @@ public EvaluationContext GetContext()
             }
         }
 
+        /// <summary>
+        /// Return the transaction context propagator.
+        /// </summary>
+        /// <returns><see cref="ITransactionContextPropagator"/>the registered transaction context propagator</returns>
+        public ITransactionContextPropagator? GetTransactionContextPropagator()
+        {
+            return this._transactionContextPropagator;
+        }
+
+        /// <summary>
+        /// Sets the transaction context propagator.
+        /// </summary>
+        /// <param name="transactionContextPropagator">the transaction context propagator to be registered</param>
+        /// <exception cref="ArgumentNullException">Transaction context propagator cannot be null</exception>
+        public void SetTransactionContextPropagator(ITransactionContextPropagator transactionContextPropagator)
+        {
+            if (transactionContextPropagator == null)
+            {
+                throw new ArgumentNullException(nameof(transactionContextPropagator),
+                    "Transaction context propagator cannot be null");
+            }
+
+            lock (this._transactionContextPropagatorLock)
+            {
+                this._transactionContextPropagator = transactionContextPropagator;
+            }
+        }
+
         /// <summary>
         /// Returns the currently defined transaction context using the registered transaction context propagator.
         /// </summary>
         /// <returns><see cref="EvaluationContext"/>The current transaction context</returns>
-        public EvaluationContext GetTransactionContext() {
-            // return this.transactionContextPropagator.getTransactionContext();
-            throw new NotImplementedException();
+        public EvaluationContext? GetTransactionContext()
+        {
+            return this._transactionContextPropagator?.GetTransactionContext();
         }
 
         /// <summary>
         /// Sets the transaction context using the registered transaction context propagator.
         /// </summary>
         /// <param name="evaluationContext">The <see cref="EvaluationContext"/> to set</param>
-        public void SetTransactionContext(EvaluationContext evaluationContext) {
-            // this.transactionContextPropagator.setTransactionContext(evaluationContext);
-            throw new NotImplementedException();
+        /// <exception cref="InvalidOperationException">Transaction context propagator is not set.</exception>
+        /// <exception cref="ArgumentNullException">Evaluation context cannot be null</exception>
+        public void SetTransactionContext(EvaluationContext evaluationContext)
+        {
+            if (evaluationContext == null)
+            {
+                throw new ArgumentNullException(nameof(evaluationContext), "Evaluation context cannot be null");
+            }
+
+            if (this._transactionContextPropagator == null)
+            {
+                throw new InvalidOperationException("Transaction context propagator is not set");
+            }
+
+            this._transactionContextPropagator?.SetTransactionContext(evaluationContext);
         }
 
         /// <summary>
@@ -300,23 +345,27 @@ private async Task AfterInitialization(FeatureProvider provider)
                 ProviderName = provider.GetMetadata()?.Name,
             };
 
-            await this._eventExecutor.EventChannel.Writer.WriteAsync(new Event { Provider = provider, EventPayload = eventPayload }).ConfigureAwait(false);
+            await this._eventExecutor.EventChannel.Writer
+                .WriteAsync(new Event { Provider = provider, EventPayload = eventPayload }).ConfigureAwait(false);
         }
 
         /// <summary>
         /// Update the provider state to ERROR and emit an ERROR after failed init.
         /// </summary>
         private async Task AfterError(FeatureProvider provider, Exception? ex)
         {
-            provider.Status = typeof(ProviderFatalException) == ex?.GetType() ? ProviderStatus.Fatal : ProviderStatus.Error;
+            provider.Status = typeof(ProviderFatalException) == ex?.GetType()
+                ? ProviderStatus.Fatal
+                : ProviderStatus.Error;
             var eventPayload = new ProviderEventPayload
             {
                 Type = ProviderEventTypes.ProviderError,
                 Message = $"Provider initialization error: {ex?.Message}",
                 ProviderName = provider.GetMetadata()?.Name,
             };
 
-            await this._eventExecutor.EventChannel.Writer.WriteAsync(new Event { Provider = provider, EventPayload = eventPayload }).ConfigureAwait(false);
+            await this._eventExecutor.EventChannel.Writer
+                .WriteAsync(new Event { Provider = provider, EventPayload = eventPayload }).ConfigureAwait(false);
         }
     }
 }