chore: refactor after review

Signed-off-by: Simon Paitrault <simon.paitrault@gmail.com>
topos-protocol · Mar 12, 2024 · ae3e1b5 · ae3e1b5
1 parent e459b6a
commit ae3e1b5
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 5 deletions.
diff --git a/crates/topos-tce-api/src/runtime/error.rs b/crates/topos-tce-api/src/runtime/error.rs
@@ -16,4 +16,7 @@ pub enum RuntimeError {
 
     #[error("Unexpected store error: {0}")]
     Store(#[from] StorageError),
+
+    #[error("Communication error: {0}")]
+    CommunicationError(String),
 }
diff --git a/crates/topos-tce-storage/src/validator/mod.rs b/crates/topos-tce-storage/src/validator/mod.rs
@@ -139,6 +139,8 @@ impl ValidatorStore {
     }
 
     /// Returns the number of certificates in the pending pool (by iterating)
+    ///
+    /// Performance can be an issue on this one as we iter over all the pending certificates.
     pub fn iter_count_pending_certificates(&self) -> Result<u64, StorageError> {
         Ok(self
             .pending_tables
@@ -158,6 +160,8 @@ impl ValidatorStore {
     }
 
     /// Returns the number of certificates in the precedence pool (by iterating)
+    ///
+    /// Performance can be an issue on this one as we iter over all the precedence certificates.
     pub fn iter_count_precedence_pool_certificates(&self) -> Result<u64, StorageError> {
         Ok(self
             .pending_tables

diff --git a/crates/topos-tce/src/app_context/api.rs b/crates/topos-tce/src/app_context/api.rs
@@ -30,17 +30,30 @@ impl AppContext {
                             certificate.id, certificate.source_subnet_id
                         );
 
-                        _ = self
+                        if let Err(_) = self
                             .tce_cli
                             .get_double_echo_channel()
                             .send(DoubleEchoCommand::Broadcast {
                                 need_gossip: true,
                                 cert: *certificate,
                                 pending_id,
                             })
-                            .await;
+                            .await
+                        {
+                            error!(
+                                "Unable to send DoubleEchoCommand::Broadcast command to double \
+                                 echo for {}",
+                                certificate.id
+                            );
 
-                        sender.send(Ok(PendingResult::InPending(pending_id)))
+                            sender.send(Err(RuntimeError::CommunicationError(
+                                "Unable to send DoubleEchoCommand::Broadcast command to double \
+                                 echo"
+                                    .to_string(),
+                            )))
+                        } else {
+                            sender.send(Ok(PendingResult::InPending(pending_id)))
+                        }
                     }
                     Ok(None) => {
                         debug!(

diff --git a/crates/topos-tce/src/app_context/network.rs b/crates/topos-tce/src/app_context/network.rs
@@ -48,15 +48,22 @@ impl AppContext {
                                     cert.id
                                 );
 
-                                _ = self
+                                if let Err(_) = self
                                     .tce_cli
                                     .get_double_echo_channel()
                                     .send(DoubleEchoCommand::Broadcast {
                                         need_gossip: false,
                                         cert,
                                         pending_id,
                                     })
-                                    .await;
+                                    .await
+                                {
+                                    error!(
+                                        "Unable to send DoubleEchoCommand::Broadcast command to \
+                                         double echo for {}",
+                                        certificate.id
+                                    );
+                                }
                             }
                             Ok(None) => {
                                 debug!(