From 53c7be32b6559ce865fd885c28bc2ef95310a9de Mon Sep 17 00:00:00 2001
From: sakridge <stephen@solana.com>
Date: Thu, 24 Oct 2019 19:27:19 -0700
Subject: [PATCH] Add more retransmit and streamer stats (#6534)

---
 core/src/retransmit_stage.rs | 13 ++++++++++++-
 core/src/streamer.rs         |  2 +-
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/core/src/retransmit_stage.rs b/core/src/retransmit_stage.rs
index 1c7cc12e2..5f07f7838 100644
--- a/core/src/retransmit_stage.rs
+++ b/core/src/retransmit_stage.rs
@@ -67,15 +67,24 @@ fn retransmit(
         .unwrap()
         .sorted_retransmit_peers_and_stakes(stakes.as_ref());
     let me = cluster_info.read().unwrap().my_data().clone();
+    let mut discard_total = 0;
+    let mut repair_total = 0;
     let mut retransmit_total = 0;
     let mut compute_turbine_peers_total = 0;
     for mut packets in packet_v {
         for packet in packets.packets.iter_mut() {
             // skip discarded packets and repair packets
-            if packet.meta.discard || packet.meta.repair {
+            if packet.meta.discard {
                 total_packets -= 1;
+                discard_total += 1;
                 continue;
             }
+            if packet.meta.repair {
+                total_packets -= 1;
+                repair_total += 1;
+                continue;
+            }
+
             let mut compute_turbine_peers = Measure::start("turbine_start");
             let (my_index, mut shuffled_stakes_and_index) = ClusterInfo::shuffle_peers_and_index(
                 &me.id,
@@ -126,6 +135,8 @@ fn retransmit(
         ("total_packets", total_packets as i64, i64),
         ("retransmit_total", retransmit_total as i64, i64),
         ("compute_turbine", compute_turbine_peers_total as i64, i64),
+        ("repair_total", i64::from(repair_total), i64),
+        ("discard_total", i64::from(discard_total), i64),
     );
     Ok(())
 }
diff --git a/core/src/streamer.rs b/core/src/streamer.rs
index edb6d6c57..a1a14b18d 100644
--- a/core/src/streamer.rs
+++ b/core/src/streamer.rs
@@ -48,7 +48,7 @@ fn recv_loop(
         }
         if recv_count > 1024 {
             datapoint_debug!(
-                "receiver-stats",
+                name,
                 ("received", recv_count as i64, i64),
                 ("call_count", i64::from(call_count), i64),
                 ("elapsed", now.elapsed().as_millis() as i64, i64),