unify request_sync logic

1 files changed, 18 insertions, 0 deletions

diff --git a/src/state/mod.rs b/src/state/mod.rs
index e8975e2..8f3a841 100644
--- a/src/state/mod.rs
+++ b/src/state/mod.rs
@@ -180,6 +180,24 @@ pub(crate) struct CatchUpSource {
 }
 
 impl AppState {
+    /// Asks the sync worker to look at `peer` (`SyncEvent::PeerNeedsSync`): whether to
+    /// send it a sync message now. Nothing if the worker is gone (the node stops).
+    pub(crate) fn request_sync(&self, peer: &str) {
+        let _ = self
+            .sync_events
+            .send(SyncEvent::PeerNeedsSync(peer.to_owned()));
+    }
+
+    /// `request_sync`, after `delay`.
+    pub(crate) fn request_sync_after(&self, peer: &str, delay: std::time::Duration) {
+        let events = self.sync_events.clone();
+        let peer = peer.to_owned();
+        tokio::spawn(async move {
+            tokio::time::sleep(delay).await;
+            let _ = events.send(SyncEvent::PeerNeedsSync(peer));
+        });
+    }
+
     /// Whether the node shuts down (`closing`).
     pub(crate) fn is_closing(&self) -> bool {
         *self.closing.borrow()