handle TEXT_EVENT_STREAM_VALUE type

Author: netbuffer

src/main/java/cn/netbuffer/springboot/demo/controller/CustomControllerAdvice.java

@@ -2,6 +2,9 @@
 
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.http.HttpMethod;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.ControllerAdvice;
 import org.springframework.web.bind.annotation.ExceptionHandler;
 import org.springframework.web.bind.annotation.ResponseBody;
@@ -23,6 +26,15 @@ public Object handleAllException(Exception e, HandlerMethod handlerMethod, HttpM
         data.put("status", false);
         data.put("msg", e.getMessage());
         data.put("processed-by", this.getClass().getSimpleName());
+        String accept = httpServletRequest.getHeader("Accept");
+        if (accept != null && accept.contains(MediaType.TEXT_EVENT_STREAM_VALUE)) {
+            // 对 SSE 请求，返回空或特定 SseEmitter 处理
+            log.error("SSE request error:{}", e.getMessage());
+            return ResponseEntity
+                    .status(HttpStatus.INTERNAL_SERVER_ERROR)
+                    .contentType(MediaType.TEXT_PLAIN)
+                    .body(e.getMessage());
+        }
         return data;
     }
 

src/main/java/cn/netbuffer/springboot/demo/controller/SSEController.java

@@ -55,7 +55,10 @@ public SseEmitter handle(String id) {
     @GetMapping("sendById")
     public void sendById(String id, String message) throws IOException {
         log.debug("send message with SseEmitter id={}", id);
-        sseEmitterMap.get(id).send(message);
+        SseEmitter emitter = sseEmitterMap.get(id);
+        if (emitter != null) {
+            emitter.send(SseEmitter.event().data(message));
+        }
     }
 
 }