From 8ec35643c7dd327f2a23bf623a058119e484be39 Mon Sep 17 00:00:00 2001
From: elonj <elonj@spotify.com>
Date: Wed, 6 Jul 2022 16:06:38 -0400
Subject: [PATCH] if non found, metrics should register and retrun created

Signed-off-by: elonj <elonj@spotify.com>
---
 plugins/catalog-backend/src/util/metrics.ts | 38 ++++++++++++++++-----
 1 file changed, 30 insertions(+), 8 deletions(-)

diff --git a/plugins/catalog-backend/src/util/metrics.ts b/plugins/catalog-backend/src/util/metrics.ts
index 4988af82ea..8edc721914 100644
--- a/plugins/catalog-backend/src/util/metrics.ts
+++ b/plugins/catalog-backend/src/util/metrics.ts
@@ -29,27 +29,49 @@ import {
 export function createCounterMetric<T extends string>(
   config: CounterConfiguration<T>,
 ): Counter<T> {
-  const existing = register.getSingleMetric(config.name) as Counter<T>;
-  return existing || new Counter<T>(config);
+  let metric = register.getSingleMetric(config.name);
+  if (!metric) {
+    const newMetric = new Counter<T>(config);
+    register.registerMetric(newMetric);
+    metric = register.getSingleMetric(config.name);
+  }
+  return metric as Counter<T>;
 }
 
 export function createGaugeMetric<T extends string>(
   config: GaugeConfiguration<T>,
 ): Gauge<T> {
-  const existing = register.getSingleMetric(config.name) as Gauge<T>;
-  return existing || new Gauge<T>(config);
+  let metric = register.getSingleMetric(config.name);
+  if (!metric) {
+    const newMetric = new Gauge<T>(config);
+    register.registerMetric(newMetric);
+    metric = register.getSingleMetric(config.name);
+  }
+  return metric as Gauge<T>;
 }
 
 export function createSummaryMetric<T extends string>(
   config: SummaryConfiguration<T>,
 ): Summary<T> {
-  const existing = register.getSingleMetric(config.name) as Summary<T>;
-  return existing || new Summary<T>(config);
+  let metric = register.getSingleMetric(config.name);
+  if (!metric) {
+    const newMetric = new Summary<T>(config);
+    register.registerMetric(newMetric);
+    metric = register.getSingleMetric(config.name);
+  }
+
+  return metric as Summary<T>;
 }
 
 export function createHistogramMetric<T extends string>(
   config: HistogramConfiguration<T>,
 ): Histogram<T> {
-  const existing = register.getSingleMetric(config.name) as Histogram<T>;
-  return existing || new Histogram<T>(config);
+  let metric = register.getSingleMetric(config.name);
+  if (!metric) {
+    const newMetric = new Histogram<T>(config);
+    register.registerMetric(newMetric);
+    metric = register.getSingleMetric(config.name);
+  }
+
+  return metric as Histogram<T>;
 }