prometheus + load-testing

apps/dispatch-server/modules/prometheus.ts

@@ -0,0 +1,50 @@
+import { prisma } from "@repo/db";
+import promClient from "prom-client";
+
+export const promRegister = new promClient.Registry();
+promClient.collectDefaultMetrics({ register: promRegister });
+
+export const socketConnections = new promClient.Gauge({
+	name: "socket_connections",
+	help: "Number of active socket connections",
+	registers: [promRegister],
+});
+
+export const aircraftPatches = new promClient.Counter({
+	name: "aircraft_patches",
+	help: "Counts patch requests for aircrafts",
+	registers: [promRegister],
+});
+
+export const connectedPilots = new promClient.Gauge({
+	name: "connected_pilots",
+	help: "Counts connected pilots",
+	registers: [promRegister],
+	collect: async () => {
+		const count = await prisma.connectedAircraft.count({
+			where: {
+				logoutTime: null,
+			},
+		});
+		connectedPilots.set(count);
+	},
+});
+
+export const connectedDispatcher = new promClient.Gauge({
+	name: "connected_dispatcher",
+	help: "Counts connected dispatchers",
+	registers: [promRegister],
+	collect: async () => {
+		const count = await prisma.connectedDispatcher.count({
+			where: {
+				logoutTime: null,
+			},
+		});
+		connectedDispatcher.set(count);
+	},
+});
+
+promRegister.registerMetric(socketConnections);
+promRegister.registerMetric(aircraftPatches);
+promRegister.registerMetric(connectedPilots);
+promRegister.registerMetric(connectedDispatcher);