Small tweak to make it more obvious when feeds are disconnected

This commit is contained in:
James Wilson
2021-08-04 10:52:53 +01:00
parent db5284a3db
commit 7d00d72baa
+5 -3
View File
@@ -132,7 +132,7 @@ async fn run_soak_test(opts: SoakTestOpts) {
// Start sending "update" messages from nodes at time intervals. // Start sending "update" messages from nodes at time intervals.
let bytes_in = Arc::new(AtomicUsize::new(0)); let bytes_in = Arc::new(AtomicUsize::new(0));
let bytes_in2 = Arc::clone(&bytes_in); let bytes_in2 = Arc::clone(&bytes_in);
let send_handle = tokio::task::spawn(async move { tokio::task::spawn(async move {
let msg = json!({ let msg = json!({
"id":1, "id":1,
"payload":{ "payload":{
@@ -175,6 +175,7 @@ async fn run_soak_test(opts: SoakTestOpts) {
bytes_out.fetch_add(num_bytes, Ordering::Relaxed); bytes_out.fetch_add(num_bytes, Ordering::Relaxed);
msgs_out.fetch_add(1, Ordering::Relaxed); msgs_out.fetch_add(1, Ordering::Relaxed);
} }
eprintln!("Error: feed has been closed unexpectedly");
}); });
} }
@@ -209,8 +210,8 @@ async fn run_soak_test(opts: SoakTestOpts) {
} }
}); });
// Wait for sending to finish before ending. // Wait forever.
send_handle.await.unwrap(); future::pending().await
} }
/// Identical to `soak_test`, except that we try to send realistic messages from fake nodes. /// Identical to `soak_test`, except that we try to send realistic messages from fake nodes.
@@ -318,6 +319,7 @@ async fn run_realistic_soak_test(opts: SoakTestOpts) {
bytes_out.fetch_add(num_bytes, Ordering::Relaxed); bytes_out.fetch_add(num_bytes, Ordering::Relaxed);
msgs_out.fetch_add(1, Ordering::Relaxed); msgs_out.fetch_add(1, Ordering::Relaxed);
} }
eprintln!("Error: feed has been closed unexpectedly");
}); });
} }