From 41afaed2c17b93ef1fc320ad2ed9814ee3bdc439 Mon Sep 17 00:00:00 2001
From: Philipp Oppermann <dev@phil-opp.com>
Date: Thu, 28 Jul 2022 15:06:35 +0200
Subject: [PATCH] Add some more log messages

---
 apis/rust/node/src/lib.rs                       | 17 +++++++++++++----
 .../examples/nodes/rust/rate_limit.rs           |  2 ++
 2 files changed, 15 insertions(+), 4 deletions(-)

diff --git a/apis/rust/node/src/lib.rs b/apis/rust/node/src/lib.rs
index 21aa79ef..168611c6 100644
--- a/apis/rust/node/src/lib.rs
+++ b/apis/rust/node/src/lib.rs
@@ -1,7 +1,7 @@
 use communication::CommunicationLayer;
 use config::{CommunicationConfig, DataId, NodeId, NodeRunConfig};
 use eyre::WrapErr;
-use futures::{stream::FuturesUnordered, StreamExt};
+use futures::{stream::FuturesUnordered, FutureExt, StreamExt};
 use futures_concurrency::Merge;
 use std::collections::HashSet;
 
@@ -84,7 +84,12 @@ impl DoraNode {
                 .wrap_err_with(|| format!("failed to subscribe on {topic}"))?;
             stop_messages.push(sub.into_future());
         }
-        let finished = Box::pin(stop_messages.all(|_| async { true }));
+        let node_id = self.id.clone();
+        let finished = Box::pin(
+            stop_messages
+                .all(|_| async { true })
+                .map(move |_| println!("all inputs finished for node {node_id}")),
+        );
 
         Ok(streams.merge().take_until(finished))
     }
@@ -121,8 +126,12 @@ impl Drop for DoraNode {
             .communication
             .publish_sync(&topic, &[])
             .wrap_err_with(|| format!("failed to send stop message for source `{self_id}`"));
-        if let Err(err) = result {
-            tracing::error!("{err}")
+        match result {
+            Ok(()) => println!("sent stop message for {self_id}"),
+            Err(err) => {
+                println!("error sending stop message for {self_id}: {err:?}");
+                tracing::error!("{err:?}")
+            }
         }
     }
 }
diff --git a/binaries/coordinator/examples/nodes/rust/rate_limit.rs b/binaries/coordinator/examples/nodes/rust/rate_limit.rs
index 4b581f13..3b3c2cea 100644
--- a/binaries/coordinator/examples/nodes/rust/rate_limit.rs
+++ b/binaries/coordinator/examples/nodes/rust/rate_limit.rs
@@ -46,5 +46,7 @@ async fn main() -> eyre::Result<()> {
         }
     }
 
+    println!("rate limit finished");
+
     Ok(())
 }