Remove indirection in initial sync completion

[rapid-gossip-sync-server] / src / tracking.rs

diff --git a/src/tracking.rs b/src/tracking.rs
index 672b75f3764fa5113afd0a98fa72c94a29f451f3..f76db3989815320c28a96906d15d488a5012e375 100644 (file)
--- a/src/tracking.rs
+++ b/src/tracking.rs
@@ -16,7 +16,9 @@ use crate::{config, TestLogger};
 use crate::downloader::GossipRouter;
 use crate::types::{GossipMessage, GossipPeerManager};
 
-pub(crate) async fn download_gossip(persistence_sender: mpsc::Sender<GossipMessage>, network_graph: Arc<NetworkGraph<TestLogger>>) {
+pub(crate) async fn download_gossip(persistence_sender: mpsc::Sender<GossipMessage>,
+               completion_sender: mpsc::Sender<()>,
+               network_graph: Arc<NetworkGraph<TestLogger>>) {
        let mut key = [0; 32];
        let mut random_data = [0; 32];
        thread_rng().fill_bytes(&mut key);
@@ -114,7 +116,7 @@ pub(crate) async fn download_gossip(persistence_sender: mpsc::Sender<GossipMessa
 
                        if needs_to_notify_persister {
                                needs_to_notify_persister = false;
-                               persistence_sender.send(GossipMessage::InitialSyncComplete).await.unwrap();
+                               completion_sender.send(()).await.unwrap();
                        }
                }
        });