DRAW-347 feat: 웹소켓 ping manager 구현

Author: hwangsunwoong

app/websocket/src/main/kotlin/com/xorker/draw/websocket/session/PingManager.kt

@@ -0,0 +1,36 @@
+package com.xorker.draw.websocket.session
+
+import com.xorker.draw.websocket.SessionEventListener
+import com.xorker.draw.websocket.WaitingQueueUseCase
+import java.util.concurrent.TimeUnit
+import org.springframework.scheduling.annotation.Scheduled
+import org.springframework.stereotype.Component
+
+@Component
+internal class PingManager(
+    private val sessionManager: SessionManager,
+    private val waitingQueueUseCase: WaitingQueueUseCase,
+    private val sessionEventListener: List<SessionEventListener>,
+) {
+
+    @Scheduled(fixedRate = 3000, timeUnit = TimeUnit.MILLISECONDS)
+    fun checkSessionStatus() {
+        val sessions = sessionManager.getSessions()
+
+        sessions.forEach { session ->
+            if (session.ping.not()) {
+                val user = session.user
+
+                sessionManager.unregisterSession(session.id)
+
+                waitingQueueUseCase.remove(user, session.locale)
+
+                sessionEventListener.forEach {
+                    it.exitSession(user.id)
+                }
+            } else {
+                session.ping = false
+            }
+        }
+    }
+}

app/websocket/src/main/kotlin/com/xorker/draw/websocket/session/SessionManager.kt

@@ -2,12 +2,9 @@ package com.xorker.draw.websocket.session
 
 import com.xorker.draw.user.UserId
 import java.util.concurrent.ConcurrentHashMap
-import org.springframework.core.Ordered
-import org.springframework.core.annotation.Order
-import org.springframework.stereotype.Service
+import org.springframework.stereotype.Component
 
-@Order(Ordered.HIGHEST_PRECEDENCE)
-@Service
+@Component
 internal class SessionManager {
     private val sessionMap: ConcurrentHashMap<SessionId, Session> = ConcurrentHashMap()
     private val userIdMap: ConcurrentHashMap<UserId, Session> = ConcurrentHashMap()
@@ -20,6 +17,12 @@ internal class SessionManager {
         return userIdMap[userId]
     }
 
+    fun getSessions(): List<Session> {
+        val sessions = sessionMap.values
+
+        return sessions.toList()
+    }
+
     fun registerSession(session: Session) {
         if (sessionMap.contains(session.id)) {
             // Init을 중복으로 호출 하면 기존 데이터를 Unregister 하고 Init 한다.
@@ -34,4 +37,12 @@ internal class SessionManager {
         val session = sessionMap.remove(sessionId)
         return userIdMap.remove(session?.user?.id)
     }
+
+    fun setPing(sessionId: SessionId) {
+        val session = sessionMap[sessionId]
+
+        session?.let {
+            session.ping = true
+        }
+    }
 }