Allow custom logger types.

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

diff --git a/src/tracking.rs b/src/tracking.rs
index af24055151bd4d52df8d32fe363ac487899eedc7..2935eb286780b8228d1d4f6123de370e71b3cbb1 100644 (file)
--- a/src/tracking.rs
+++ b/src/tracking.rs
@@ -1,41 +1,66 @@
+use std::collections::hash_map::RandomState;
+use std::hash::{BuildHasher, Hasher};
 use std::net::SocketAddr;
 use std::sync::Arc;
 use std::time::{Duration, Instant};
 
 use bitcoin::hashes::hex::ToHex;
-use bitcoin::secp256k1::{PublicKey, SecretKey};
+use bitcoin::secp256k1::PublicKey;
 use lightning;
 use lightning::ln::peer_handler::{
        ErroringMessageHandler, IgnoringMessageHandler, MessageHandler, PeerManager,
 };
 use lightning::routing::gossip::NetworkGraph;
-use rand::{Rng, thread_rng};
+use lightning::sign::KeysManager;
+use lightning::util::logger::Logger;
 use tokio::sync::mpsc;
 
-use crate::{config, TestLogger};
+use crate::config;
 use crate::downloader::GossipRouter;
 use crate::types::{GossipMessage, GossipPeerManager};
 
-pub(crate) async fn download_gossip(persistence_sender: mpsc::Sender<GossipMessage>, network_graph: Arc<NetworkGraph<Arc<TestLogger>>>) {
-       let mut key = [0; 32];
-       let mut random_data = [0; 32];
-       thread_rng().fill_bytes(&mut key);
-       thread_rng().fill_bytes(&mut random_data);
-       let our_node_secret = SecretKey::from_slice(&key).unwrap();
-
-       let router = Arc::new(GossipRouter::new(network_graph, persistence_sender.clone()));
+pub(crate) async fn download_gossip<L: Logger + Send + Sync + 'static>(persistence_sender: mpsc::Sender<GossipMessage>,
+               completion_sender: mpsc::Sender<()>,
+               network_graph: Arc<NetworkGraph<Arc<L>>>,
+               logger: Arc<L>
+) {
+       let mut key = [42; 32];
+       let mut random_data = [43; 32];
+       // Get something psuedo-random from std.
+       let mut key_hasher = RandomState::new().build_hasher();
+       key_hasher.write_u8(1);
+       key[0..8].copy_from_slice(&key_hasher.finish().to_ne_bytes());
+       let mut rand_hasher = RandomState::new().build_hasher();
+       rand_hasher.write_u8(2);
+       random_data[0..8].copy_from_slice(&rand_hasher.finish().to_ne_bytes());
+
+       let keys_manager = Arc::new(KeysManager::new(&key, 0xdeadbeef, 0xdeadbeef));
+
+       let router = Arc::new(GossipRouter::new(network_graph, persistence_sender.clone(), logger.clone()));
 
        let message_handler = MessageHandler {
                chan_handler: ErroringMessageHandler::new(),
                route_handler: Arc::clone(&router),
+               onion_message_handler: IgnoringMessageHandler {},
+               custom_message_handler: IgnoringMessageHandler {},
        };
        let peer_handler = Arc::new(PeerManager::new(
                message_handler,
-               our_node_secret,
+               0xdeadbeef,
                &random_data,
-               Arc::new(TestLogger::new()),
-               IgnoringMessageHandler {},
+               logger,
+               keys_manager,
        ));
+       router.set_pm(Arc::clone(&peer_handler));
+
+       let ph_timer = Arc::clone(&peer_handler);
+       tokio::spawn(async move {
+               let mut intvl = tokio::time::interval(Duration::from_secs(10));
+               loop {
+                       intvl.tick().await;
+                       ph_timer.timer_tick_occurred();
+               }
+       });
 
        println!("Connecting to Lightning peers...");
        let peers = config::ln_peers();
@@ -114,13 +139,13 @@ 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();
                        }
                }
        });
 }
 
-async fn connect_peer(current_peer: (PublicKey, SocketAddr), peer_manager: GossipPeerManager) -> bool {
+async fn connect_peer<L: Logger + Send + Sync + 'static>(current_peer: (PublicKey, SocketAddr), peer_manager: GossipPeerManager<L>) -> bool {
        eprintln!("Connecting to peer {}@{}...", current_peer.0.to_hex(), current_peer.1.to_string());
        let connection = lightning_net_tokio::connect_outbound(
                Arc::clone(&peer_manager),
@@ -139,9 +164,8 @@ async fn connect_peer(current_peer: (PublicKey, SocketAddr), peer_manager: Gossi
                                        current_peer.1,
                                ).await {
                                        disconnection_future.await;
-                               } else {
-                                       tokio::time::sleep(Duration::from_secs(10)).await;
                                }
+                               tokio::time::sleep(Duration::from_secs(10)).await;
                        }
                });
                true