Something went wrong. Try again.
[READ-ONLY] Mirror of https://github.com/openstatusHQ/openstatus. 🫖 Status page with uptime monitoring & API monitoring as code 🫖 openstatus.dev
bun drizzle-orm monitoring monitoring-as-code nextjs observability on-call open-source shadcn-ui status-page statuspage synthetic-monitoring tinybird turso uptime uptime-checker uptime-monitor
Something went wrong. Try again.
9.8 kB · 301 lines
TypeScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302import { and, asc, db as defaultDb, desc, eq, gte, inArray, type SQL, sql,} from "@openstatus/db";import { page as pageTable, pageComponent, type PageComponentImpact, selectPageComponentSchema, selectPageSchema, statusReport, statusReportsToPageComponents, statusReportUpdate, statusReportUpdateToPageComponents,} from "@openstatus/db/src/schema";
import { type AttributedUserDetail, loadAttributedUsers } from "../attribution";import { batchReads, type DB, type ServiceContext } from "../context";import type { Page, PageComponent, StatusReport, StatusReportUpdate,} from "../types";import { getReportInWorkspace } from "./internal";import { GetStatusReportInput, ListStatusReportsInput, type StatusReportListPeriod,} from "./schemas";
function periodToSince(period: StatusReportListPeriod): Date { const day = 24 * 60 * 60 * 1000; const now = Date.now(); switch (period) { case "1d": return new Date(now - 1 * day); case "7d": return new Date(now - 7 * day); case "14d": return new Date(now - 14 * day); }}
type Attributed = { createdByUser: AttributedUserDetail | null; updatedByUser: AttributedUserDetail | null;};
export type StatusReportUpdateWithImpacts = StatusReportUpdate & Attributed & { /** Impacts this update set; empty for legacy reports. */ componentImpacts: { pageComponentId: number; impact: PageComponentImpact; }[]; };
export type StatusReportWithRelations = StatusReport & Attributed & { updates: StatusReportUpdateWithImpacts[]; pageComponents: PageComponent[]; /** Flat list of associated component ids. Convenience for proto conversion. */ pageComponentIds: number[]; /** * The owning page with its full component roster. `null` when the report * has no `pageId` — rare today but schema-allowed. */ page: (Page & { pageComponents: PageComponent[] }) | null; };
export type ListStatusReportsResult = { items: StatusReportWithRelations[]; totalSize: number;};
/** * Load relations for a set of status reports in three round-trips regardless * of how many reports were passed in. Avoids the O(N) per-row pattern that * pairs badly with the dashboard's effectively-unlimited list request. */async function enrichReportsBatch( db: DB, rows: StatusReport[],): Promise<StatusReportWithRelations[]> { if (rows.length === 0) return [];
const reportIds = rows.map((r) => r.id); const pageIdsSet = new Set<number>(); for (const r of rows) if (r.pageId != null) pageIdsSet.add(r.pageId); const pageIds = Array.from(pageIdsSet); // Keeps the batch a fixed four statements when no report carries a pageId; // `inArray` with an empty list is not a valid predicate. const anyPage = pageIds.length > 0 ? inArray(pageTable.id, pageIds) : sql`0`; const anyPageComponent = pageIds.length > 0 ? inArray(pageComponent.pageId, pageIds) : sql`0`;
// Everything that keys off the reports themselves — updates, component // associations, owning pages and their component rosters — in one // round-trip. Only the impact rows below depend on a prior result. const [allUpdates, assocRows, pageRows, pageSiblings] = await batchReads(db, [ db .select() .from(statusReportUpdate) .where(inArray(statusReportUpdate.statusReportId, reportIds)) .orderBy(desc(statusReportUpdate.date), desc(statusReportUpdate.id)), // Explicit column selection with aliases avoids depending on drizzle's // auto-derived `row.<object_name>` keys — those are named after the // exported JS variable, so a rename in the schema silently breaks the // row shape at runtime. db .select({ reportId: statusReportsToPageComponents.statusReportId, component: pageComponent, }) .from(pageComponent) .innerJoin( statusReportsToPageComponents, eq(statusReportsToPageComponents.pageComponentId, pageComponent.id), ) .where(inArray(statusReportsToPageComponents.statusReportId, reportIds)), db.select().from(pageTable).where(anyPage), db.select().from(pageComponent).where(anyPageComponent), ]);
// One query: all impact rows for all updates. const updateIds = allUpdates.map((u) => u.id); const impactRows = updateIds.length > 0 ? await db .select({ statusReportUpdateId: statusReportUpdateToPageComponents.statusReportUpdateId, pageComponentId: statusReportUpdateToPageComponents.pageComponentId, impact: statusReportUpdateToPageComponents.impact, }) .from(statusReportUpdateToPageComponents) .where( inArray( statusReportUpdateToPageComponents.statusReportUpdateId, updateIds, ), ) .all() : []; const impactsByUpdate = new Map< number, { pageComponentId: number; impact: PageComponentImpact }[] >(); for (const row of impactRows) { const arr = impactsByUpdate.get(row.statusReportUpdateId); const entry = { pageComponentId: row.pageComponentId, impact: row.impact, }; if (arr) arr.push(entry); else impactsByUpdate.set(row.statusReportUpdateId, [entry]); }
const users = await loadAttributedUsers(db, [ ...rows.flatMap((r) => [r.createdBy, r.updatedBy]), ...allUpdates.flatMap((u) => [u.createdBy, u.updatedBy]), ]); const attributed = (row: { createdBy: number | null; updatedBy: number | null; }): Attributed => ({ createdByUser: row.createdBy != null ? (users.get(row.createdBy) ?? null) : null, updatedByUser: row.updatedBy != null ? (users.get(row.updatedBy) ?? null) : null, });
const updatesByReport = new Map<number, StatusReportUpdateWithImpacts[]>(); for (const u of allUpdates) { const withImpacts = { ...u, ...attributed(u), componentImpacts: impactsByUpdate.get(u.id) ?? [], }; const arr = updatesByReport.get(u.statusReportId); if (arr) arr.push(withImpacts); else updatesByReport.set(u.statusReportId, [withImpacts]); }
const componentsByReport = new Map<number, PageComponent[]>(); for (const row of assocRows) { const component = selectPageComponentSchema.parse(row.component); const arr = componentsByReport.get(row.reportId); if (arr) arr.push(component); else componentsByReport.set(row.reportId, [component]); }
// Keyed by pageId so multiple reports sharing a page share the same Page. const pageById = new Map< number, Page & { pageComponents: PageComponent[] } >(); const siblingsByPageId = new Map<number, PageComponent[]>(); for (const c of pageSiblings) { const parsed = selectPageComponentSchema.parse(c); const arr = siblingsByPageId.get(parsed.pageId); if (arr) arr.push(parsed); else siblingsByPageId.set(parsed.pageId, [parsed]); } for (const p of pageRows) { pageById.set(p.id, { ...selectPageSchema.parse(p), pageComponents: siblingsByPageId.get(p.id) ?? [], }); }
return rows.map((r) => { const components = componentsByReport.get(r.id) ?? []; return { ...r, ...attributed(r), updates: updatesByReport.get(r.id) ?? [], pageComponents: components, pageComponentIds: components.map((c) => c.id), page: r.pageId != null ? (pageById.get(r.pageId) ?? null) : null, }; });}
export async function listStatusReports(args: { ctx: ServiceContext; input: ListStatusReportsInput;}): Promise<ListStatusReportsResult> { const { ctx } = args; const input = ListStatusReportsInput.parse(args.input); const db = ctx.db ?? defaultDb;
const conditions: SQL[] = [eq(statusReport.workspaceId, ctx.workspace.id)]; if (input.statuses.length > 0) { conditions.push(inArray(statusReport.status, input.statuses)); } if (input.pageId !== undefined) { conditions.push(eq(statusReport.pageId, input.pageId)); } if (input.period !== undefined) { conditions.push(gte(statusReport.createdAt, periodToSince(input.period))); } const whereClause = and(...conditions);
const rows = await db .select() .from(statusReport) .where(whereClause) .orderBy( input.order === "asc" ? asc(statusReport.createdAt) : desc(statusReport.createdAt), ) .limit(input.limit) .offset(input.offset) .all();
// A short page is the last page, so the total is already known and the // extra `count(*)` is only paid when a full page comes back — or when an // empty page leaves it ambiguous whether we ran off the end. Both tRPC // consumers discard `totalSize` entirely. let totalSize = input.offset + rows.length; if (rows.length === input.limit || (rows.length === 0 && input.offset > 0)) { const countRow = await db .select({ count: sql<number>`count(*)` }) .from(statusReport) .where(whereClause) .get(); totalSize = countRow?.count ?? totalSize; }
const items = await enrichReportsBatch(db, rows); return { items, totalSize };}
export async function getStatusReport(args: { ctx: ServiceContext; input: GetStatusReportInput;}): Promise<StatusReportWithRelations> { const { ctx } = args; const input = GetStatusReportInput.parse(args.input); const db = ctx.db ?? defaultDb;
const report = await getReportInWorkspace({ tx: db, id: input.id, workspaceId: ctx.workspace.id, }); const [enriched] = await enrichReportsBatch(db, [report]); // `enrichReportsBatch` guarantees a 1:1 mapping for a non-empty input. // oxlint-disable-next-line typescript/no-non-null-assertion -- always non-null for len === 1 return enriched!;}