fix duration for small network sizees

broker/src/lib.rs

@@ -377,6 +377,19 @@ where
 
         info!("received connection from user {:?}", connection_id.data());
 
+        // If we have a small amount of users, send the updates immediately
+        if get_lock!(inner.user_connection_lookup, read).get_connection_count() < 50 {
+            // TODO NEXT: Move this into just asking the task nicely to do it
+            let _ = inner
+                .send_updates_to_brokers(
+                    vec![],
+                    get_lock!(inner.broker_connection_lookup, read)
+                        .get_all_connections()
+                        .clone(),
+                )
+                .await;
+        }
+
         // This runs the main loop for receiving information from the user
         let () = inner.user_recv_loop(connection_id, receiver).await;
 

client/src/main.rs

@@ -64,10 +64,7 @@ async fn main() -> Result<()> {
         })
         .await?,
     );
-
-    // Wait 5 seconds so broker has time to propagate info about us
-    sleep(Duration::from_secs(5)).await;
-
+
     // Run our first client, which sends a message to our second.
     let client1 = spawn(async move {
         // Clone our client