Skip to content

3/6 Add OtlpParamSource for OTLP corpora#2136

Open
gareth-ellis wants to merge 3 commits into
masterfrom
otlp-pr3-param-source
Open

3/6 Add OtlpParamSource for OTLP corpora#2136
gareth-ellis wants to merge 3 commits into
masterfrom
otlp-pr3-param-source

Conversation

@gareth-ellis

@gareth-ellis gareth-ellis commented May 29, 2026

Copy link
Copy Markdown
Member

Summary

Adds OtlpParamSource in esrally/track/params.py — the parameter source that drives OTLP ingest operations.

  • Streams length-prefixed protobuf records out of an OtlpProtobufFile rather than materializing into memory (a 1 MB-per-record corpus with 6k records per partition would otherwise consume 6+ GB per worker).
  • Partitions records across workers using the offset sidecar so workers seek directly to their slice.
  • Surfaces percent_completed so Rally's progress bar tracks real progress.
  • Supports a looped mode that cycles the partition indefinitely for time-bound benchmarks.
  • Honors a corpora operation param so a track with multiple OTLP corpora can target a specific one.

Depends on #2135 — merge after #2135. Part 3 of 6.

Series

Test plan

  • 12 new unit tests cover partitioning, looping, percent-completed, corpus filtering, StopIteration semantics
  • pre-commit clean

🤖 Generated with Claude Code

gareth-ellis and others added 3 commits May 29, 2026 14:41
The ApiError handler in execute_single() decodes `e.body`, `e.error`, and
`e.info` as UTF-8 to build a human-readable error message. When the body is
binary (e.g., binary protobuf returned by ES OTLP endpoints on 4xx/5xx), the
strict decode raises UnicodeDecodeError, which crashes the worker mid-task.

Switch the six decode() calls to use errors="replace" so undecodable bytes
become U+FFFD instead of aborting the worker. No semantic change for valid
UTF-8 (the common case).

This is a latent bug independent of OTLP — any operation that surfaces a
binary error body would have hit it.
Introduces OtlpProtobufFile in esrally/utils/io.py for reading/writing
length-prefixed OTLP ExportMetricsServiceRequest protobufs, plus an
offset sidecar to allow worker partitions to seek without scanning.

Wires preparation into esrally/track/loader.py and esrally/track/track.py:
- New OTLP document set fields (otlp_pb_size_in_bytes, etc.)
- prepare_otlp_document_set tries to download a .pb from the corpus base
  URL, otherwise converts a local JSON corpus to .pb on disk.
- set_absolute_data_path picks up the .pb when present.

Adds OTLP protobuf bindings to pyproject.toml (opentelemetry-proto).

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
OtlpParamSource streams length-prefixed protobuf records out of an
OtlpProtobufFile, partitions them across workers using the offset
sidecar, and surfaces percent_completed so the progress bar tracks
real progress. Supports a "looped" mode that cycles the partition
indefinitely for time-bound benchmarks.

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
@gareth-ellis gareth-ellis requested a review from a team as a code owner May 29, 2026 14:10
@gareth-ellis gareth-ellis changed the base branch from otlp-pr2-core-io to master May 29, 2026 14:31
@gareth-ellis gareth-ellis changed the title Add OtlpParamSource for OTLP corpora 3/6 Add OtlpParamSource for OTLP corpora May 29, 2026

@ebadyano ebadyano left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm

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.

2 participants