From 73efb1437c0ca8b5fbdf135597d0e468f41a3946 Mon Sep 17 00:00:00 2001
From: aler9 <46489434+aler9@users.noreply.github.com>
Date: Sat, 13 Mar 2021 15:14:20 +0100
Subject: [PATCH] fix readers removal in examples (#26)

---
 examples/server-tls/main.go | 2 ++
 examples/server-udp/main.go | 2 ++
 examples/server/main.go     | 2 ++
 serverconf_test.go          | 2 ++
 4 files changed, 8 insertions(+)

diff --git a/examples/server-tls/main.go b/examples/server-tls/main.go
index db672d2f..dc832a51 100644
--- a/examples/server-tls/main.go
+++ b/examples/server-tls/main.go
@@ -144,6 +144,8 @@ func handleConn(conn *gortsplib.ServerConn) {
 	if conn == publisher {
 		publisher = nil
 		sdp = nil
+	} else {
+		delete(readers, conn)
 	}
 }
 
diff --git a/examples/server-udp/main.go b/examples/server-udp/main.go
index 91d61c54..b5501b9e 100644
--- a/examples/server-udp/main.go
+++ b/examples/server-udp/main.go
@@ -143,6 +143,8 @@ func handleConn(conn *gortsplib.ServerConn) {
 	if conn == publisher {
 		publisher = nil
 		sdp = nil
+	} else {
+		delete(readers, conn)
 	}
 }
 
diff --git a/examples/server/main.go b/examples/server/main.go
index 318903a1..45b95f2b 100644
--- a/examples/server/main.go
+++ b/examples/server/main.go
@@ -143,6 +143,8 @@ func handleConn(conn *gortsplib.ServerConn) {
 	if conn == publisher {
 		publisher = nil
 		sdp = nil
+	} else {
+		delete(readers, conn)
 	}
 }
 
diff --git a/serverconf_test.go b/serverconf_test.go
index a1589e91..04388054 100644
--- a/serverconf_test.go
+++ b/serverconf_test.go
@@ -287,6 +287,8 @@ func (ts *testServ) handleConn(conn *ServerConn) {
 	if conn == ts.publisher {
 		ts.publisher = nil
 		ts.sdp = nil
+	} else {
+		delete(ts.readers, conn)
 	}
 }