Add reservation benchmark

default.nix

@@ -72,6 +72,8 @@ let
         # This is needed for the libsqlite3-sys crate to find the correct sqlite3.
         export PKG_CONFIG_PATH="${pkgs.sqlite.dev}/lib/pkgconfig''${PKG_CONFIG_PATH:+:''${PKG_CONFIG_PATH}}"
       '';
+      # Fixes: libstdc++.so.6: cannot open shared object file: No such file or directory
+      LD_LIBRARY_PATH = pkgs.lib.makeLibraryPath [ pkgs.stdenv.cc.cc.lib ];
     };
   };
 in

justfile

@@ -75,6 +75,16 @@ nix-test-integration *TEST_ARGS: nix-build
 
   timeout --signal term --kill-after {{pytest_timeout_kill_seconds}} {{pytest_timeout_seconds}} pytest --color=yes {{TEST_ARGS}}
 
+# Benchmark query cost and render a graph.
+[group('bench')]
+bench-chunks-select:
+  #!/usr/bin/env bash
+  set -euo pipefail
+  cargo bench --bench chunks_select
+
+  source "./libs/opsqueue_python/.setup_local_venv.sh"
+  python opsqueue/benches/plot_chunks_select.py
+
 # Run all linters, fast and slow
 [group('lint')]
 lint: (lint-light "--all-files") lint-heavy

libs/opsqueue_python/.setup_local_venv.sh

@@ -1,15 +1,19 @@
+relative_dir_path="$(dirname "${BASH_SOURCE[0]}")"
+absolute_dir_path="$(cd "$relative_dir_path" && pwd)"
+
 python_version_hash=$(python --version --version | sha1sum | cut -c1-40)
+venv_path="$absolute_dir_path/.venv-$python_version_hash"
 
 # Then, we (create iff necessary) and activate an (empty!) virtual env
 # so Maturin doesn't complain when running `maturin develop`.
-if [ ! -d ".venv-$python_version_hash" ]; then
+if [ ! -d "$venv_path" ]; then
   echo "Creating an empty Python virtualenv to be able to run 'maturin develop', for Python version '$(python --version --version)' (hash $python_version_hash)..."
-  python -m venv ".venv-$python_version_hash"
+  python -m venv "$venv_path"
   echo "Done!"
 fi
 
-source ".venv-$python_version_hash/bin/activate"
+source "$venv_path/bin/activate"
 
 # Ensure `pytest` is available in the venv;
 # If we were to use `pytest` from Nix, it would not see the locally built python package!
-uv pip install -r pyproject.toml --all-extras --quiet
+uv pip install -r "$absolute_dir_path/pyproject.toml" --all-extras --quiet

libs/opsqueue_python/pyproject.toml

@@ -45,6 +45,7 @@ test = [
     "pytest-xdist==3.8.0",
     "multiprocess==0.70.19",
     "pytest-timeout==2.4.0",
+    "matplotlib>=3.11.1",
 ]
 
 [tool.pytest.ini_options]

opsqueue/Cargo.toml

@@ -79,7 +79,14 @@ workspace-hack.workspace = true
 workspace = true
 
 [dev-dependencies]
+criterion.workspace = true
 insta.workspace = true
+assert_matches.workspace = true
+
+[[bench]]
+name = "chunks_select"
+harness = false
+required-features = ["server-logic"]
 
 [features]
 # Dependencies only in use by the server-logic: