Merge pull request #1641 from Cratis/feature/configureable-polly

Deal with annoying polly logging
Cratis · Jan 10, 2025 · bbdfa5e · bbdfa5e
2 parents 7229cc3 + d5a66fb
commit bbdfa5e
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/Source/Kernel/Setup/ResilientStorageOptions.cs b/Source/Kernel/Setup/ResilientStorageOptions.cs
@@ -3,6 +3,7 @@
 
 using System.ComponentModel.DataAnnotations;
 using Polly.Retry;
+using Polly.Telemetry;
 using Polly.Timeout;
 namespace Cratis.Chronicle.Setup;
 
@@ -22,4 +23,14 @@ public class ResilientStorageOptions
     /// </summary>
     [Required]
     public TimeoutStrategyOptions Timeout { get; set; } = new();
+
+    /// <summary>
+    /// Gets or sets the value indicating whether to enable Polly recording telemetry for each "OnRetry" event.
+    /// </summary>
+#pragma warning disable MA0016
+    public Dictionary<string, ResilienceEventSeverity> ResilienceEventSeverities { get; set; } = new(StringComparer.InvariantCulture)
+#pragma warning restore MA0016
+    {
+        { "OnRetry", ResilienceEventSeverity.Debug }
+    };
 }
diff --git a/Source/Kernel/Setup/StorageProviderExtensions.cs b/Source/Kernel/Setup/StorageProviderExtensions.cs
@@ -8,6 +8,7 @@
 using Orleans.Storage;
 using Polly;
 using Polly.Registry;
+using Polly.Telemetry;
 
 namespace Orleans.Hosting;
 
@@ -33,6 +34,20 @@ public static ISiloBuilder AddStorageProviders(this ISiloBuilder builder)
             var options = context.GetOptions<ResilientStorageOptions>();
             builder.AddRetry(options.Retry);
             builder.AddTimeout(options.Timeout);
+
+            var telemetryOptions = new TelemetryOptions(context.GetOptions<TelemetryOptions>())
+            {
+                SeverityProvider = ev =>
+                {
+                    if (options.ResilienceEventSeverities.TryGetValue(ev.Event.EventName, out var severity))
+                    {
+                        return severity;
+                    }
+
+                    return ev.Event.Severity;
+                }
+            };
+            builder.ConfigureTelemetry(telemetryOptions);
         });
 
         builder.ConfigureServices(services =>