Class IsolatedBenchmark

benchmarks/dirty_benchmark.py:192–367

Run benchmarks directly against the dirty pool without HTTP. Spawns a standalone dirty arbiter and workers, then runs concurrent clients to measure performance.

class IsolatedBenchmark:
    """
    Run benchmarks directly against the dirty pool without HTTP.

    Spawns a standalone dirty arbiter and workers, then runs concurrent
    clients to measure performance.
    """

    def __init__(
        self,
        dirty_workers: int = 2,
        dirty_threads: int = 1,
        dirty_timeout: int = 300,
        verbose: bool = False,
    ):
        self.dirty_workers = dirty_workers
        self.dirty_threads = dirty_threads
        self.dirty_timeout = dirty_timeout
        self.verbose = verbose

        self.arbiter = None
        self.arbiter_pid = None
        self.socket_path = None
        self._tmpdir = None

    def start(self):
        """Start the dirty arbiter and workers."""
        # Create temp directory for socket
        self._tmpdir = tempfile.mkdtemp(prefix="dirty-bench-")
        self.socket_path = os.path.join(self._tmpdir, "arbiter.sock")

        # Create config and logger
        cfg = MockConfig(
            dirty_apps=[BENCHMARK_APP],
            dirty_workers=self.dirty_workers,
            dirty_threads=self.dirty_threads,
            dirty_timeout=self.dirty_timeout,
        )
        log = MockLogger(verbose=self.verbose)

        # Fork arbiter process
        pid = os.fork()
        if pid == 0:
            # Child process - run arbiter
            try:
                arbiter = DirtyArbiter(cfg, log, socket_path=self.socket_path)
                arbiter.run()
            except Exception as e:
                print(f"Arbiter error: {e}")
            finally:
                os._exit(0)

        # Parent process
        self.arbiter_pid = pid

        # Wait for arbiter socket to be ready
        for _ in range(50):  # 5 seconds max
            if os.path.exists(self.socket_path):