From 8cda5cd986f289c6219ad4c9e6ce95d29754cf4b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gast=C3=B3n=20Fournier?= <gaston@getunleash.io>
Date: Tue, 15 Oct 2024 11:38:05 +0200
Subject: [PATCH] Migrate feature_toggles_total

---
 src/lib/metrics.ts | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/lib/metrics.ts b/src/lib/metrics.ts
index 64e0caac2b..429f3478f4 100644
--- a/src/lib/metrics.ts
+++ b/src/lib/metrics.ts
@@ -117,10 +117,12 @@ export default class MetricsMonitor {
             help: 'Number of times a feature flag has been used',
             labelNames: ['toggle', 'active', 'appName'],
         });
-        const featureFlagsTotal = createGauge({
+        dbMetrics.registerGaugeDbMetric({
             name: 'feature_toggles_total',
             help: 'Number of feature flags',
             labelNames: ['version'],
+            query: () => instanceStatsService.getToggleCount(),
+            map: (count) => ({ count, labels: { version } }),
         });
         dbMetrics.registerGaugeDbMetric({
             name: 'max_feature_environment_strategies',
@@ -446,9 +448,6 @@ export default class MetricsMonitor {
                         : Promise.resolve([]),
                 ]);
 
-                featureFlagsTotal.reset();
-                featureFlagsTotal.labels({ version }).set(stats.featureToggles);
-
                 featureTogglesArchivedTotal.reset();
                 featureTogglesArchivedTotal.set(stats.archivedFeatureToggles);