From ca0a832d45d5ee63ac01d8bbe221cede8700b215 Mon Sep 17 00:00:00 2001
From: xieb <vip_xiaobin810@163.com>
Date: Fri, 22 Sep 2023 14:56:49 +0800
Subject: [PATCH] 支持多设备
---
src/main/java/com/dji/sample/component/websocket/service/impl/WebSocketManageServiceImpl.java | 2 +-
src/main/java/com/dji/sample/component/websocket/config/AuthPrincipalHandler.java | 3 ++-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/main/java/com/dji/sample/component/websocket/config/AuthPrincipalHandler.java b/src/main/java/com/dji/sample/component/websocket/config/AuthPrincipalHandler.java
index 05fbd2d..bf58c80 100644
--- a/src/main/java/com/dji/sample/component/websocket/config/AuthPrincipalHandler.java
+++ b/src/main/java/com/dji/sample/component/websocket/config/AuthPrincipalHandler.java
@@ -15,6 +15,7 @@
import java.security.Principal;
import java.util.Map;
import java.util.Optional;
+import java.util.UUID;
/**
* @author sean.zhou
@@ -63,7 +64,7 @@
CustomClaim claim = (CustomClaim) ((ServletServerHttpRequest) request).getServletRequest()
.getAttribute(AuthInterceptor.TOKEN_CLAIM);
- return () -> claim.getWorkspaceId() + "/" + claim.getUserType() + "/" + claim.getId();
+ return () -> UUID.randomUUID().toString() + "/" + claim.getUserType() + "/" + claim.getId();
}
return () -> null;
}
diff --git a/src/main/java/com/dji/sample/component/websocket/service/impl/WebSocketManageServiceImpl.java b/src/main/java/com/dji/sample/component/websocket/service/impl/WebSocketManageServiceImpl.java
index a0070d4..02e7925 100644
--- a/src/main/java/com/dji/sample/component/websocket/service/impl/WebSocketManageServiceImpl.java
+++ b/src/main/java/com/dji/sample/component/websocket/service/impl/WebSocketManageServiceImpl.java
@@ -75,7 +75,7 @@
return RedisOpsUtils.hashKeys(key)
.stream()
.map(SESSIONS::get)
- .filter(this.getValueWithWorkspace(workspaceId)::contains)
+// .filter(this.getValueWithWorkspace(workspaceId)::contains)
.collect(Collectors.toSet());
}
--
Gitblit v1.9.3