DRAW-259 fix: 대기 큐 exit 오류 수정

YAPP-Github · Aug 24, 2024 · 584dc8a · 584dc8a
1 parent 73e2bc6
commit 584dc8a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/app/websocket/src/main/kotlin/com/xorker/draw/websocket/MainWebSocketHandler.kt b/app/websocket/src/main/kotlin/com/xorker/draw/websocket/MainWebSocketHandler.kt
@@ -19,6 +19,7 @@ internal class MainWebSocketHandler(
     private val waitingQueueSessionUseCase: WaitingQueueSessionUseCase,
     private val router: WebSocketRouter,
     private val requestParser: WebSocketRequestParser,
+    private val waitingQueueSessionEventListener: List<WaitingQueueSessionEventListener>,
     private val sessionEventListener: List<SessionEventListener>,
     private val webSocketExceptionHandler: WebSocketExceptionHandler,
     private val metricManager: MetricManager,
@@ -51,7 +52,9 @@ internal class MainWebSocketHandler(
         val waitingQueueSessionDto = waitingQueueSessionUseCase.getSession(SessionId(session.id))
 
         if (waitingQueueSessionDto != null) {
-            waitingQueueSessionUseCase.unregisterSession(SessionId(session.id))
+            waitingQueueSessionEventListener.forEach {
+                it.exitSession(waitingQueueSessionDto)
+            }
             return
         }