From 26fdf3fb07723c64209221e35002be5d825c62d2 Mon Sep 17 00:00:00 2001
From: "mergify[bot]" <37929162+mergify[bot]@users.noreply.github.com>
Date: Wed, 23 Jun 2021 12:43:37 -0500
Subject: [PATCH] Add batch stats (#18096) (#18142)

(cherry picked from commit e808f34b0b22c881bdba5c8caa317fca69782e8b)

Co-authored-by: sakridge <sakridge@gmail.com>
---
 core/src/progress_map.rs           | 10 ++++++++++
 ledger/src/blockstore_processor.rs |  2 ++
 runtime/src/bank.rs                |  4 ++++
 3 files changed, 16 insertions(+)

diff --git a/core/src/progress_map.rs b/core/src/progress_map.rs
index 9724c57f92bbc9..36348ca534090a 100644
--- a/core/src/progress_map.rs
+++ b/core/src/progress_map.rs
@@ -63,6 +63,16 @@ impl ReplaySlotStats {
             ("load_us", self.execute_timings.load_us, i64),
             ("execute_us", self.execute_timings.execute_us, i64),
             ("store_us", self.execute_timings.store_us, i64),
+            (
+                "total_batches_len",
+                self.execute_timings.total_batches_len,
+                i64
+            ),
+            (
+                "num_execute_batches",
+                self.execute_timings.num_execute_batches,
+                i64
+            ),
             (
                 "serialize_us",
                 self.execute_timings.details.serialize_us,
diff --git a/ledger/src/blockstore_processor.rs b/ledger/src/blockstore_processor.rs
index 957b604f900e5a..16346c06587866 100644
--- a/ledger/src/blockstore_processor.rs
+++ b/ledger/src/blockstore_processor.rs
@@ -195,6 +195,8 @@ fn execute_batches(
             })
         });
 
+    timings.total_batches_len += batches.len();
+    timings.num_execute_batches += 1;
     for timing in new_timings {
         timings.accumulate(&timing);
     }
diff --git a/runtime/src/bank.rs b/runtime/src/bank.rs
index 09f025dbe6887a..a276ea7936a625 100644
--- a/runtime/src/bank.rs
+++ b/runtime/src/bank.rs
@@ -157,6 +157,8 @@ pub struct ExecuteTimings {
     pub load_us: u64,
     pub execute_us: u64,
     pub store_us: u64,
+    pub total_batches_len: usize,
+    pub num_execute_batches: u64,
     pub details: ExecuteDetailsTimings,
 }
 
@@ -166,6 +168,8 @@ impl ExecuteTimings {
         self.load_us += other.load_us;
         self.execute_us += other.execute_us;
         self.store_us += other.store_us;
+        self.total_batches_len += other.total_batches_len;
+        self.num_execute_batches += other.num_execute_batches;
         self.details.accumulate(&other.details);
     }
 }