From 6cc49de2ed75eb4a17a44e5dccc0eb41c7b100f7 Mon Sep 17 00:00:00 2001
From: "Spencer C. Imbleau" <spencer@imbleau.com>
Date: Sat, 6 Jan 2024 18:12:26 -0500
Subject: [PATCH] fix: issue 391

---
 matchbox_socket/src/webrtc_socket/socket.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/matchbox_socket/src/webrtc_socket/socket.rs b/matchbox_socket/src/webrtc_socket/socket.rs
index cc442b6c..11fbe0d0 100644
--- a/matchbox_socket/src/webrtc_socket/socket.rs
+++ b/matchbox_socket/src/webrtc_socket/socket.rs
@@ -777,7 +777,7 @@ async fn run_socket(
         select! {
             msgloop = message_loop_done => {
                 match msgloop {
-                    Ok(()) => {
+                    Ok(()) | Err(SignalingError::StreamExhausted) => {
                         debug!("Message loop completed");
                         break Ok(())
                     },
@@ -792,6 +792,10 @@ async fn run_socket(
             sigloop = signaling_loop_done => {
                 match sigloop {
                     Ok(()) => debug!("Signaling loop completed"),
+                    Err(SignalingError::StreamExhausted) => {
+                        debug!("Signaling loop completed");
+                        break Ok(());
+                    },
                     Err(e) => {
                         // TODO: Reconnect X attempts if configured to reconnect.
                         error!("The signaling loop finished with an error: {e:?}");