cleveragents-core/benchmarks/subplan_execution_bench.py

"""Airspeed Velocity benchmarks for subplan execution scheduler overhead.

Measures scheduling, merge, and integrated execution+merge performance for
SubplanExecutionService and SubplanMergeService across all execution modes
and merge strategies.
"""

from __future__ import annotations

from cleveragents.application.services.subplan_execution_service import (
    SubplanExecutionOutput,
    SubplanExecutionService,
)
from cleveragents.application.services.subplan_merge_service import (
    SubplanMergeService,
)
from cleveragents.domain.models.core.plan import (
    ExecutionMode,
    SubplanConfig,
    SubplanMergeStrategy,
    SubplanStatus,
)

_S1 = "01HGZ6FE0AQDYTR4BXVQZ6EA00"
_S2 = "01HGZ6FE0AQDYTR4BXVQZ6EB00"
_S3 = "01HGZ6FE0AQDYTR4BXVQZ6EC00"


def _make_status(subplan_id: str) -> SubplanStatus:
    return SubplanStatus(
        subplan_id=subplan_id,
        action_name="local/bench-sub",
    )


def _noop_executor(status: SubplanStatus) -> SubplanExecutionOutput:
    return SubplanExecutionOutput(
        subplan_id=status.subplan_id,
        success=True,
        files={f"src/{status.subplan_id[-4:]}.py": f"# {status.subplan_id}\n"},
        files_changed=1,
    )


class SubplanExecutionSchedulerSuite:
    """Benchmark SubplanExecutionService scheduling overhead."""

    def setup(self) -> None:
        """Prepare fixtures for scheduler benchmarks."""
        self.statuses = [_make_status(_S1), _make_status(_S2), _make_status(_S3)]
        self.seq_config = SubplanConfig(execution_mode=ExecutionMode.SEQUENTIAL)
        self.par_config = SubplanConfig(
            execution_mode=ExecutionMode.PARALLEL, max_parallel=3
        )
        self.dep_config = SubplanConfig(execution_mode=ExecutionMode.DEPENDENCY_ORDERED)
        self.dep_graph = {_S1: [], _S2: [_S1], _S3: [_S2]}

    def time_sequential_execution(self) -> None:
        """Time sequential execution of 3 subplans."""
        service = SubplanExecutionService(
            config=self.seq_config, executor_fn=_noop_executor
        )
        service.execute_all(subplan_statuses=self.statuses, base_files={})

    def time_parallel_execution(self) -> None:
        """Time parallel execution of 3 subplans."""
        service = SubplanExecutionService(
            config=self.par_config, executor_fn=_noop_executor
        )
        service.execute_all(subplan_statuses=self.statuses, base_files={})

    def time_dependency_ordered_execution(self) -> None:
        """Time dependency-ordered execution of 3 subplans."""
        service = SubplanExecutionService(
            config=self.dep_config, executor_fn=_noop_executor
        )
        service.execute_all(
            subplan_statuses=self.statuses,
            base_files={},
            dependency_graph=self.dep_graph,
        )

    def time_service_construction(self) -> None:
        """Time constructing SubplanExecutionService."""
        SubplanExecutionService(config=self.seq_config, executor_fn=_noop_executor)


class SubplanMergeStrategySuite:
    """Benchmark SubplanMergeService strategy overhead."""

    def setup(self) -> None:
        """Prepare fixtures for merge benchmarks."""
        self.base_files = {"src/main.py": "line1\nline2\nline3\n"}
        self.outputs = [
            (_S1, {"src/main.py": "line1\nline2\nline3\nnew_a\n"}),
            (_S2, {"src/main.py": "new_b\nline1\nline2\nline3\n"}),
        ]
        self.overlapping_outputs = [
            (_S1, {"src/main.py": "first version\n"}),
            (_S2, {"src/main.py": "second version\n"}),
        ]

    def time_git_three_way_merge(self) -> None:
        """Time git three-way merge of non-overlapping changes."""
        service = SubplanMergeService(SubplanMergeStrategy.GIT_THREE_WAY)
        service.merge(self.base_files, self.outputs)

    def time_sequential_apply_merge(self) -> None:
        """Time sequential apply merge."""
        service = SubplanMergeService(SubplanMergeStrategy.SEQUENTIAL_APPLY)
        service.merge(self.base_files, self.overlapping_outputs)

    def time_last_wins_merge(self) -> None:
        """Time last-wins merge."""
        service = SubplanMergeService(SubplanMergeStrategy.LAST_WINS)
        service.merge(self.base_files, self.overlapping_outputs)

    def time_merge_service_construction(self) -> None:
        """Time constructing SubplanMergeService."""
        SubplanMergeService(SubplanMergeStrategy.GIT_THREE_WAY)

    def time_single_file_merge(self) -> None:
        """Time merging a single file output."""
        service = SubplanMergeService(SubplanMergeStrategy.LAST_WINS)
        service.merge(
            {"src/main.py": "base\n"},
            [(_S1, {"src/main.py": "modified\n"})],
        )


class SubplanIntegrationSuite:
    """Benchmark integrated execution+merge overhead."""

    def setup(self) -> None:
        """Prepare fixtures for integration benchmarks."""
        self.statuses = [_make_status(_S1), _make_status(_S2)]

    def time_sequential_with_last_wins(self) -> None:
        """Time sequential execution with last-wins merge."""
        config = SubplanConfig(
            execution_mode=ExecutionMode.SEQUENTIAL,
            merge_strategy=SubplanMergeStrategy.LAST_WINS,
        )
        service = SubplanExecutionService(config=config, executor_fn=_noop_executor)
        service.execute_all(subplan_statuses=self.statuses, base_files={})

    def time_parallel_with_git_merge(self) -> None:
        """Time parallel execution with git three-way merge."""
        config = SubplanConfig(
            execution_mode=ExecutionMode.PARALLEL,
            max_parallel=2,
            merge_strategy=SubplanMergeStrategy.GIT_THREE_WAY,
        )
        service = SubplanExecutionService(config=config, executor_fn=_noop_executor)
        service.execute_all(subplan_statuses=self.statuses, base_files={})