Increase SSE timeout from 10 to 90 secs

dRAGon-Okinawa · Jun 11, 2024 · 0ae9240 · 0ae9240
1 parent c48f52c
commit 0ae9240
Showing 1 changed file with 14 additions and 7 deletions.
diff --git a/backend/src/main/java/ai/dragon/service/SseService.java b/backend/src/main/java/ai/dragon/service/SseService.java
@@ -11,12 +11,15 @@
 
 @Service
 public class SseService {
+    // TODO Custom timeout :
+    public static final long DEFAULT_TIMEOUT = 90L * 1000;
+
     private final Logger logger = LoggerFactory.getLogger(this.getClass());
 
     private ConcurrentHashMap<UUID, SseEmitter> emitters = new ConcurrentHashMap<>();
 
     public UUID createEmitter() {
-        return createEmitter(10L * 1000L);
+        return createEmitter(DEFAULT_TIMEOUT);
     }
 
     public UUID createEmitter(Long timeout) {
@@ -36,27 +39,31 @@ public SseEmitter retrieveEmitter(UUID id) {
         return emitters.get(id);
     }
 
-    public void complete(UUID id) {
+    public boolean complete(UUID id) {
         SseEmitter emitter = emitters.get(id);
         if (emitter == null) {
-            logger.warn("No emitter found for id: {}", id);
-            return;
+            logger.warn("Can't complete : No emitter found for id '{}'", id);
+            return false;
         }
         emitter.complete();
         emitters.remove(id);
+        return true;
     }
 
-    public void sendEvent(UUID id, Object event) {
+    public boolean sendEvent(UUID id, Object event) {
         SseEmitter emitter = emitters.get(id);
         if (emitter == null) {
-            logger.warn("No emitter found for id: {}", id);
-            return;
+            logger.info(event.toString());
+            logger.warn("Can't send event : No emitter found for id '{}'", id);
+            return false;
         }
         try {
             emitter.send(event);
         } catch (IOException e) {
             emitter.complete();
             emitters.remove(id);
+            return false;
         }
+        return true;
     }
 }