Merge pull request #75 from Particular/gatekeeping-backport-3

Fix for plugin not sending metric data to ServiceControl (3.0)
Particular · Nov 20, 2019 · bb2f8a4 · bb2f8a4
2 parents daff04f + 492d6af
commit bb2f8a4
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 7 deletions.
diff --git a/src/NServiceBus.Metrics.ServiceControl/MetricsOptionsExtensions.cs b/src/NServiceBus.Metrics.ServiceControl/MetricsOptionsExtensions.cs
@@ -23,6 +23,8 @@ public static void SendMetricDataToServiceControl(this MetricsOptions options, s
             reporting.ServiceControlMetricsAddress = serviceControlMetricsAddress;
             reporting.ServiceControlReportingInterval = interval;
             reporting.EndpointInstanceIdOverride = instanceId;
+
+            options.RegisterObservers(context => reporting.CreateReporters());
         }
 
         /// <summary>

diff --git a/src/NServiceBus.Metrics.ServiceControl/ReportingFeature.cs b/src/NServiceBus.Metrics.ServiceControl/ReportingFeature.cs
@@ -232,15 +232,18 @@ public ServiceControlRawDataReporting(IBuilder builder, ReportingOptions options
 
             protected override Task OnStart(IMessageSession session)
             {
-                foreach (var metric in metrics)
+                options.OnCreateReporters(() =>
                 {
-                    reporters.Add(CreateReporter(metric.Key, metric.Value.Item1, metric.Value.Item2));
-                }
+                    foreach (var metric in metrics)
+                    {
+                        reporters.Add(CreateReporter(metric.Key, metric.Value.Item1, metric.Value.Item2));
+                    }
 
-                foreach (var reporter in reporters)
-                {
-                    reporter.Start();
-                }
+                    foreach (var reporter in reporters)
+                    {
+                        reporter.Start();
+                    }
+                });                
 
                 return Task.FromResult(0);
             }

diff --git a/src/NServiceBus.Metrics.ServiceControl/ReportingOptions.cs b/src/NServiceBus.Metrics.ServiceControl/ReportingOptions.cs
@@ -10,6 +10,8 @@ class ReportingOptions
         internal TimeSpan ServiceControlReportingInterval;
         internal string EndpointInstanceIdOverride;
         public TimeSpan TimeToBeReceived { get; set; } = TimeSpan.FromDays(7);
+        Action createMetricReporters;
+        bool createReportersCalled;
 
         internal bool TryGetValidEndpointInstanceIdOverride(out string instanceId)
         {
@@ -24,5 +26,32 @@ internal bool TryGetValidEndpointInstanceIdOverride(out string instanceId)
         }
 
         public static ReportingOptions Get(MetricsOptions options) => reporting.GetOrAdd(options, metricsOptions => new ReportingOptions());
+
+        internal void CreateReporters()
+        {
+            if(createReportersCalled)
+            {
+                throw new Exception("CreateReporters has already been called, and can only be called once.");
+            }
+
+            createReportersCalled = true;
+            if(createMetricReporters != null)
+            {
+                createMetricReporters();
+                createMetricReporters = null;
+            }
+        }
+
+        internal void OnCreateReporters(Action createMetricReporters)
+        {
+            if(createReportersCalled)
+            {
+                createMetricReporters();
+            }
+            else
+            {
+                this.createMetricReporters = createMetricReporters;
+            }
+        }
     }
 }