batpak 0.2.0

Event sourcing with causal graphs and policy gates. Sync API, zero async.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

#!/usr/bin/env python3
"""Run or compare Criterion benchmarks against per-OS, per-surface baselines."""

from __future__ import annotations

import argparse
import platform
import subprocess
import sys
from pathlib import Path


def os_slug() -> str:
    system = platform.system().lower()
    if system.startswith("windows"):
        return "windows"
    if system.startswith("linux"):
        return "linux"
    if system.startswith("darwin"):
        return "macos"
    return "other"


def cargo_command(surface: str) -> list[str]:
    if surface == "neutral":
        return [
            "cargo",
            "bench",
            "--bench",
            "cold_start",
            "--bench",
            "compaction",
            "--bench",
            "projection_latency",
            "--bench",
            "subscription_fanout",
            "--bench",
            "write_throughput",
        ]
    if surface == "redb":
        return ["cargo", "bench", "--features", "redb", "--bench", "projection_latency"]
    if surface == "lmdb":
        return ["cargo", "bench", "--features", "lmdb", "--bench", "projection_latency"]
    raise ValueError(f"unsupported surface: {surface}")


def main() -> int:
    parser = argparse.ArgumentParser()
    parser.add_argument("--surface", required=True, choices=["neutral", "redb", "lmdb"])
    parser.add_argument("--save", action="store_true")
    parser.add_argument("--compare", action="store_true")
    args = parser.parse_args()

    if args.save and args.compare:
        parser.error("--save and --compare are mutually exclusive")

    baseline = f"{os_slug()}-{args.surface}-v3"
    command = cargo_command(args.surface)

    if args.save:
        command += ["--", "--save-baseline", baseline]
        print(f"Running {args.surface} benchmarks and saving baseline {baseline}...")
    elif args.compare:
        target_dir = Path("target") / "criterion"
        baseline_exists = target_dir.exists() and any(
            path.is_dir() and path.name == baseline for path in target_dir.rglob(baseline)
        )
        if not baseline_exists:
            print(
                f"Baseline {baseline} does not exist yet. Run "
                f"'python ./scripts/bench-report --surface {args.surface} --save' first."
            )
            return 1
        command += ["--", "--baseline", baseline]
        print(f"Comparing {args.surface} benchmarks against baseline {baseline}...")
    else:
        print(f"Running {args.surface} benchmarks...")
        print(f"Baseline name: {baseline}")

    return subprocess.call(command)


if __name__ == "__main__":
    sys.exit(main())