支持多设备

xieb

2023-09-22 ca0a832d45d5ee63ac01d8bbe221cede8700b215

支持多设备

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

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