Refactor: KEYS 대신 Redis 카운터를 사용하여 온라인 유저 수 조회 성능 개선

Author: jueunk617

src/main/java/com/back/global/websocket/config/WebSocketConstants.java

@@ -46,6 +46,13 @@ private WebSocketConstants() {
     public static final String ROOM_USERS_KEY_PREFIX = "ws:room:";
     public static final String ROOM_USERS_KEY_SUFFIX = ":users";
 
+    /**
+     * 전체 온라인 사용자 수 저장 Key
+     * - 패턴: ws:online_users:count
+     * - 값: Long (카운트)
+     */
+    public static final String ONLINE_USER_COUNT_KEY = "ws:online_users:count";
+
     // ===== Key 빌더 헬퍼 메서드 =====
 
     public static String buildUserSessionKey(Long userId) {

src/main/java/com/back/global/websocket/service/UserSessionService.java

@@ -34,6 +34,8 @@ public void registerSession(Long userId, String sessionId) {
         redisSessionStore.saveUserSession(userId, newSession);
         redisSessionStore.saveSessionUserMapping(sessionId, userId);
 
+        redisSessionStore.incrementOnlineUserCount();
+
         log.info("WebSocket 세션 등록 완료 - 사용자: {}, 세션: {}", userId, sessionId);
     }
 
@@ -44,6 +46,9 @@ public void terminateSession(String sessionId) {
         if (userId != null) {
             redisSessionStore.deleteUserSession(userId);
             redisSessionStore.deleteSessionUserMapping(sessionId);
+
+            redisSessionStore.decrementOnlineUserCount();
+
             log.info("WebSocket 세션 종료 완료 - 세션: {}, 사용자: {}", sessionId, userId);
         } else {
             log.warn("종료할 세션을 찾을 수 없음 - 세션: {}", sessionId);

src/main/java/com/back/global/websocket/store/RedisSessionStore.java

@@ -182,11 +182,35 @@ public long getRoomUserCount(Long roomId) {
 
     public long getTotalOnlineUserCount() {
         try {
-            Set<String> userKeys = redisTemplate.keys(WebSocketConstants.buildUserSessionKeyPattern());
-            return userKeys != null ? userKeys.size() : 0;
+            // 카운터 키에서 직접 값을 가져옴
+            Object count = redisTemplate.opsForValue().get(WebSocketConstants.ONLINE_USER_COUNT_KEY);
+            if (count instanceof Number) {
+                return ((Number) count).longValue();
+            }
+            return 0L;
         } catch (Exception e) {
             log.error("전체 온라인 사용자 수 조회 실패", e);
-            return 0;
+            return 0; // 에러 발생 시 0 반환
+        }
+    }
+
+    public void incrementOnlineUserCount() {
+        try {
+            redisTemplate.opsForValue().increment(WebSocketConstants.ONLINE_USER_COUNT_KEY);
+        } catch (Exception e) {
+            log.error("온라인 사용자 수 증가 실패", e);
+        }
+    }
+
+    public void decrementOnlineUserCount() {
+        try {
+            // 카운터가 0보다 작아지지 않도록 방지
+            Long currentValue = redisTemplate.opsForValue().decrement(WebSocketConstants.ONLINE_USER_COUNT_KEY);
+            if (currentValue != null && currentValue < 0) {
+                redisTemplate.opsForValue().set(WebSocketConstants.ONLINE_USER_COUNT_KEY, 0L);
+            }
+        } catch (Exception e) {
+            log.error("온라인 사용자 수 감소 실패", e);
         }
     }