new : emitter 관리 service, controller 생성

Author: EpicFn

src/main/java/org/tuna/zoopzoop/backend/domain/SSE/controller/ApiV1NotificationController.java

@@ -0,0 +1,34 @@
+package org.tuna.zoopzoop.backend.domain.SSE.controller;
+
+import lombok.RequiredArgsConstructor;
+import org.springframework.security.core.Authentication;
+import org.springframework.security.core.annotation.AuthenticationPrincipal;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
+import org.tuna.zoopzoop.backend.domain.SSE.service.EmitterService;
+import org.tuna.zoopzoop.backend.global.security.jwt.CustomUserDetails;
+
+@RestController
+@RequestMapping("/api/v1/notifications")
+@RequiredArgsConstructor
+public class ApiV1NotificationController {
+    private final EmitterService emitterService;
+
+    /**
+     * SSE 구독 엔드포인트
+     * @param userDetails - 현재 인증된 사용자 정보
+     * @return SseEmitter - 클라이언트와의 SSE 연결을 관리하는 객체
+     */
+    @GetMapping(value = "/subscribe", produces = "text/event-stream")
+    public SseEmitter subscribe(
+            @AuthenticationPrincipal CustomUserDetails userDetails
+    ) {
+        // 1. 현재 로그인한 사용자의 ID를 가져옴
+        Long memberId = (long) userDetails.getMember().getId();
+
+        // 2. EmitterService를 통해 Emitter를 생성하고 반환
+        return emitterService.addEmitter(memberId);
+    }
+}

src/main/java/org/tuna/zoopzoop/backend/domain/SSE/service/EmitterService.java

@@ -0,0 +1,56 @@
+package org.tuna.zoopzoop.backend.domain.SSE.service;
+
+import org.springframework.stereotype.Service;
+import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
+
+import java.io.IOException;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+@Service
+public class EmitterService {
+    // 1. 모든 Emitter를 저장하는 ConcurrentHashMap
+    private final Map<Long, SseEmitter> emitters = new ConcurrentHashMap<>();
+
+    /**
+     * 새로운 Emitter 생성 및 저장
+     * @param memberId - 사용자 ID
+     * @return SseEmitter - 생성된 Emitter 객체
+     */
+    public SseEmitter addEmitter(Long memberId) {
+        // 1시간 타임아웃 설정
+        SseEmitter emitter = new SseEmitter(3600L * 1000);
+        this.emitters.put(memberId, emitter);
+
+        // Emitter 완료 또는 타임아웃 시 Map에서 삭제
+        emitter.onCompletion(() -> this.emitters.remove(memberId));
+        emitter.onTimeout(() -> this.emitters.remove(memberId));
+
+        // 503 에러 방지를 위한 더미 이벤트 전송
+        try {
+            emitter.send(SseEmitter.event().name("connect").data("SSE connected!"));
+        } catch (IOException e) {
+            // 예외 처리
+        }
+
+        return emitter;
+    }
+
+    /**
+     * 특정 사용자에게 이벤트 전송
+     * @param memberId - 사용자 ID
+     * @param eventName - 이벤트 이름
+     * @param data - 전송할 데이터 객체
+     */
+    public void sendNotification(Long memberId, String eventName, Object data) {
+        SseEmitter emitter = this.emitters.get(memberId);
+        if (emitter != null) {
+            try {
+                // data 객체를 JSON 문자열로 변환하여 전송해야 함 (Controller에서는 자동 변환)
+                emitter.send(SseEmitter.event().name(eventName).data(data));
+            } catch (IOException e) {
+                this.emitters.remove(memberId);
+            }
+        }
+    }
+}

src/main/java/org/tuna/zoopzoop/backend/domain/space/space/controller/ApiV1SpaceController.java

@@ -220,8 +220,8 @@ public RsData<ResBodyForSpaceInfo> getSpace(
     @Operation(summary = "Liveblocks 접속 토큰 발급")
     public ResponseEntity<RsData<ResBodyForAuthToken>> getAuthToken(
             @PathVariable Integer spaceId,
-            @AuthenticationPrincipal CustomUserDetails userDetails) throws AccessDeniedException {
-
+            @AuthenticationPrincipal CustomUserDetails userDetails
+    ) throws AccessDeniedException {
         Member member = userDetails.getMember();
         String token = dashboardService.getAuthTokenForSpace(spaceId, member);