From 0e30a0d25090089ee1a76686ca9a378bd408940a Mon Sep 17 00:00:00 2001
From: Bas950 <me@bas950.com>
Date: Tue, 24 Sep 2024 13:47:39 +0200
Subject: [PATCH] chore: scan keys instead

---
 apps/api-master/src/tracing.ts | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/apps/api-master/src/tracing.ts b/apps/api-master/src/tracing.ts
index 17df184..a3b4cbc 100644
--- a/apps/api-master/src/tracing.ts
+++ b/apps/api-master/src/tracing.ts
@@ -3,6 +3,8 @@ import { Counter, Gauge, Registry, collectDefaultMetrics } from "prom-client";
 import { updateActivePresenceGauge, updateActivePresenceGaugeLimit } from "./functions/updateActivePresenceGauge.js";
 import { redis } from "./index.js";
 
+const scanCount = Number.parseInt(process.env.SCAN_COUNT || "1000", 10);
+
 export const register = new Registry();
 collectDefaultMetrics({ register });
 
@@ -11,7 +13,13 @@ export const activeSessionsCounter = new Counter({
 	help: "Number of active sessions",
 	async collect() {
 		this.reset();
-		const length = await redis.hlen("pmd-api.sessions");
+		let length = 0;
+		let cursor = "0";
+		do {
+			const reply = await redis.scan(cursor, "MATCH", "pmd-api.sessions.*", "COUNT", scanCount);
+			cursor = reply[0];
+			length += reply[1].length;
+		} while (cursor !== "0");
 		this.inc(length);
 	},
 });