Skip to content

Commit 3d22067

Browse files
author
Andrei Neagu
committed
replace rate check with size check
1 parent 7756e79 commit 3d22067

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

services/director-v2/src/simcore_service_director_v2/modules/dynamic_sidecar/scheduler/_core/_events_utils.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -173,10 +173,10 @@ async def service_save_state(
173173
size = await sidecars_client.save_service_state(
174174
scheduler_data.endpoint, progress_callback=progress_callback
175175
)
176-
if rate := get_rate(size, duration.to_float()) >= 1:
176+
if size > 0:
177177
get_instrumentation(app).dynamic_sidecar_metrics.push_service_state_rate.labels(
178178
**get_metrics_labels(scheduler_data)
179-
).observe(rate)
179+
).observe(get_rate(size, duration.to_float()))
180180

181181
await sidecars_client.update_volume_state(
182182
scheduler_data.endpoint,
@@ -480,13 +480,13 @@ async def _pull_output_ports_with_metrics() -> None:
480480
size: int = await sidecars_client.pull_service_output_ports(
481481
dynamic_sidecar_endpoint
482482
)
483-
if rate := get_rate(size, duration.to_float()) >= 1:
483+
if size > 0:
484484
get_instrumentation(
485485
app
486486
).dynamic_sidecar_metrics.output_ports_pull_rate.labels(
487487
**get_metrics_labels(scheduler_data)
488488
).observe(
489-
rate
489+
get_rate(size, duration.to_float())
490490
)
491491

492492
async def _pull_user_services_images_with_metrics() -> None:
@@ -505,13 +505,13 @@ async def _restore_service_state_with_metrics() -> None:
505505
with track_duration() as duration:
506506
size = await sidecars_client.restore_service_state(dynamic_sidecar_endpoint)
507507

508-
if rate := get_rate(size, duration.to_float()) >= 1:
508+
if size > 0:
509509
get_instrumentation(
510510
app
511511
).dynamic_sidecar_metrics.pull_service_state_rate.labels(
512512
**get_metrics_labels(scheduler_data)
513513
).observe(
514-
rate
514+
get_rate(size, duration.to_float())
515515
)
516516

517517
tasks = [

services/director-v2/src/simcore_service_director_v2/modules/dynamic_sidecar/scheduler/_core/_scheduler.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -464,13 +464,13 @@ async def retrieve_service_inputs(
464464
)
465465
duration = time.time() - started
466466

467-
if rate := get_rate(transferred_bytes, duration) >= 1:
467+
if transferred_bytes > 0:
468468
get_instrumentation(
469469
self.app
470470
).dynamic_sidecar_metrics.input_ports_pull_rate.labels(
471471
**get_metrics_labels(scheduler_data)
472472
).observe(
473-
rate
473+
get_rate(transferred_bytes, duration)
474474
)
475475

476476
if scheduler_data.restart_policy == RestartPolicy.ON_INPUTS_DOWNLOADED:

0 commit comments

Comments
 (0)