Add optional port restriction for /metrics endpoint

prom-client-net · Feb 25, 2019 · e6ac027 · e6ac027
1 parent 4233ece
commit e6ac027
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 6 deletions.
diff --git a/src/Prometheus.Client.AspNetCore/PrometheusExtensions.cs b/src/Prometheus.Client.AspNetCore/PrometheusExtensions.cs
@@ -1,7 +1,9 @@
-using System;
+using System;
+using System.Net;
 using System.Threading.Tasks;
 using Prometheus.Client.Collectors;
 using Microsoft.AspNetCore.Builder;
+using Microsoft.AspNetCore.Http;
 
 namespace Prometheus.Client.AspNetCore
 {
@@ -37,6 +39,19 @@ public static IApplicationBuilder UsePrometheusServer(this IApplicationBuilder a
 
             RegisterCollectors(options);
 
+            if (options.Port == null)
+            {
+                MapToRoute(app, options);
+                return app;
+            }
+
+            Func<HttpContext, bool> portMatches = context => context.Connection.LocalPort == options.Port;
+            app.MapWhen(portMatches, coreapp => MapToRoute(coreapp, options));
+            return app;
+        }
+
+        private static void MapToRoute(IApplicationBuilder app, PrometheusOptions options)
+        {
             app.Map(options.MapPath, coreapp =>
             {
                 coreapp.Run(async context =>
@@ -52,17 +67,15 @@ public static IApplicationBuilder UsePrometheusServer(this IApplicationBuilder a
                     await Task.FromResult(0).ConfigureAwait(false);
                 });
             });
-
-            return app;
         }
 
 
         private static void RegisterCollectors(PrometheusOptions options)
         {
             if (options.UseDefaultCollectors)
             {
-                var metricFactory = options.CollectorRegistryInstance == CollectorRegistry.Instance 
-                    ? Metrics.DefaultFactory 
+                var metricFactory = options.CollectorRegistryInstance == CollectorRegistry.Instance
+                    ? Metrics.DefaultFactory
                     : new MetricFactory(options.CollectorRegistryInstance);
 
                 options.Collectors.AddRange(DefaultCollectors.Get(metricFactory));

diff --git a/src/Prometheus.Client.AspNetCore/PrometheusOptions.cs b/src/Prometheus.Client.AspNetCore/PrometheusOptions.cs
@@ -14,6 +14,11 @@ public class PrometheusOptions
         /// </summary>
         public string MapPath { get; set; } = "/metrics";
 
+        /// <summary>
+        ///     When specified only allow access to metrics on this port, otherwise return 404, default = null.
+        /// </summary>
+        public int? Port { get; set; }
+
         /// <summary>
         ///     CollectorRegistry intance
         /// </summary>
@@ -29,4 +34,4 @@ public class PrometheusOptions
         /// </summary>
         public bool UseDefaultCollectors { get; set; } = true;
     }
-}
+}