孙永军 1 rok temu
rodzic
commit
d539666ea7

+ 2 - 2
src/main/java/com/sxtvs/open/api/review/controller/SseController.java

@@ -19,9 +19,9 @@ public class SseController {
     private SSEService sseService;
 
     @GetMapping(path = "/sse", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
-    public SseEmitter handleSse(@RequestParam(value = "token", required = false, defaultValue = "") String token) throws InterruptedException {
+    public SseEmitter handleSse() throws InterruptedException {
         SseEmitter sseEmitter = new SseEmitter(0L); // 设置超时时间
-        sseService.monitor(token, sseEmitter);
+        sseService.monitor(sseEmitter);
         Thread.sleep(500);
         return sseEmitter;
     }

+ 6 - 9
src/main/java/com/sxtvs/open/api/review/service/impl/SSEService.java

@@ -22,16 +22,13 @@ public class SSEService {
 
     private Map<Long, SseEmitter>  sseMap = new HashMap<>();
 
-    public void monitor(String token, SseEmitter sseEmitter) {
-        if (TextUtils.isEmpty(token)){
-            Optional<String> oToken = HttpContextUtil.getToken();
-            if (oToken.isPresent()){
-                token = oToken.get();
-            }
-        }
-        String uid = AESUtil.decryptStr(token);
-        sseMap.put(Long.valueOf(uid), sseEmitter);
+    public void monitor(SseEmitter sseEmitter) {
+        Optional<String> token = HttpContextUtil.getToken();
 
+        if (token.isPresent()){
+            String uid = AESUtil.decryptStr(token.get());
+            sseMap.put(Long.valueOf(uid), sseEmitter);
+        }
     }
 
     public Boolean sendMsg(Long uid, String msg) {