Orders Bench

1import { bench, describe } from "vitest";
2import { query } from "./clickhouse.js";
3import { type Reader, Cursor } from "../src/readers/core.js";
4import {
5  type OrderRow,
6  readOrderRow,
7  readOrderRowFast,
8} from "../src/examples/orders.js";
9
10/**
11 * Benchmark: API-combinator `readOrderRow` (BigInt `formatUUID`) vs
12 * `readOrderRowFast` (lookup-table `formatUUIDTable` + inlined reads) over the
13 * same large buffer. Since every row stringifies a UUID, the formatter swap is
14 * expected to dominate the win.
15 */
16const N = 20_000;
17const BUF = await query(
18  `SELECT toUInt8(number % 251) AS id, generateUUIDv4() AS uid, ` +
19    `toDecimal64(number / 100, 2) AS price, ` +
20    `CAST(toInt8(number % 3 + 1) AS Enum8('new' = 1, 'shipped' = 2, 'done' = 3)) AS status ` +
21    `FROM numbers(${N}) FORMAT RowBinary`,
22);
23
24function decodeAll(read: Reader<OrderRow>): OrderRow[] {
25  const s = new Cursor(BUF);
26  const out: OrderRow[] = [];
27  while (s.pos < s.buf.length) out.push(read(s));
28  return out;
29}
30
31const norm = (rows: OrderRow[]): string =>
32  JSON.stringify(rows, (_k, v) => (typeof v === "bigint" ? `${v}n` : v));
33{
34  const a = decodeAll(readOrderRow);
35  const b = decodeAll(readOrderRowFast);
36  if (a.length !== N)
37    throw new Error(`orders: decoded ${a.length} rows, expected ${N}`);
38  if (norm(a) !== norm(b)) throw new Error("orders: API vs fast mismatch");
39}
40
41describe("example orders: API vs optimized", () => {
42  bench("API (formatUUID + combinators)", () => {
43    decodeAll(readOrderRow);
44  });
45  bench("optimized (formatUUIDTable + inlined)", () => {
46    decodeAll(readOrderRowFast);
47  });
48});