Feat: implement event listeners for subscribe, unsubscribe, disconnect events for dealing user's state in session

Author: beomzun

src/main/java/org/example/plzdrawing/common/listener/websocket/ChatDisconnectEventListener.java

@@ -0,0 +1,22 @@
+package org.example.plzdrawing.common.listener.websocket;
+
+import lombok.RequiredArgsConstructor;
+import org.example.plzdrawing.api.chat.repository.ChatPresenceRepository;
+import org.springframework.context.ApplicationListener;
+import org.springframework.messaging.simp.stomp.StompHeaderAccessor;
+import org.springframework.stereotype.Component;
+import org.springframework.web.socket.messaging.SessionDisconnectEvent;
+
+@Component
+@RequiredArgsConstructor
+public class ChatDisconnectEventListener implements ApplicationListener<SessionDisconnectEvent> {
+
+    private final ChatPresenceRepository chatPresenceRepository;
+
+    @Override
+    public void onApplicationEvent(SessionDisconnectEvent event) {
+        StompHeaderAccessor accessor = StompHeaderAccessor.wrap(event.getMessage());
+        Long memberId = Long.parseLong(accessor.getFirstNativeHeader("memberId"));
+        chatPresenceRepository.removeActive(memberId);
+    }
+}

src/main/java/org/example/plzdrawing/common/listener/websocket/ChatSubscriptionEventListener.java

@@ -0,0 +1,34 @@
+package org.example.plzdrawing.common.listener.websocket;
+
+import lombok.RequiredArgsConstructor;
+import org.example.plzdrawing.api.chat.repository.ChatPresenceRepository;
+import org.example.plzdrawing.api.chatRoom.repository.UnreadCountRepository;
+import org.example.plzdrawing.util.websocket.MessagePublisher;
+import org.springframework.context.ApplicationListener;
+import org.springframework.messaging.simp.stomp.StompHeaderAccessor;
+import org.springframework.stereotype.Component;
+import org.springframework.web.socket.messaging.SessionSubscribeEvent;
+
+@Component
+@RequiredArgsConstructor
+public class ChatSubscriptionEventListener implements ApplicationListener<SessionSubscribeEvent> {
+
+    private final ChatPresenceRepository chatPresenceRepository;
+    private final UnreadCountRepository unreadCountRepository;
+    private final MessagePublisher messagePublisher;
+
+    @Override
+    public void onApplicationEvent(SessionSubscribeEvent event) {
+        StompHeaderAccessor accessor = StompHeaderAccessor.wrap(event.getMessage());
+        String destination = accessor.getDestination();
+        if (destination.startsWith("/topic/join/")) {
+            String chatRoomId = destination.substring("/topic/join/".length());
+
+            Long memberId = Long.parseLong(accessor.getFirstNativeHeader("memberId"));
+            chatPresenceRepository.setActive(chatRoomId, memberId);
+            unreadCountRepository.resetUnreadCount(chatRoomId, memberId);
+
+            messagePublisher.removeUnreadCheck(chatRoomId, memberId);
+        }
+    }
+}

src/main/java/org/example/plzdrawing/common/listener/websocket/ChatUnsubscribeEventListener.java

@@ -0,0 +1,25 @@
+package org.example.plzdrawing.common.listener.websocket;
+
+import lombok.RequiredArgsConstructor;
+import org.example.plzdrawing.api.chat.repository.ChatPresenceRepository;
+import org.springframework.context.ApplicationListener;
+import org.springframework.messaging.simp.stomp.StompHeaderAccessor;
+import org.springframework.stereotype.Component;
+import org.springframework.web.socket.messaging.SessionUnsubscribeEvent;
+
+@Component
+@RequiredArgsConstructor
+public class ChatUnsubscribeEventListener implements ApplicationListener<SessionUnsubscribeEvent> {
+
+    private final ChatPresenceRepository chatPresenceRepository;
+
+    @Override
+    public void onApplicationEvent(SessionUnsubscribeEvent event) {
+        StompHeaderAccessor accessor = StompHeaderAccessor.wrap(event.getMessage());
+        String destination = accessor.getDestination();
+        if (destination.startsWith("/topic/join/")) {
+            Long memberId = Long.parseLong(accessor.getFirstNativeHeader("memberId"));
+            chatPresenceRepository.removeActive(memberId);
+        }
+    }
+}