Made the retry policy for service bus configurable

#2803
SignalR · Jan 28, 2014 · de7fa9b · de7fa9b
1 parent c6e0ead
commit de7fa9b
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/src/Microsoft.AspNet.SignalR.ServiceBus/ServiceBusConnection.cs b/src/Microsoft.AspNet.SignalR.ServiceBus/ServiceBusConnection.cs
@@ -36,13 +36,21 @@ public ServiceBusConnection(ServiceBusScaleoutConfiguration configuration, Trace
             {
                 _namespaceManager = NamespaceManager.CreateFromConnectionString(_connectionString);
                 _factory = MessagingFactory.CreateFromConnectionString(_connectionString);
-                _factory.RetryPolicy = RetryExponential.Default;
             }
             catch (ConfigurationErrorsException)
             {
                 _trace.TraceError("The configured Service Bus connection string contains an invalid property. Check the exception details for more information.");
             }
 
+            if (configuration.RetryPolicy != null)
+            {
+                _factory.RetryPolicy = configuration.RetryPolicy;
+            }
+            else
+            {
+                _factory.RetryPolicy = RetryExponential.Default;
+            }
+
             _idleSubscriptionTimeout = configuration.IdleSubscriptionTimeout;
             _configuration = configuration;
         }

diff --git a/src/Microsoft.AspNet.SignalR.ServiceBus/ServiceBusScaleoutConfiguration.cs b/src/Microsoft.AspNet.SignalR.ServiceBus/ServiceBusScaleoutConfiguration.cs
@@ -107,5 +107,11 @@ public string BuildConnectionString()
 
             return ConnectionString;
         }
+
+        /// <summary>
+        /// Gets or sets the retry policy for service bus
+        /// Default value is RetryExponential.Default
+        /// </summary>
+        public RetryPolicy RetryPolicy { get; set; }
     }
 }