ref(rust): Do not do extra work when merging if not needed

getsentry · Jan 3, 2024 · c217fe5 · c217fe5
1 parent b211283
commit c217fe5
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/rust_snuba/rust_arroyo/src/processing/strategies/reduce.rs b/rust_snuba/rust_arroyo/src/processing/strategies/reduce.rs
@@ -43,8 +43,12 @@ impl<T, TResult> BatchState<T, TResult> {
 
         let tmp = self.value.take().unwrap();
         let payload = message.into_payload();
-        self.message_count += (self.compute_batch_size)(&payload);
-        self.value = Some((self.accumulator)(tmp, payload));
+        let batch_size = (self.compute_batch_size)(&payload);
+
+        if batch_size > 0 {
+            self.message_count += batch_size;
+            self.value = Some((self.accumulator)(tmp, payload));
+        }
     }
 }