114 lines
3.6 KiB
TypeScript
114 lines
3.6 KiB
TypeScript
import { writeFileSync } from "node:fs";
|
|
import { resolve } from "node:path";
|
|
import { AppEnv } from "@autumn/shared";
|
|
import { sql } from "drizzle-orm";
|
|
import {
|
|
initDrizzle,
|
|
prodTestCustomerId,
|
|
prodTestEntityId,
|
|
prodTestOrgId,
|
|
} from "./experimentEnv";
|
|
|
|
const { getFullSubjectQuery } = await import(
|
|
"../src/internal/customers/repos/getFullSubject/getFullSubjectQuery"
|
|
);
|
|
const { RELEVANT_STATUSES } = await import(
|
|
"../src/internal/customers/cusProducts/CusProductService"
|
|
);
|
|
|
|
// Run with:
|
|
// bun run experiments/explainGetFullSubject.ts
|
|
// Or scoped to an entity:
|
|
// PROD_TEST_ENTITY_ID=... bun run experiments/explainGetFullSubject.ts
|
|
|
|
const main = async () => {
|
|
const orgId = prodTestOrgId;
|
|
const env = AppEnv.Live;
|
|
const customerId = prodTestCustomerId;
|
|
const entityId = prodTestEntityId;
|
|
|
|
const { db } = initDrizzle();
|
|
|
|
const query = getFullSubjectQuery({
|
|
orgId,
|
|
env,
|
|
customerId,
|
|
entityId,
|
|
inStatuses: RELEVANT_STATUSES,
|
|
});
|
|
|
|
console.log(
|
|
`--- Running full subject query (customer=${customerId}${entityId ? `, entity=${entityId}` : ""}) ---`,
|
|
);
|
|
const start = performance.now();
|
|
const result = await db.execute(query);
|
|
const elapsed = performance.now() - start;
|
|
console.log(`Rows returned: ${result.length}`);
|
|
console.log(`Wall-clock time: ${elapsed.toFixed(2)}ms`);
|
|
|
|
// The full row payload is enormous — dump it to a file and print a summary.
|
|
const outDir = resolve(import.meta.dir, "out");
|
|
try {
|
|
const { mkdirSync } = await import("node:fs");
|
|
mkdirSync(outDir, { recursive: true });
|
|
} catch {
|
|
// ignore
|
|
}
|
|
const resultPath = resolve(outDir, "getFullSubject-result.json");
|
|
writeFileSync(resultPath, JSON.stringify(result, null, 2));
|
|
console.log(`Result written to: ${resultPath}`);
|
|
|
|
const row0 = (result[0] ?? {}) as Record<string, unknown>;
|
|
const summarize = (value: unknown): unknown => {
|
|
if (Array.isArray(value)) return `Array(len=${value.length})`;
|
|
if (value && typeof value === "object")
|
|
return `Object(keys=${Object.keys(value).length})`;
|
|
return value;
|
|
};
|
|
console.log("\n--- Top-level fields (summary) ---");
|
|
for (const [key, value] of Object.entries(row0)) {
|
|
console.log(` ${key}: ${summarize(value)}`);
|
|
}
|
|
|
|
const aggregated = (row0.aggregated_customer_entitlements ?? []) as Array<
|
|
Record<string, unknown>
|
|
>;
|
|
if (aggregated.length > 0) {
|
|
console.log(
|
|
`\n--- aggregated_customer_entitlements (${aggregated.length}) ---`,
|
|
);
|
|
for (const ae of aggregated) {
|
|
const entities = (ae.entities ?? {}) as Record<string, unknown>;
|
|
console.log(
|
|
` ${ae.feature_id}: balance=${ae.balance}, adj=${ae.adjustment}, add=${ae.additional_balance}, rollover=${ae.rollover_balance}, entities.keys=${Object.keys(entities).length}`,
|
|
);
|
|
}
|
|
}
|
|
console.log();
|
|
|
|
console.log("--- EXPLAIN (ANALYZE, BUFFERS) ---\n");
|
|
const explainQuery = sql`EXPLAIN (ANALYZE, BUFFERS, FORMAT TEXT) ${query}`;
|
|
const explainResult = await db.execute(explainQuery);
|
|
|
|
const explainPath = resolve(outDir, "getFullSubject-explain.txt");
|
|
const explainLines: string[] = [];
|
|
for (const row of explainResult) {
|
|
const line = (row as Record<string, unknown>)["QUERY PLAN"];
|
|
if (typeof line === "string") explainLines.push(line);
|
|
}
|
|
writeFileSync(explainPath, explainLines.join("\n"));
|
|
console.log(`EXPLAIN written to: ${explainPath}`);
|
|
|
|
// Print just the top of the plan (total time, first ~20 lines) inline
|
|
const head = explainLines.slice(0, 20);
|
|
for (const line of head) console.log(line);
|
|
const execLine = explainLines.find((line) => /Execution Time:/.test(line));
|
|
const planLine = explainLines.find((line) => /Planning Time:/.test(line));
|
|
if (planLine) console.log(planLine);
|
|
if (execLine) console.log(execLine);
|
|
|
|
process.exit(0);
|
|
};
|
|
|
|
await main();
|