孙永军 il y a 1 an
Parent
commit
dfad895df0

+ 4 - 1
src/main/java/com/sxtvs/open/api/review/service/impl/SSEService.java

@@ -45,7 +45,9 @@ public class SSEService {
     }
 
     public void monitor(Long groupId, SseEmitter sseEmitter) {
-
+        if (sseDataMap.containsKey(groupId)){
+            throw new BizException("请求繁忙");
+        }
         sseDataMap.put(groupId, sseEmitter);
     }
 
@@ -66,6 +68,7 @@ public class SSEService {
         if (sseDataMap.containsKey(gid)){
             SseEmitter sseEmitter = sseDataMap.get(gid);
             sseEmitter.complete();
+            sseDataMap.remove(gid);
         }
     }
 }