HomePhorge
Diffusion Ant (master) Code

ant/docs/reports/src/reports.ts
201d063fe28amaster

Last Change
Raw File

reports.ts
View Options

import { sha256 } from './helpers';
import { drizzle } from 'drizzle-orm/d1';
import { asc, eq, sql } from 'drizzle-orm';
import { generate as generateShortUuid } from 'short-uuid';
import { frames as frameTable, reportFrames, reports, type CrashReport } from './schema';
type ReportDb = ReturnType<typeof drizzle>;
type ReportFrameRow = {
frame: string;
frameHash: string;
frameIndex: number;
};
export type RawReport = {
id: string;
trace: string;
firstSeenAt: string;
lastSeenAt: string;
hitCount: number;
expiresAt: string;
report: CrashReport;
frames: ReportFrameRow[];
};
function dedupeFrame(frame: string): string {
return frame.replace(/\b0x[0-9a-fA-F]+\b/g, '0x');
}
async function reportTrace(report: CrashReport): Promise<string> {
const traceInput = JSON.stringify({
runtime: report.runtime,
version: report.version,
target: report.target,
os: report.os,
arch: report.arch,
code: report.code,
reason: report.reason,
addr: report.addr,
frames: report.frames.map(dedupeFrame),
});
return sha256(traceInput);
}
const reportFromRows = (
row: typeof reports.$inferSelect,
frameRows: ReportFrameRow[],
): CrashReport => ({
schema: 1,
runtime: 'ant',
version: row.version,
target: row.target,
os: row.os,
arch: row.arch,
kind: row.kind,
code: row.code,
reason: row.reason,
addr: row.faultAddress,
elapsedMs: row.elapsedMs,
peakRss: row.peakRss,
frames: frameRows.map(row => row.frame),
});
async function insertReportFrames(db: ReportDb, reportId: string, frames: string[]): Promise<void> {
if (!frames.length) return;
const frameHashes = await Promise.all(frames.map(frame => sha256(frame)));
const uniqueFrames = new Map<string, string>();
frames.forEach((frame, index) => {
uniqueFrames.set(frameHashes[index], frame);
});
await db
.insert(frameTable)
.values([...uniqueFrames].map(([hash, frame]) => ({ hash, frame })))
.onConflictDoNothing();
await db.insert(reportFrames).values(
frameHashes.map((frameHash, index) => ({
frameHash,
reportId,
frameIndex: index,
})),
);
}
async function getReportFrames(db: ReportDb, reportId: string): Promise<ReportFrameRow[]> {
return db
.select({
frame: frameTable.frame,
frameHash: reportFrames.frameHash,
frameIndex: reportFrames.frameIndex,
})
.from(reportFrames)
.innerJoin(frameTable, eq(reportFrames.frameHash, frameTable.hash))
.where(eq(reportFrames.reportId, reportId))
.orderBy(asc(reportFrames.frameIndex));
}
async function getReportRow(db: ReportDb, id: string): Promise<typeof reports.$inferSelect | null> {
const [row] = await db.select().from(reports).where(eq(reports.id, id)).limit(1);
return row ?? null;
}
export async function getReport(db: ReportDb, id: string): Promise<CrashReport | null> {
const row = await getReportRow(db, id);
if (!row) return null;
const frames = await getReportFrames(db, row.id);
return reportFromRows(row, frames);
}
export async function getRawReport(db: ReportDb, id: string): Promise<RawReport | null> {
const row = await getReportRow(db, id);
if (!row) return null;
const frames = await getReportFrames(db, row.id);
return {
id: row.id,
trace: row.trace,
firstSeenAt: row.firstSeenAt,
lastSeenAt: row.lastSeenAt,
hitCount: row.hitCount,
expiresAt: row.expiresAt,
report: reportFromRows(row, frames),
frames,
};
}
export async function createReport(db: ReportDb, report: CrashReport): Promise<string> {
const id = generateShortUuid();
const trace = await reportTrace(report);
const now = new Date();
const expires = new Date(now.getTime() + 30 * 24 * 60 * 60 * 1000);
const nowIso = now.toISOString();
const expiresIso = expires.toISOString();
await db
.insert(reports)
.values({
id,
trace,
hitCount: 1,
runtime: report.runtime,
version: report.version,
kind: report.kind,
reason: report.reason,
code: report.code,
target: report.target,
os: report.os,
arch: report.arch,
faultAddress: report.addr,
elapsedMs: report.elapsedMs,
peakRss: report.peakRss,
firstSeenAt: nowIso,
lastSeenAt: nowIso,
expiresAt: expiresIso,
})
.onConflictDoNothing({ target: reports.trace });
const [row] = await db
.select({
id: reports.id,
hitCount: reports.hitCount,
})
.from(reports)
.where(eq(reports.trace, trace))
.limit(1);
if (!row) return id;
if (row.id === id) {
await insertReportFrames(db, id, report.frames);
return id;
}
await db
.update(reports)
.set({
hitCount: sql`${reports.hitCount} + 1`,
lastSeenAt: nowIso,
expiresAt: expiresIso,
})
.where(eq(reports.id, row.id));
return row.id;
}

Owner Packages

  • No Owners