chore: adding republish_count

Signed-off-by: Simon Paitrault <[email protected]>

crates/topos-tce-api/tests/runtime.rs

@@ -717,7 +717,9 @@ async fn get_pending_pool(
         create_validator_store(vec![], futures::future::ready(fullnode_store.clone())).await;
 
     for certificate in &certificates {
-        _ = store.insert_pending_certificate(&certificate.certificate);
+        _ = store
+            .insert_pending_certificate(&certificate.certificate)
+            .await;
     }
 
     let storage_client = StorageClient::new(store.clone());
@@ -795,7 +797,9 @@ async fn check_precedence(
         create_validator_store(vec![], futures::future::ready(fullnode_store.clone())).await;
 
     for certificate in &certificates {
-        _ = store.insert_pending_certificate(&certificate.certificate);
+        _ = store
+            .insert_pending_certificate(&certificate.certificate)
+            .await;
     }
 
     let storage_client = StorageClient::new(store.clone());

crates/topos-tce-broadcast/src/double_echo/broadcast_state.rs

@@ -32,6 +32,7 @@ pub struct BroadcastState {
     readies: HashSet<Ready>,
     pub(crate) expected_position: Option<Position>,
     republish_count: u64,
+    need_gossip: bool,
 }
 
 impl BroadcastState {
@@ -61,6 +62,7 @@ impl BroadcastState {
             readies: HashSet::new(),
             expected_position: None,
             republish_count: 0,
+            need_gossip,
         };
 
         _ = state.event_sender.try_send(ProtocolEvents::Broadcast {
@@ -85,20 +87,23 @@ impl BroadcastState {
     }
 
     pub(crate) fn republish_certificate(&mut self) {
-        if let Some(count) = self.republish_count.checked_add(1) {
-            self.republish_count = count;
-
-            // TODO: catch error
-            let _ = self.event_sender.try_send(ProtocolEvents::Gossip {
-                cert: self.certificate.clone(),
-                republish_count: self.republish_count,
-            });
+        if self.need_gossip {
+            if let Some(count) = self.republish_count.checked_add(1) {
+                self.republish_count = count;
+
+                // TODO: catch error
+                let _ = self.event_sender.try_send(ProtocolEvents::Gossip {
+                    cert: self.certificate.clone(),
+                    republish_count: self.republish_count,
+                });
+            }
+        }
 
-            self.status = Status::Pending;
+        self.status = Status::Pending;
 
-            self.update_status();
-        }
+        self.update_status();
     }
+
     pub fn into_delivered(&self) -> CertificateDelivered {
         CertificateDelivered {
             certificate: self.certificate.clone(),

crates/topos-tce-broadcast/src/task_manager/task.rs

@@ -10,7 +10,7 @@ use topos_tce_storage::errors::StorageError;
 use topos_tce_storage::store::{ReadStore, WriteStore};
 use topos_tce_storage::types::CertificateDeliveredWithPositions;
 use topos_tce_storage::validator::ValidatorStore;
-use tracing::{error, info, warn};
+use tracing::{debug, error};
 
 use crate::double_echo::broadcast_state::{BroadcastState, Status};
 use crate::{DoubleEchoCommand, TaskStatus};
@@ -99,6 +99,9 @@ impl IntoFuture for Task {
 
             let mut gossip_timer = tokio::time::interval(Duration::from_secs(5));
             gossip_timer.set_missed_tick_behavior(tokio::time::MissedTickBehavior::Skip);
+            // First tick is instant
+            gossip_timer.tick().await;
+
             loop {
                 tokio::select! {
                     _tick = gossip_timer.tick(), if !self.broadcast_state.received_echo() => {