Use new variant to return entry count

solana-labs · Sep 21, 2023 · 6fc2a3f · 6fc2a3f
1 parent e0603fe
commit 6fc2a3f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/core/src/replay_stage.rs b/core/src/replay_stage.rs
@@ -2927,6 +2927,7 @@ impl ReplayStage {
                         Some(bank.clock().unix_timestamp),
                         Some(bank.block_height()),
                         bank.executed_transaction_count(),
+                        r_replay_progress.num_entries as u64,
                     )
                 }
                 bank_complete_time.stop();

diff --git a/geyser-plugin-manager/src/block_metadata_notifier.rs b/geyser-plugin-manager/src/block_metadata_notifier.rs
@@ -6,7 +6,7 @@ use {
     log::*,
     solana_accounts_db::stake_rewards::RewardInfo,
     solana_geyser_plugin_interface::geyser_plugin_interface::{
-        ReplicaBlockInfoV2, ReplicaBlockInfoVersions,
+        ReplicaBlockInfoV3, ReplicaBlockInfoVersions,
     },
     solana_measure::measure::Measure,
     solana_metrics::*,
@@ -31,6 +31,7 @@ impl BlockMetadataNotifier for BlockMetadataNotifierImpl {
         block_time: Option<UnixTimestamp>,
         block_height: Option<u64>,
         executed_transaction_count: u64,
+        entry_count: u64,
     ) {
         let plugin_manager = self.plugin_manager.read().unwrap();
         if plugin_manager.plugins.is_empty() {
@@ -49,8 +50,9 @@ impl BlockMetadataNotifier for BlockMetadataNotifierImpl {
                 block_time,
                 block_height,
                 executed_transaction_count,
+                entry_count,
             );
-            let block_info = ReplicaBlockInfoVersions::V0_0_2(&block_info);
+            let block_info = ReplicaBlockInfoVersions::V0_0_3(&block_info);
             match plugin.notify_block_metadata(block_info) {
                 Err(err) => {
                     error!(
@@ -103,8 +105,9 @@ impl BlockMetadataNotifierImpl {
         block_time: Option<UnixTimestamp>,
         block_height: Option<u64>,
         executed_transaction_count: u64,
-    ) -> ReplicaBlockInfoV2<'a> {
-        ReplicaBlockInfoV2 {
+        entry_count: u64,
+    ) -> ReplicaBlockInfoV3<'a> {
+        ReplicaBlockInfoV3 {
             parent_slot,
             parent_blockhash,
             slot,
@@ -113,6 +116,7 @@ impl BlockMetadataNotifierImpl {
             block_time,
             block_height,
             executed_transaction_count,
+            entry_count,
         }
     }
 

diff --git a/geyser-plugin-manager/src/block_metadata_notifier_interface.rs b/geyser-plugin-manager/src/block_metadata_notifier_interface.rs
@@ -17,6 +17,7 @@ pub trait BlockMetadataNotifier {
         block_time: Option<UnixTimestamp>,
         block_height: Option<u64>,
         executed_transaction_count: u64,
+        entry_count: u64,
     );
 }