alexion
4def357d09
Imported as a linear vendor commit so forge rebases do not replay the upstream root tree. git-subtree-dir: third_party/pi git-subtree-split: d981de1229ef899957bbe968bc8dcda02a21f477
34 lines
1.0 KiB
TypeScript
34 lines
1.0 KiB
TypeScript
import { mkdtemp, rm } from "node:fs/promises";
|
|
import { tmpdir } from "node:os";
|
|
import { join } from "node:path";
|
|
import { BACKGROUND_CONTEXT, type SessionRepo } from "@earendil-works/pi-agent-core";
|
|
import { createNodeSqliteFactory, SqliteSessionRepo } from "../../src/index.ts";
|
|
import type { BenchmarkTarget } from "../../../../agent/benchmark/session/benchmark.ts";
|
|
|
|
export interface SessionRepoBenchmarkFixture extends AsyncDisposable {
|
|
readonly repo: SessionRepo;
|
|
}
|
|
|
|
const NOW = 1_700_000_000_000;
|
|
|
|
export const sessionRepoBenchmarkTargets = [
|
|
{
|
|
name: "sqlite",
|
|
async createFixture() {
|
|
const directory = await mkdtemp(join(tmpdir(), "pi-sqlite-session-repo-benchmark-"));
|
|
const repo = new SqliteSessionRepo({
|
|
directory,
|
|
databaseFactory: createNodeSqliteFactory(),
|
|
now: () => NOW,
|
|
});
|
|
return {
|
|
repo,
|
|
async [Symbol.asyncDispose]() {
|
|
await repo.close(BACKGROUND_CONTEXT);
|
|
await rm(directory, { recursive: true, force: true });
|
|
},
|
|
};
|
|
},
|
|
},
|
|
] satisfies readonly BenchmarkTarget<SessionRepoBenchmarkFixture>[];
|