WebSocket Stomp over SockJS-超文本传输协议自定义标头

Question

我用的是跺脚。我的javascript客户端中的js over SockJS。我正在使用连接到websocket

stompClient.connect({}, function (frame) {

stomp over sockJS连接有2个http请求：

1. 请求发送至/info
2. http升级请求

客户端发送所有cookie。我也想发送自定义头（例如XSRF头），但没有找到方法。谢谢你的帮助。

Answer 1

如果在服务器上使用Spring boot，请在方法中使用@Header（name=“token”）注释。

用法-

@Controller
public class SocketController {

    static final String token = "1234";

    @MessageMapping("/send")
    @SendTo("/receive/changes")
    public Object notify(MessageModel message, @Header(name = "token") String header)throws Exception {
        if(!header.equals(token)) {
            // return when headers do not match
            return("Unauthorized");
        }
        // return the model object with associated sent message
        return new MessageModel(message.getMessage());
    } 
}

您应该有一个MessageModel类，该类包含message变量和必需的getter、setters和constructor。

在前端使用跺脚

用法-

function sendMessage() {
     var text = document.getElementById('text').value;
            stompClient.send("/send/message", {'token':'1234'},
               JSON.stringify({'message':text}));
}

为了增加更多的安全性，你可以在Spring使用CORS

Answer 2

自定义标题：

stompClient.connect({token: "ABC123"}, function(frame) { ... code ...});

没有自定义标题：

stompClient.connect({}, function(frame) { ... code ...});

在Javascript中，您可以使用以下方法提取STOMP标头：

  username = frame.headers['user-name'];

在服务器端，如果您使用的是Spring框架，那么您可以实现一个拦截器来将HTTP参数复制到WebSockets STOMP头。

public class HttpSessionHandshakeInterceptor_personalised implements HandshakeInterceptor {

    @Override
    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response,
            WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {


        // Set ip attribute to WebSocket session
        attributes.put("ip", request.getRemoteAddress());

        // ============================================= CODIGO PERSONAL
        ServletServerHttpRequest servletRequest = (ServletServerHttpRequest) request;
        HttpServletRequest httpServletRequest = servletRequest.getServletRequest();
//        httpServletRequest.getCookies();
//        httpServletRequest.getParameter("inquiryId");
//        httpServletRequest.getRemoteUser();

         String token = httpServletRequest.getParameter("token");


      ...
    }
}

对于发送不带STOMP参数的消息：

function sendMessage() {
     var from = document.getElementById('from').value;
     var text = document.getElementById('text').value;
            stompClient.send("/app/chatchannel", {},
               JSON.stringify({'from':from, 'text':text}));
}

在这里，您将参数传递到STOMP头中。

function sendMessage() {
     var from = document.getElementById('from').value;
     var text = document.getElementById('text').value;
            stompClient.send("/app/chatchannel", {'token':'AA123'},
               JSON.stringify({'from':from, 'text':text}));
}

Answer 3

@rohitdev所以基本上你不能发送任何HTTP标头使用StompClient，因为STOMP是层在webSocket，只有当webSocket握手发生时，我们有可能发送自定义标头。所以只有SockJS可以发送这个标题，但是出于某些原因不要这样做：https://github.com/sockjs/sockjs-client/issues/196