WIP

clients/client-core/src/client/real_messages_control/real_traffic_stream.rs

@@ -384,7 +384,10 @@ where
             match Pin::new(&mut self.real_receiver).poll_recv(cx) {
                 // in the case our real message channel stream was closed, we should also indicate we are closed
                 // (and whoever is using the stream should panic)
-                Poll::Ready(None) => Poll::Ready(None),
+                Poll::Ready(None) => {
+                    log::error!("real receiver ready(none)");
+                    Poll::Ready(None)
+                }
 
                 Poll::Ready(Some((real_messages, conn_id))) => {
                     log::trace!("handling real_messages: size: {}", real_messages.len());

common/client-libs/gateway-client/src/socket_state.rs

@@ -100,7 +100,7 @@ impl PartiallyDelegated {
 
             let ret_err = loop {
                 tokio::select! {
-                    _ = shutdown.recv() => {
+                    _ = shutdown.recv_with_delay() => {
                         log::trace!("GatewayClient listener: Received shutdown");
                         log::debug!("GatewayClient listener: Exiting");
                         return;

common/task/src/manager.rs

@@ -254,12 +254,13 @@ impl TaskClient {
     }
 
     pub async fn recv_with_delay(&mut self) {
-        self.recv()
-            .then(|msg| async move {
-                sleep(Duration::from_secs(1)).await;
-                msg
-            })
-            .await
+        pending().await
+        //self.recv()
+        //    .then(|msg| async move {
+        //        sleep(Duration::from_secs(1)).await;
+        //        msg
+        //    })
+        //    .await
     }
 
     pub async fn recv_timeout(&mut self) {