[C] Increment work_count based on the actual work done.

real-logic · Oct 15, 2024 · 9a04bd6 · 9a04bd6
1 parent 57ff2fe
commit 9a04bd6
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/aeron-driver/src/main/c/aeron_driver_receiver.c b/aeron-driver/src/main/c/aeron_driver_receiver.c
@@ -176,34 +176,37 @@ int aeron_driver_receiver_do_work(void *clientd)
 
  if (NULL != image->endpoint)
  {
- if (aeron_publication_image_send_pending_status_message(image, now_ns) < 0)
+ int send_result = aeron_publication_image_send_pending_status_message(image, now_ns);
+ if (send_result < 0)
  {
  AERON_APPEND_ERR("%s", "receiver send SM");
  aeron_driver_receiver_log_error(receiver);
  }
  else
  {
- work_count++;
+ work_count += send_result;
  }
 
- if (aeron_publication_image_send_pending_loss(image) < 0)
+ send_result = aeron_publication_image_send_pending_loss(image);
+ if (send_result < 0)
  {
  AERON_APPEND_ERR("%s", "receiver send NAK");
  aeron_driver_receiver_log_error(receiver);
  }
  else
  {
- work_count++;
+ work_count += send_result;
  }
 
- if (aeron_publication_image_initiate_rttm(image, now_ns) < 0)
+ send_result = aeron_publication_image_initiate_rttm(image, now_ns);
+ if (send_result < 0)
  {
  AERON_APPEND_ERR("%s", "receiver send RTTM");
  aeron_driver_receiver_log_error(receiver);
  }
  else
  {
- work_count++;
+ work_count += send_result;
  }
  }
  }