vm: notify engine of new block

Signed-off-by: Sam Batschelet <[email protected]>

Author: hexfusion

mini-kvvm/src/chain/vm.rs

@@ -14,4 +14,5 @@ pub struct Context {
 pub trait Vm: avalanche_types::rpcchainvm::vm::Vm {
     async fn is_bootstrapped(&self) -> bool;
     async fn submit(&self, txs: Vec<Transaction>) -> Result<()>;
+    async fn notify_block_ready(&self);
 }

mini-kvvm/src/vm.rs

@@ -21,7 +21,7 @@ use crate::{
     api,
     block::Block as StatefulBlock,
     block::{self, state::State},
-    chain::{self, tx::Transaction},
+    chain::{self, tx::Transaction, vm::Vm},
     genesis::Genesis,
 };
 
@@ -133,6 +133,25 @@ impl crate::chain::vm::Vm for ChainVm {
         }
         Ok(())
     }
+
+    /// Sends a signal to the consensus engine that a new block
+    /// is ready to be created.
+    async fn notify_block_ready(&self) {
+        let vm = self.inner.read().await;
+
+        if let Some(engine) = &vm.to_engine {
+            if let Err(_) = engine
+                .send(rpcchainvm::common::message::Message::PendingTxs)
+                .await
+            {
+                log::warn!("dropping message to consensus engine");
+            };
+            return;
+        }
+
+        log::error!("consensus engine channel failed to initialized");
+        return;
+    }
 }
 
 #[tonic::async_trait]
@@ -466,6 +485,8 @@ impl rpcchainvm::snowman::block::ChainVm for ChainVm {
             .await
             .map_err(|e| Error::new(ErrorKind::Other, e.to_string()))?;
 
+        self.notify_block_ready().await;
+
         Ok(Box::new(block))
     }