Wire planner metrics gauges to actual session/client counts

plannerActiveSessions and plannerConnectedClients were registered
but never updated — they always reported 0. Now incremented on
WebSocket connect and decremented on close.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Ullrich Schäfer 2026-04-02 17:02:15 +01:00
parent ca057eb9d4
commit e083ee134d
No known key found for this signature in database
GPG key ID: A32FF691A0F752D9

View file

@ -6,6 +6,7 @@ import * as encoding from "lib0/encoding";
import * as decoding from "lib0/decoding";
import type { IncomingMessage, Server } from "node:http";
import { saveSessionState, loadSessionState, touchSession } from "./sessions.ts";
import { plannerActiveSessions, plannerConnectedClients } from "./metrics.server.ts";
const messageSync = 0;
const messageAwareness = 1;
@ -159,10 +160,13 @@ export function setupYjsWebSocket(server: Server): WebSocketServer {
});
wss.on("connection", async (ws: WebSocket, _request: IncomingMessage, sessionId: string) => {
const isNewSession = !docs.has(sessionId);
const doc = await getOrLoadDoc(sessionId);
const awareness = getAwareness(sessionId, doc);
conns.set(ws, { sessionId, clientIds: new Set() });
plannerConnectedClients.inc();
if (isNewSession) plannerActiveSessions.inc();
// Broadcast doc updates to all connections in this session
const onUpdate = (update: Uint8Array, origin: unknown) => {
@ -218,11 +222,13 @@ export function setupYjsWebSocket(server: Server): WebSocketServer {
);
}
conns.delete(ws);
plannerConnectedClients.dec();
doc.off("update", onUpdate);
// Save when last client leaves
const hasClients = Array.from(conns.values()).some((m) => m.sessionId === sessionId);
if (!hasClients) {
plannerActiveSessions.dec();
saveSessionState(sessionId).catch(() => {});
}
});