Comprehensive admin + user dashboards (production-ready)

lib/queue/health.ts

@@ -0,0 +1,74 @@
+import { prisma } from "@/lib/db";
+
+const STALE_MS = 60_000;
+
+/** Worker calls this on an interval so the admin can tell it's alive. */
+export async function recordHeartbeat(
+  name: string,
+  stats?: { queued?: number; running?: number }
+): Promise<void> {
+  const now = new Date();
+  await prisma.workerHeartbeat.upsert({
+    where: { name },
+    create: { name, lastBeatAt: now, queued: stats?.queued, running: stats?.running },
+    update: { lastBeatAt: now, queued: stats?.queued, running: stats?.running },
+  });
+}
+
+export interface WorkerHealth {
+  name: string;
+  alive: boolean;
+  lastBeatAt: Date;
+  secondsAgo: number;
+  queued: number | null;
+  running: number | null;
+}
+
+export async function getWorkerHealth(): Promise<WorkerHealth[]> {
+  const rows = await prisma.workerHeartbeat.findMany({ orderBy: { name: "asc" } });
+  const now = Date.now();
+  return rows.map((r) => ({
+    name: r.name,
+    alive: now - r.lastBeatAt.getTime() < STALE_MS,
+    lastBeatAt: r.lastBeatAt,
+    secondsAgo: Math.round((now - r.lastBeatAt.getTime()) / 1000),
+    queued: r.queued,
+    running: r.running,
+  }));
+}
+
+export interface QueueStat {
+  queue: string;
+  queued: number;
+  active: number;
+  completed: number;
+  retry: number;
+  failed: number;
+}
+
+/**
+ * Per-queue job counts straight from pg-boss's own tables (same Postgres).
+ * Returns [] if the pgboss schema isn't reachable (e.g. worker never started).
+ */
+export async function getQueueStats(): Promise<QueueStat[]> {
+  try {
+    const rows = await prisma.$queryRawUnsafe<{ name: string; state: string; n: number }[]>(
+      `SELECT name, state, count(*)::int AS n FROM pgboss.job GROUP BY name, state`
+    );
+    const map = new Map<string, QueueStat>();
+    for (const r of rows) {
+      const stat =
+        map.get(r.name) ??
+        ({ queue: r.name, queued: 0, active: 0, completed: 0, retry: 0, failed: 0 } as QueueStat);
+      if (r.state === "created") stat.queued += r.n;
+      else if (r.state === "active") stat.active += r.n;
+      else if (r.state === "completed") stat.completed += r.n;
+      else if (r.state === "retry") stat.retry += r.n;
+      else if (r.state === "failed") stat.failed += r.n;
+      map.set(r.name, stat);
+    }
+    return Array.from(map.values()).sort((a, b) => a.queue.localeCompare(b.queue));
+  } catch {
+    return [];
+  }
+}