Skip to content

feat(cli): expose insert-rate, search-stages, streaming-concurrencies for StreamingPerformanceCase - #5

Open
Saccilotto wants to merge 1 commit into
mainfrom
feat/streaming-cli-flags
Open

Saccilotto wants to merge 1 commit into
mainfrom
feat/streaming-cli-flags

Conversation

@Saccilotto

Copy link
Copy Markdown
Collaborator

Problem

--case-type StreamingPerformanceCase is accepted, but get_custom_case_config
has no branch for it, so insert_rate, search_stages, and concurrencies
always fall back to the class defaults (500 rows/s, [0.5, 0.8], [5, 10]) with
no way to override them from the CLI.

This is what GRINPROJS-PUCRS/expdb#404 needs: the expdb adapter builds a
streaming profile and expects --insert-rate, --search-stages, and
--streaming-concurrencies to exist on this CLI.

Change

  • --insert-rate INTEGER (default 500)
  • --search-stages TEXT — comma-separated stream-completion fractions, e.g. 0.5,0.8
  • --streaming-concurrencies TEXT — comma-separated concurrency levels, e.g. 5,10

Parsed the same way --num-concurrency already is, and only routed into
custom_case_config when --case-type StreamingPerformanceCase; every other
case type is untouched.

Validation

test_streaming_cli.py (new, 4 cases: options exposed, defaults, overrides,
ignored for non-streaming case types) and test_cli_note.py pass. black --check / ruff check clean on the touched files.

… for StreamingPerformanceCase

The CLI accepted --case-type StreamingPerformanceCase but had no way to
override its insert_rate, search_stages, or concurrencies knobs, so every
run silently used the class defaults (500 rows/s, 0.5/0.8 stages, 5/10
concurrencies).

Add --insert-rate, --search-stages, and --streaming-concurrencies options
(comma-separated for the two list values, mirroring --num-concurrency) and
wire them into get_custom_case_config for StreamingPerformanceCase. Other
case types are unaffected.
@Saccilotto Saccilotto self-assigned this Sep 14, 2026
@Saccilotto

Copy link
Copy Markdown
Collaborator Author

Tracked end-to-end in GRINPROJS-PUCRS/expdb#411 (this repo has Issues disabled).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant