Skip to content

Commit

Permalink
Add python consumer flag
Browse files Browse the repository at this point in the history
  • Loading branch information
ayirr7 committed Dec 7, 2024
1 parent 1a8c0eb commit b240f35
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion snuba/cli/rust_consumer.py
Original file line number Diff line number Diff line change
Expand Up @@ -148,6 +148,11 @@
This is only to be used for the mutability consumer
""",
)
@click.option(
"--max-dlq-buffer-size",
default=None,
help="Set a per-partition limit to the size (length) of the DLQ buffer",
)
@click.option(
"--health-check-file",
default=None,
Expand Down Expand Up @@ -212,7 +217,8 @@ def rust_consumer(
stop_at_timestamp: Optional[int],
batch_write_timeout_ms: Optional[int],
max_bytes_before_external_group_by: Optional[int],
mutations_mode: bool
mutations_mode: bool,
max_dlq_buffer_size: Optional[int]
) -> None:
"""
Experimental alternative to `snuba consumer`
Expand Down Expand Up @@ -265,6 +271,7 @@ def rust_consumer(
stop_at_timestamp,
batch_write_timeout_ms,
max_bytes_before_external_group_by,
max_dlq_buffer_size,
)

sys.exit(exitcode)

0 comments on commit b240f35

Please sign in to comment.