Allow publishers to re-connect

Resolves #134
Glimesh · Jun 4, 2024 · c1a00a1 · c1a00a1
1 parent 4a3b4a1
commit c1a00a1
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/internal/webrtc/webrtc.go b/internal/webrtc/webrtc.go
@@ -122,15 +122,18 @@ func peerConnectionDisconnected(streamKey string, whepSessionId string) {
 		return
 	}
 
+	stream.whepSessionsLock.Lock()
+	defer stream.whepSessionsLock.Unlock()
+
 	if whepSessionId != "" {
-		stream.whepSessionsLock.Lock()
-		defer stream.whepSessionsLock.Unlock()
 		delete(stream.whepSessions, whepSessionId)
+	} else {
+		stream.hasWHIPClient.Store(false)
+	}
 
-		// Only delete stream if all WHEP Sessions are gone and have no WHIP Client
-		if len(stream.whepSessions) != 0 || stream.hasWHIPClient.Load() {
-			return
-		}
+	// Only delete stream if all WHEP Sessions are gone and have no WHIP Client
+	if len(stream.whepSessions) != 0 || stream.hasWHIPClient.Load() {
+		return
 	}
 
 	stream.whipActiveContextCancel()