From b064f01f97cd91aefb76d539d47cfa14058bb3f7 Mon Sep 17 00:00:00 2001
From: kharhamel <oognic@gmail.com>
Date: Thu, 12 Nov 2020 14:51:19 +0100
Subject: [PATCH] FEAT: added a prometheus gauge for the number of active rooms

---
 back/src/Services/GaugeManager.ts  | 12 ++++++++++++
 back/src/Services/SocketManager.ts |  2 ++
 2 files changed, 14 insertions(+)

diff --git a/back/src/Services/GaugeManager.ts b/back/src/Services/GaugeManager.ts
index f8af822b..80712856 100644
--- a/back/src/Services/GaugeManager.ts
+++ b/back/src/Services/GaugeManager.ts
@@ -6,8 +6,13 @@ class GaugeManager {
     private nbClientsPerRoomGauge: Gauge<string>;
     private nbGroupsPerRoomGauge: Gauge<string>;
     private nbGroupsPerRoomCounter: Counter<string>;
+    private nbRoomsGauge: Gauge<string>;
 
     constructor() {
+        this.nbRoomsGauge = new Gauge({
+            name: 'workadventure_nb_rooms',
+            help: 'Number of active rooms'
+        });
         this.nbClientsGauge = new Gauge({
             name: 'workadventure_nb_sockets',
             help: 'Number of connected sockets',
@@ -31,6 +36,13 @@ class GaugeManager {
         });
     }
 
+    incNbRoomGauge(): void {
+        this.nbRoomsGauge.inc();
+    }
+    decNbRoomGauge(): void {
+        this.nbRoomsGauge.dec();
+    }
+
     incNbClientPerRoomGauge(roomId: string): void {
         this.nbClientsGauge.inc();
         this.nbClientsPerRoomGauge.inc({ room: roomId });
diff --git a/back/src/Services/SocketManager.ts b/back/src/Services/SocketManager.ts
index 4bd26778..97f008c4 100644
--- a/back/src/Services/SocketManager.ts
+++ b/back/src/Services/SocketManager.ts
@@ -351,6 +351,7 @@ export class SocketManager {
                     world.leave(Client);
                     if (world.isEmpty()) {
                         this.Worlds.delete(Client.roomId);
+                        gaugeManager.decNbRoomGauge();
                     }
                 }
                 //user leave previous room
@@ -383,6 +384,7 @@ export class SocketManager {
                 world.tags = data.tags
                 world.policyType = Number(data.policy_type)
             }
+            gaugeManager.incNbRoomGauge();
             this.Worlds.set(roomId, world);
         }
         return Promise.resolve(world)