统一风格

rain

2024-06-14 8d9a2d656e4ae007590c622e5f7c228adacdca49

 src/main/java/com/dji/sample/component/websocket/config/AuthPrincipalHandler.java

@@ -31,17 +31,19 @@
        if (request instanceof ServletServerHttpRequest) {
            HttpServletRequest servletRequest = ((ServletServerHttpRequest) request).getServletRequest();
            String token = servletRequest.getParameter(AuthInterceptor.PARAM_TOKEN);
            String workspaceId = servletRequest.getParameter(AuthInterceptor.WORKSPACE_ID);

            if (!StringUtils.hasText(token)) {
                return false;
            }

            log.debug("token:" + token);
            Optional<CustomClaim> customClaim = JwtUtil.parseToken(token);
            if (customClaim.isEmpty()) {
                return false;
            }

            servletRequest.setAttribute(AuthInterceptor.TOKEN_CLAIM, customClaim.get());
            servletRequest.setAttribute(AuthInterceptor.WORKSPACE_ID, workspaceId);
            return true;
        }
        return false;
@@ -63,7 +65,10 @@
            CustomClaim claim = (CustomClaim) ((ServletServerHttpRequest) request).getServletRequest()
                    .getAttribute(AuthInterceptor.TOKEN_CLAIM);

            return () -> claim.getWorkspaceId() + "/" + claim.getUserType() + "/" + claim.getId();
            String workspaceId = (String) ((ServletServerHttpRequest) request).getServletRequest()
                    .getAttribute(AuthInterceptor.WORKSPACE_ID);

            return () -> workspaceId + "/" + claim.getUserType() + "/" + claim.getId();
        }
        return () -> null;
    }