Correct num_packets stats (solana-labs#33630)

Do not count the empty packets from cache warmer in num_packets stats as they are not sent.
anza-xyz · Oct 11, 2023 · 21b2fce · 21b2fce
1 parent 295d610
commit 21b2fce
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/quic-client/src/nonblocking/quic_client.rs b/quic-client/src/nonblocking/quic_client.rs
@@ -582,10 +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
+        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);
+                self.connection_stats
+                    .add_client_stats(&stats, num_packets, true);
                 v
             })
             .map_err(|e| {
@@ -595,7 +598,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