Correct num_packets stats

solana-labs · Oct 10, 2023 · ff78f0d · ff78f0d
1 parent 6ea5128
commit ff78f0d
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/quic-client/src/nonblocking/quic_client.rs b/quic-client/src/nonblocking/quic_client.rs
@@ -582,11 +582,13 @@ impl ClientConnection for QuicClientConnection {
 
     async fn send_data(&self, data: &[u8]) -> TransportResult<()> {
         let stats = Arc::new(ClientStats::default());
+        // When data is empty which is from cache warmer, we are not sending packets actually, do not count it in stats.
+        let num_packets = if data.is_empty() { 0 } else { 1 };
         self.client
             .send_buffer(data, &stats, self.connection_stats.clone())
             .map_ok(|v| {
-                self.connection_stats.add_client_stats(&stats, 1, true);
-                v
+                self.connection_stats
+                    .add_client_stats(&stats, num_packets, true);
             })
             .map_err(|e| {
                 warn!(
@@ -595,7 +597,8 @@ impl ClientConnection for QuicClientConnection {
                     e
                 );
                 datapoint_warn!("send-wire-async", ("failure", 1, i64),);
-                self.connection_stats.add_client_stats(&stats, 1, false);
+                self.connection_stats
+                    .add_client_stats(&stats, num_packets, false);
                 e.into()
             })
             .await