Benchmark.java

package fastaimcp.benchmark;

import fastaimcp.McpTool;
import fastairuntime.FastAIRuntime;
import fastairuntime.FastCommand;
import fastairuntime.FastObservation;
import fastairuntime.FastTool;
import org.openjdk.jmh.annotations.*;

import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
 * JMH Microbenchmark for FastAIMCP.
 * Measures tool schema mapping, FastAIRuntime adaptation, and dispatch throughput.
 */
@BenchmarkMode(Mode.Throughput)
@OutputTimeUnit(TimeUnit.MILLISECONDS)
@State(Scope.Benchmark)
@Warmup(iterations = 2, time = 1)
@Measurement(iterations = 3, time = 1)
@Fork(1)
public class Benchmark {

    private FastAIRuntime runtime;
    private FastCommand command;

    @Setup
    public void setup() {
        runtime = new FastAIRuntime();

        McpTool tool = new McpTool(
            "read_file",
            "Reads file contents via MCP bridge",
            Map.of("path", "string")
        );

        FastTool adapted = new FastTool() {
            @Override
            public String name() {
                return tool.name();
            }

            @Override
            public FastObservation execute(Map<String, Object> args) {
                return new FastObservation() {
                    @Override
                    public boolean success() {
                        return true;
                    }

                    @Override
                    public String message() {
                        return "mock-content";
                    }
                };
            }
        };

        runtime.register(adapted);
        command = new FastCommand("read_file", Map.of("path", "README.md"));
    }

    @Benchmark
    public FastObservation benchmarkMcpToolDispatch() {
        return runtime.execute(command);
    }
}