server: add ott_balancer_connections metric (#1759)

dyc3 · Apr 29, 2024 · 6c38cf4 · 6c38cf4
1 parent aa795a2
commit 6c38cf4
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/server/balancer.ts b/server/balancer.ts
@@ -12,6 +12,7 @@ import roommanager from "./roommanager";
 import type { RoomListItem } from "./api/room";
 import _ from "lodash";
 import type { MsgB2M, MsgM2B, UnloadReason } from "./generated";
+import { Gauge } from "prom-client";
 export type { MsgB2M, MsgM2B };
 
 const log = getLogger("balancer");
@@ -435,3 +436,11 @@ function gossip() {
 const gossipDebounced = _.debounce(gossip, 1000 * 20, { trailing: true, maxWait: 1000 * 20 });
 
 interface GossipRoom extends RoomListItem {}
+
+const gaugeBalancerConnections = new Gauge({
+	name: "ott_balancer_connections",
+	help: "Number of balancer connections",
+	collect() {
+		this.set(balancerManager.balancerConnections.length);
+	},
+});