Skip to content

Commit c16ab3d

Browse files
committed
changes improvements
1 parent b7bab3b commit c16ab3d

File tree

2 files changed

+15
-1
lines changed

2 files changed

+15
-1
lines changed

robusta_krr/core/integrations/kubernetes/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -277,7 +277,7 @@ async def _list_namespaced_or_global_objects_batched(
277277
limit: Optional[int] = None,
278278
continue_ref: Optional[str] = None,
279279
) -> tuple[list[Any], Optional[str]]:
280-
logger.debug(f"Listing {kind}s in {self.cluster} with batching (limit={limit})")
280+
logger.debug("Listing %s in %s with batching (limit=%d)", kind, self.cluster, limit)
281281
loop = asyncio.get_running_loop()
282282

283283
try:

robusta_krr/main.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -232,6 +232,18 @@ def run_strategy(
232232
help="Maximum number of jobs/pods to query per GroupedJob group (default: 500).",
233233
rich_help_panel="Job Grouping Settings",
234234
),
235+
discovery_job_batch_size: int = typer.Option(
236+
1000,
237+
"--discovery-job-batch-size",
238+
help="Batch size for Kubernetes job API calls (default: 1000).",
239+
rich_help_panel="Job Discovery Settings",
240+
),
241+
discovery_job_max_batches: int = typer.Option(
242+
50,
243+
"--discovery-job-max-batches",
244+
help="Maximum number of job batches to process to prevent infinite loops (default: 50).",
245+
rich_help_panel="Job Discovery Settings",
246+
),
235247
format: str = typer.Option(
236248
"table",
237249
"--formatter",
@@ -371,6 +383,8 @@ def run_strategy(
371383
max_workers=max_workers,
372384
job_grouping_labels=job_grouping_labels,
373385
job_grouping_limit=job_grouping_limit,
386+
discovery_job_batch_size=discovery_job_batch_size,
387+
discovery_job_max_batches=discovery_job_max_batches,
374388
format=format,
375389
show_cluster_name=show_cluster_name,
376390
verbose=verbose,

0 commit comments

Comments
 (0)