mirror of
https://github.com/Youzini-afk/ST-Bionic-Memory-Ecology.git
synced 2026-05-15 22:30:38 +08:00
perf: optimize persist delta gating and diagnostics
This commit is contained in:
Youzini-afk
158
tests/perf/graph-layout-bench.mjs
Normal file
158
tests/perf/graph-layout-bench.mjs
Normal file
@@ -0,0 +1,158 @@
|
||||
import { performance } from "node:perf_hooks";
|
||||
|
||||
import { solveLayoutWithJs } from "../../ui/graph-layout-solver.js";
|
||||
import {
|
||||
getNativeModuleStatus,
|
||||
solveLayout as solveNativeLayout,
|
||||
} from "../../vendor/wasm/stbme_core.js";
|
||||
|
||||
const SCALES = [
|
||||
{ nodes: 600, edgeMultiplier: 3 },
|
||||
{ nodes: 1200, edgeMultiplier: 4 },
|
||||
{ nodes: 2000, edgeMultiplier: 4 },
|
||||
];
|
||||
|
||||
const RUNS = 3;
|
||||
|
||||
function buildPayload(seed = 7, nodeCount = 600, edgeMultiplier = 4) {
|
||||
let state = seed >>> 0;
|
||||
const rand = () => {
|
||||
state = (state * 1664525 + 1013904223) >>> 0;
|
||||
return state / 0xffffffff;
|
||||
};
|
||||
|
||||
const regionRect = { x: 0, y: 0, w: 1280, h: 780 };
|
||||
const nodes = new Array(nodeCount).fill(null).map(() => ({
|
||||
x: regionRect.x + rand() * regionRect.w,
|
||||
y: regionRect.y + rand() * regionRect.h,
|
||||
vx: 0,
|
||||
vy: 0,
|
||||
pinned: false,
|
||||
radius: 5.5 + rand() * 8,
|
||||
regionKey: "objective",
|
||||
regionRect,
|
||||
}));
|
||||
|
||||
const edgeCount = Math.max(1, Math.floor(nodeCount * edgeMultiplier));
|
||||
const edges = [];
|
||||
for (let i = 0; i < edgeCount; i++) {
|
||||
const from = Math.floor(rand() * nodeCount);
|
||||
let to = Math.floor(rand() * nodeCount);
|
||||
if (to === from) to = (to + 1) % nodeCount;
|
||||
edges.push({
|
||||
from,
|
||||
to,
|
||||
strength: 0.25 + rand() * 0.75,
|
||||
});
|
||||
}
|
||||
|
||||
return {
|
||||
nodes,
|
||||
edges,
|
||||
config: {
|
||||
iterations: 56,
|
||||
repulsion: 2600,
|
||||
springK: 0.05,
|
||||
damping: 0.87,
|
||||
centerGravity: 0.015,
|
||||
minGap: 11,
|
||||
speedCap: 3.2,
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
function summarize(values = []) {
|
||||
if (!values.length) return { avg: 0, p95: 0, min: 0, max: 0 };
|
||||
const sorted = [...values].sort((a, b) => a - b);
|
||||
const sum = sorted.reduce((acc, value) => acc + value, 0);
|
||||
const p95Index = Math.min(sorted.length - 1, Math.floor(sorted.length * 0.95));
|
||||
return {
|
||||
avg: sum / sorted.length,
|
||||
p95: sorted[p95Index],
|
||||
min: sorted[0],
|
||||
max: sorted[sorted.length - 1],
|
||||
};
|
||||
}
|
||||
|
||||
async function runNative(payload) {
|
||||
const start = performance.now();
|
||||
const result = await solveNativeLayout(payload);
|
||||
const elapsed = performance.now() - start;
|
||||
return { elapsed, result };
|
||||
}
|
||||
|
||||
function runJs(payload) {
|
||||
const start = performance.now();
|
||||
const result = solveLayoutWithJs(payload);
|
||||
const elapsed = performance.now() - start;
|
||||
return { elapsed, result };
|
||||
}
|
||||
|
||||
async function warmUp() {
|
||||
const payload = buildPayload(12345, 320, 3);
|
||||
runJs(payload);
|
||||
await runNative(payload);
|
||||
}
|
||||
|
||||
async function main() {
|
||||
const originalLoader = globalThis.__stBmeLoadRustWasmLayout;
|
||||
if (typeof originalLoader !== "function") {
|
||||
globalThis.__stBmeLoadRustWasmLayout = async () => ({
|
||||
solve_layout(payload) {
|
||||
const jsResult = solveLayoutWithJs(payload);
|
||||
return {
|
||||
ok: true,
|
||||
positions: Array.from(jsResult.positions),
|
||||
diagnostics: {
|
||||
solver: "mock-rust-wasm",
|
||||
nodeCount: jsResult.diagnostics.nodeCount,
|
||||
edgeCount: jsResult.diagnostics.edgeCount,
|
||||
iterations: jsResult.diagnostics.iterations,
|
||||
},
|
||||
};
|
||||
},
|
||||
});
|
||||
}
|
||||
|
||||
try {
|
||||
await warmUp();
|
||||
const nativeStatus = getNativeModuleStatus();
|
||||
console.log(`[ST-BME][bench] graph-layout runs=${RUNS}`);
|
||||
console.log(
|
||||
`[ST-BME][bench] graph-layout native-source=${nativeStatus.source || "unknown"}`,
|
||||
);
|
||||
for (const scale of SCALES) {
|
||||
const jsTimes = [];
|
||||
const nativeTimes = [];
|
||||
for (let run = 0; run < RUNS; run++) {
|
||||
const payload = buildPayload(
|
||||
scale.nodes * 31 + run,
|
||||
scale.nodes,
|
||||
scale.edgeMultiplier,
|
||||
);
|
||||
const js = runJs(payload);
|
||||
jsTimes.push(js.elapsed);
|
||||
|
||||
const native = await runNative(payload);
|
||||
nativeTimes.push(native.elapsed);
|
||||
}
|
||||
|
||||
const jsSummary = summarize(jsTimes);
|
||||
const nativeSummary = summarize(nativeTimes);
|
||||
console.log(
|
||||
`[ST-BME][bench] nodes=${scale.nodes} edges≈${Math.floor(scale.nodes * scale.edgeMultiplier)} | js avg=${jsSummary.avg.toFixed(2)}ms p95=${jsSummary.p95.toFixed(2)}ms | native avg=${nativeSummary.avg.toFixed(2)}ms p95=${nativeSummary.p95.toFixed(2)}ms`,
|
||||
);
|
||||
}
|
||||
} finally {
|
||||
if (typeof originalLoader === "function") {
|
||||
globalThis.__stBmeLoadRustWasmLayout = originalLoader;
|
||||
} else {
|
||||
delete globalThis.__stBmeLoadRustWasmLayout;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
main().catch((error) => {
|
||||
console.error("[ST-BME][bench] graph-layout failed:", error?.message || String(error));
|
||||
process.exitCode = 1;
|
||||
});
|
||||
161
tests/perf/persist-delta-bench.mjs
Normal file
161
tests/perf/persist-delta-bench.mjs
Normal file
@@ -0,0 +1,161 @@
|
||||
import { performance } from "node:perf_hooks";
|
||||
|
||||
import { buildPersistDelta } from "../../sync/bme-db.js";
|
||||
import {
|
||||
getNativeModuleStatus,
|
||||
installNativePersistDeltaHook,
|
||||
} from "../../vendor/wasm/stbme_core.js";
|
||||
|
||||
const RUNS = 5;
|
||||
|
||||
function buildSnapshots(seed = 5, nodeCount = 5000, edgeCount = 12000, churn = 0.1) {
|
||||
let state = seed >>> 0;
|
||||
const rand = () => {
|
||||
state = (state * 1664525 + 1013904223) >>> 0;
|
||||
return state / 0xffffffff;
|
||||
};
|
||||
|
||||
const beforeNodes = [];
|
||||
for (let i = 0; i < nodeCount; i++) {
|
||||
beforeNodes.push({
|
||||
id: `n-${i}`,
|
||||
type: "event",
|
||||
fields: {
|
||||
text: `node-${i}`,
|
||||
v: Math.floor(rand() * 1000),
|
||||
},
|
||||
archived: false,
|
||||
updatedAt: 1000 + i,
|
||||
});
|
||||
}
|
||||
|
||||
const beforeEdges = [];
|
||||
for (let i = 0; i < edgeCount; i++) {
|
||||
const from = Math.floor(rand() * nodeCount);
|
||||
let to = Math.floor(rand() * nodeCount);
|
||||
if (to === from) to = (to + 1) % nodeCount;
|
||||
beforeEdges.push({
|
||||
id: `e-${i}`,
|
||||
fromId: `n-${from}`,
|
||||
toId: `n-${to}`,
|
||||
relation: "related",
|
||||
strength: rand(),
|
||||
updatedAt: 1000 + i,
|
||||
});
|
||||
}
|
||||
|
||||
const afterNodes = beforeNodes.map((node) => ({ ...node, fields: { ...node.fields } }));
|
||||
const afterEdges = beforeEdges.map((edge) => ({ ...edge }));
|
||||
|
||||
const mutateNodeCount = Math.floor(nodeCount * churn);
|
||||
for (let i = 0; i < mutateNodeCount; i++) {
|
||||
const index = Math.floor(rand() * afterNodes.length);
|
||||
afterNodes[index].fields.v = Math.floor(rand() * 5000);
|
||||
afterNodes[index].updatedAt += 100;
|
||||
}
|
||||
|
||||
const addNodeCount = Math.max(1, Math.floor(nodeCount * churn * 0.25));
|
||||
const baseNodeId = afterNodes.length;
|
||||
for (let i = 0; i < addNodeCount; i++) {
|
||||
afterNodes.push({
|
||||
id: `n-new-${baseNodeId + i}`,
|
||||
type: "event",
|
||||
fields: { text: `new-${i}`, v: Math.floor(rand() * 3000) },
|
||||
archived: false,
|
||||
updatedAt: 5000 + i,
|
||||
});
|
||||
}
|
||||
|
||||
const removeEdgeCount = Math.max(1, Math.floor(edgeCount * churn * 0.2));
|
||||
afterEdges.splice(0, removeEdgeCount);
|
||||
|
||||
return {
|
||||
before: {
|
||||
meta: { chatId: "bench-chat", revision: 1, lastModified: 1000 },
|
||||
state: { lastProcessedFloor: 1, extractionCount: 1 },
|
||||
nodes: beforeNodes,
|
||||
edges: beforeEdges,
|
||||
tombstones: [],
|
||||
},
|
||||
after: {
|
||||
meta: { chatId: "bench-chat", revision: 2, lastModified: 2000 },
|
||||
state: { lastProcessedFloor: 2, extractionCount: 2 },
|
||||
nodes: afterNodes,
|
||||
edges: afterEdges,
|
||||
tombstones: [],
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
function summarize(values = []) {
|
||||
if (!values.length) return { avg: 0, p95: 0, min: 0, max: 0 };
|
||||
const sorted = [...values].sort((a, b) => a - b);
|
||||
const sum = sorted.reduce((acc, value) => acc + value, 0);
|
||||
const p95Index = Math.min(sorted.length - 1, Math.floor(sorted.length * 0.95));
|
||||
return {
|
||||
avg: sum / sorted.length,
|
||||
p95: sorted[p95Index],
|
||||
min: sorted[0],
|
||||
max: sorted[sorted.length - 1],
|
||||
};
|
||||
}
|
||||
|
||||
async function main() {
|
||||
await installNativePersistDeltaHook();
|
||||
const nativeStatus = getNativeModuleStatus();
|
||||
const jsSamples = [];
|
||||
const nativeSamples = [];
|
||||
for (let run = 0; run < RUNS; run++) {
|
||||
const snapshots = buildSnapshots(17 + run, 5000, 12000, 0.12);
|
||||
const jsStartedAt = performance.now();
|
||||
const jsDelta = buildPersistDelta(snapshots.before, snapshots.after, {
|
||||
useNativeDelta: false,
|
||||
});
|
||||
const jsElapsedMs = performance.now() - jsStartedAt;
|
||||
jsSamples.push({
|
||||
elapsedMs: jsElapsedMs,
|
||||
upsertNodes: jsDelta.upsertNodes.length,
|
||||
upsertEdges: jsDelta.upsertEdges.length,
|
||||
deleteNodeIds: jsDelta.deleteNodeIds.length,
|
||||
deleteEdgeIds: jsDelta.deleteEdgeIds.length,
|
||||
});
|
||||
|
||||
const nativeStartedAt = performance.now();
|
||||
const nativeDelta = buildPersistDelta(snapshots.before, snapshots.after, {
|
||||
useNativeDelta: true,
|
||||
minSnapshotRecords: 0,
|
||||
minStructuralDelta: 0,
|
||||
minCombinedSerializedChars: 0,
|
||||
nativeFailOpen: false,
|
||||
});
|
||||
const nativeElapsedMs = performance.now() - nativeStartedAt;
|
||||
nativeSamples.push({
|
||||
elapsedMs: nativeElapsedMs,
|
||||
upsertNodes: nativeDelta.upsertNodes.length,
|
||||
upsertEdges: nativeDelta.upsertEdges.length,
|
||||
deleteNodeIds: nativeDelta.deleteNodeIds.length,
|
||||
deleteEdgeIds: nativeDelta.deleteEdgeIds.length,
|
||||
});
|
||||
}
|
||||
|
||||
const jsTimingSummary = summarize(jsSamples.map((sample) => sample.elapsedMs));
|
||||
const nativeTimingSummary = summarize(nativeSamples.map((sample) => sample.elapsedMs));
|
||||
const avgUpserts =
|
||||
jsSamples.reduce((acc, sample) => acc + sample.upsertNodes + sample.upsertEdges, 0) /
|
||||
jsSamples.length;
|
||||
const avgDeletes =
|
||||
jsSamples.reduce((acc, sample) => acc + sample.deleteNodeIds + sample.deleteEdgeIds, 0) /
|
||||
jsSamples.length;
|
||||
|
||||
console.log(
|
||||
`[ST-BME][bench] persist-delta native-source=${nativeStatus.source || "unknown"}`,
|
||||
);
|
||||
console.log(
|
||||
`[ST-BME][bench] persist-delta runs=${RUNS} | js avg=${jsTimingSummary.avg.toFixed(2)}ms p95=${jsTimingSummary.p95.toFixed(2)}ms min=${jsTimingSummary.min.toFixed(2)}ms max=${jsTimingSummary.max.toFixed(2)}ms | native avg=${nativeTimingSummary.avg.toFixed(2)}ms p95=${nativeTimingSummary.p95.toFixed(2)}ms min=${nativeTimingSummary.min.toFixed(2)}ms max=${nativeTimingSummary.max.toFixed(2)}ms | avgUpserts=${avgUpserts.toFixed(1)} avgDeletes=${avgDeletes.toFixed(1)}`,
|
||||
);
|
||||
}
|
||||
|
||||
main().catch((error) => {
|
||||
console.error("[ST-BME][bench] persist-delta failed:", error?.message || String(error));
|
||||
process.exitCode = 1;
|
||||
});
|
||||
Reference in New Issue
Block a user