알림 반환문제(메모리) 테스트

beyond-sw-camp · Sep 5, 2024 · 0eeefc2 · 0eeefc2
1 parent 6db71f1
commit 0eeefc2
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/src/main/java/com/example/want/api/sse/NotificationService.java b/src/main/java/com/example/want/api/sse/NotificationService.java
@@ -27,10 +27,19 @@ public SseEmitter addEmitter(Long projectId, Long memberId) {
     }
 
     private void removeExistingEmitter(Long projectId, Long memberId) {
-        projectMemberEmitters.computeIfPresent(projectId, (k, members) -> {
+        // 프로젝트 ID에 해당하는 회원 목록을 가져옵니다.
+        Map<Long, SseEmitter> members = projectMemberEmitters.get(projectId);
+
+        if (members != null) {
+            // 해당 멤버 ID의 Emitter를 제거합니다.
             members.remove(memberId);
-            return members.isEmpty() ? null : members;
-        });
+            log.info("Removed emitter for project {} and member {}.", projectId, memberId);
+            // 만약 멤버 목록이 비어 있다면 프로젝트 ID도 제거합니다.
+            if (members.isEmpty()) {
+                projectMemberEmitters.remove(projectId);
+                log.info("Removed project {} from emitters.", projectId);
+            }
+        }
     }
 
     private SseEmitter createNewEmitter() {