From 55b2412a8a38f8be3bf328ffa114f1a2f20af252 Mon Sep 17 00:00:00 2001
From: Ian Gulliver <ian@flamingcow.tv>
Date: Fri, 21 May 2021 16:22:27 +0000
Subject: [PATCH] Be consistent with other loop break

---
 main.go | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index 48ff1e3..590f4f0 100644
--- a/main.go
+++ b/main.go
@@ -377,13 +377,16 @@ func present(w http.ResponseWriter, r *http.Request) {
 			delete(room.present, controlChan)
 			close(controlChan)
 			mu.Unlock()
-			return
 
 		case <-ticker.C:
 			writePresentHeartbeat(w, flusher)
 
-		case ctrl := <-controlChan:
-			writePresentEvent(ctrl, w, flusher)
+		case ctrl, ok := <-controlChan:
+			if ok {
+				writePresentEvent(ctrl, w, flusher)
+			} else {
+				return
+			}
 		}
 	}
 }