来自客户端的 Java (Tomcat) websocket 参数

Question

我需要将一些参数从 javascript 客户端传递到 Tomcat 中的 Java 端点。

使用 Tomcat，您可以使用@ServerEndpoint(value="/websocket")注释您的 java 类，然后这个类将成为您的 websocket 的端点。

即每次 javascript 调用 new WebSocket("ws://localhost/websocket") 时，您的注释类都会被实例化。

现在我想在实例化期间（即在构造函数中）从 javascript 客户端将一个短字符串参数（或 int，并不重要）传递给我的注释类最好。

我该怎么做？

我曾想过将其放入 url，但随后我需要做其他事情而不是注释，因此感觉不太安全。

Answer 1

您可以将其作为标头传递，并且暴露 Web 套接字的 Java Tomcat 端点必须读取标头并将其设置为端点配置的用户属性映射 示例

公共类 WsHandShakeInterceptor 扩展 ServerEndpointConfig.Configurator {

@Override
public void modifyHandshake(ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) {
    super.modifyHandshake(sec, request, response);

    setUserHeadersAsProps(sec, request);
}

private void setUserHeadersAsProps(ServerEndpointConfig sec, HandshakeRequest request) {
    Map<String, List<String>> headers = request.getHeaders();
    if (headers != null) {
        List<String> userNameHeader = headers.get(HeadersConst.OIDC_CLAIM_USERNAME);
        if (userNameHeader != null && !userNameHeader.isEmpty()) {
            sec.getUserProperties().put(HeadersConst.X_FORWARDED_USER, userNameHeader.get(0));
        }
    }
}

}