From ebc278ec98bb24f2852b61fde2a9bf2e3d83818b Mon Sep 17 00:00:00 2001
From: Dave Henderson <dhenderson@gmail.com>
Date: Wed, 30 Dec 2020 13:44:02 -0500
Subject: metrics: allow disabling OpenMetrics negotiation (#3944)

* metrics: allow disabling OpenMetrics negotiation

Signed-off-by: Dave Henderson <dhenderson@gmail.com>

* fixup! metrics: allow disabling OpenMetrics negotiation
---
 modules/metrics/metrics.go | 23 +++++++++++++++++++----
 1 file changed, 19 insertions(+), 4 deletions(-)

(limited to 'modules/metrics/metrics.go')

diff --git a/modules/metrics/metrics.go b/modules/metrics/metrics.go
index eff364e..e6d4345 100644
--- a/modules/metrics/metrics.go
+++ b/modules/metrics/metrics.go
@@ -38,6 +38,10 @@ func init() {
 // unlike AdminMetrics.
 type Metrics struct {
 	metricsHandler http.Handler
+
+	// Disable OpenMetrics negotiation, enabled by default. May be necessary if
+	// the produced metrics cannot be parsed by the service scraping metrics.
+	DisableOpenMetrics bool `json:"disable_openmetrics,omitempty"`
 }
 
 // CaddyModule returns the Caddy module information.
@@ -59,7 +63,7 @@ func (l *zapLogger) Println(v ...interface{}) {
 // Provision sets up m.
 func (m *Metrics) Provision(ctx caddy.Context) error {
 	log := ctx.Logger(m)
-	m.metricsHandler = createMetricsHandler(&zapLogger{log})
+	m.metricsHandler = createMetricsHandler(&zapLogger{log}, !m.DisableOpenMetrics)
 	return nil
 }
 
@@ -71,7 +75,9 @@ func parseCaddyfile(h httpcaddyfile.Helper) (caddyhttp.MiddlewareHandler, error)
 
 // UnmarshalCaddyfile sets up the handler from Caddyfile tokens. Syntax:
 //
-//     metrics <matcher>
+//     metrics [<matcher>] {
+//         disable_openmetrics
+//     }
 //
 func (m *Metrics) UnmarshalCaddyfile(d *caddyfile.Dispenser) error {
 	for d.Next() {
@@ -79,6 +85,15 @@ func (m *Metrics) UnmarshalCaddyfile(d *caddyfile.Dispenser) error {
 		if len(args) > 0 {
 			return d.ArgErr()
 		}
+
+		for d.NextBlock(0) {
+			switch d.Val() {
+			case "disable_openmetrics":
+				m.DisableOpenMetrics = true
+			default:
+				return d.Errf("unrecognized subdirective %q", d.Val())
+			}
+		}
 	}
 	return nil
 }
@@ -95,7 +110,7 @@ var (
 	_ caddyfile.Unmarshaler       = (*Metrics)(nil)
 )
 
-func createMetricsHandler(logger promhttp.Logger) http.Handler {
+func createMetricsHandler(logger promhttp.Logger, enableOpenMetrics bool) http.Handler {
 	return promhttp.InstrumentMetricHandler(prometheus.DefaultRegisterer,
 		promhttp.HandlerFor(prometheus.DefaultGatherer, promhttp.HandlerOpts{
 			// will only log errors if logger is non-nil
@@ -103,7 +118,7 @@ func createMetricsHandler(logger promhttp.Logger) http.Handler {
 
 			// Allow OpenMetrics format to be negotiated - largely compatible,
 			// except quantile/le label values always have a decimal.
-			EnableOpenMetrics: true,
+			EnableOpenMetrics: enableOpenMetrics,
 		}),
 	)
 }
-- 
cgit v1.2.3