Use time.AfterFunc to schedule room termination

olzhasar · Mar 16, 2024 · 1a1f86f · 1a1f86f
1 parent 0df9ef8
commit 1a1f86f
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/backend/pkg/chat/hub.go b/backend/pkg/chat/hub.go
@@ -175,12 +175,13 @@ func (h *Hub) RoomCount() int {
 
 func (h *Hub) scheduleRoomTermination(room *Room) {
 	go func() {
-		time.Sleep(EMPTY_ROOM_TIMEOUT)
-		if room.ClientCount() > 0 || h.rooms[room.ID] == nil {
-			return
-		}
-		delete(h.rooms, room.ID)
-		metrics.RoomCount.Dec()
+		time.AfterFunc(EMPTY_ROOM_TIMEOUT, func() {
+			if room.ClientCount() > 0 || h.rooms[room.ID] == nil {
+				return
+			}
+			delete(h.rooms, room.ID)
+			metrics.RoomCount.Dec()
+		})
 	}()
 }