stop session garbage collector when lobby done

shigde · Mar 11, 2024 · 489967f · 489967f
1 parent 282f888
commit 489967f
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/internal/lobby/lobby.go b/internal/lobby/lobby.go
@@ -51,9 +51,16 @@ func newLobby(id uuid.UUID, entity *LobbyEntity) *lobby {
 		sessionGarbageCollector: sessionGarbageCollector,
 	}
 	go func() {
-		for id := range sessionGarbageCollector {
-			if ok := lobby.sessions.Delete(id); !ok {
-				slog.Warn("session could not delete", "session id", id)
+		for {
+			select {
+			case sessionId := <-sessionGarbageCollector:
+
+				if ok := lobby.sessions.Delete(sessionId); !ok {
+					slog.Warn("session could not delete", "session id", sessionId)
+				}
+			case <-lobby.ctx.Done():
+				slog.Debug("stop session garbage collector")
+				return
 			}
 		}
 	}()