Server: Fix issue with PULL mode connection status

AgustinSRG · Nov 23, 2024 · 99c10c2 · 99c10c2
1 parent b1bead9
commit 99c10c2
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/server/connection.go b/server/connection.go
@@ -436,6 +436,10 @@ func (ch *ConnectionHandler) HandlePull(msg *WebsocketProtocolMessage) bool {
 			// Pull
 			go ch.PullFromHlsSource(source, ch.pullingInterruptChannel, maxInitialFragments)
 
+			// Switch mode
+			ch.streamId = streamId
+			ch.mode = CONNECTION_MODE_PULL
+
 			return true
 		}
 	}
@@ -452,6 +456,10 @@ func (ch *ConnectionHandler) HandlePull(msg *WebsocketProtocolMessage) bool {
 			// Pull
 			go ch.PullFromHlsRelay(relay, ch.pullingInterruptChannel, maxInitialFragments)
 
+			// Switch mode
+			ch.streamId = streamId
+			ch.mode = CONNECTION_MODE_PULL
+
 			return true
 		}
 	}