孙永军 1 年之前
父節點
當前提交
dfad895df0
共有 1 個文件被更改,包括 4 次插入1 次删除
  1. 4 1
      src/main/java/com/sxtvs/open/api/review/service/impl/SSEService.java

+ 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);
         }
     }
 }