51 lines
1.3 KiB
TypeScript
51 lines
1.3 KiB
TypeScript
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);
|