如何在Spring Boot应用程序中将自定义标题添加到STOMP CREATED消息中？

Question

我正在尝试将自定义标头添加到客户端在第一次连接时收到的STOMP'CREATED'消息。这里是连接到使用STOMP JavaScript的WebSocket的功能：

function connect() { 
    socket = new SockJS('/chat'); 
    stompClient = Stomp.over(socket); 
    stompClient.connect('', '', function(frame) { 
     whoami = frame.headers['user-name']; 
     console.log(frame); 
     stompClient.subscribe('/user/queue/messages', function(message) { 
      console.log("MESSAGE RECEIVED:"); 
      console.log(message); 

     showMessage(JSON.parse(message.body)); 
     }); 
     stompClient.subscribe('/topic/active', function(activeMembers) { 
     showActive(activeMembers); 
     }); 
    }); 
    } 

此功能打印以下浏览器的控制台：

body: "" 
command: "CONNECTED" 
headers: Object 
    heart-beat: "0,0" 
    user-name: "someuser" 
    version: "1.1" 

，我想添加自定义标题，以便输出要像：

body: "" 
command: "CONNECTED" 
headers: Object 
    heart-beat: "0,0" 
    user-name: "someuser" 
    version: "1.1" 
    custom-header: "foo" 

我在我的Spring Boot应用程序中有以下WebSocket配置。

WebSocketConfig.java

@Configuration 
@EnableWebSocketMessageBroker 
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer { 

    @Override 
    public void configureMessageBroker(MessageBrokerRegistry config) { 
    config.enableSimpleBroker("/queue", "/topic"); 
    config.setApplicationDestinationPrefixes("/app"); 
    } 

    @Override 
    public void registerStompEndpoints(StompEndpointRegistry registry) { 
    registry.addEndpoint("/chat", "/activeUsers") 
      .withSockJS() 
      .setInterceptors(customHttpSessionHandshakeInterceptor()); 
    } 

    ... 

    @Bean 
    public CustomHttpSessionHandshakeInterceptor 
     customHttpSessionHandshakeInterceptor() { 
     return new CustomHttpSessionHandshakeInterceptor(); 

    } 

} 

我试图注册 'HandshakeInterceptor' 设置自定义标题，但没有奏效。这里是 'CustomHttpSessionHandshakeInterceptor'：

CustomHttpSessionHandshakeInterceptor.java

public class CustomHttpSessionHandshakeInterceptor implements 

HandshakeInterceptor { 

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


       ServletServerHttpRequest servletRequest = 
        (ServletServerHttpRequest) request; 
       attributes.put("custom-header", "foo"); 
      } 
      return true; 
     } 

     public void afterHandshake(ServerHttpRequest request, 
      ServerHttpResponse response, 
      WebSocketHandler wsHandler, 
      Exception ex) { } 
} 

我发现在https://dzone.com/articles/spring-boot-based-websocket
这个代码片断有人可以解释我为什么这个方法行不通？在Spring Boot应用程序中，是否有另一种方法将自定义标头设置为服务器端的STOMP'CREATED'消息？
谢谢！

Answer 1

你试过这样吗？ MessageHeaderAccessor也有一个setHeader方法。 https://docs.spring.io/spring/docs/current/spring-framework-reference/html/websocket.html#websocket-stomp-authentication-token-based

Answer 2

也许为时已晚，但迟到总比不到好...

服务器消息（例如连接）是不可变的，这意味着它们不能被修改。

我会做的是注册一个客户端出站拦截器并通过覆盖preSend（...）方法来捕获连接的消息，并用我的自定义头创建一条新消息。

@Override 
    public Message<?> preSend(Message<?> message, MessageChannel channel) { 

     LOGGER.info("Outbound channel pre send ..."); 

     final StompHeaderAccessor headerAccessor = StompHeaderAccessor.wrap(message); 
     final StompCommand command = headerAccessor.getCommand(); 

     if (!isNull(command)) { 

      switch (command) { 

      case CONNECTED: 
final StompHeaderAccessor accessor = StompHeaderAccessor.create(headerAccessor.getCommand()); 
       accessor.setSessionId(headerAccessor.getSessionId()); 
     @SuppressWarnings("unchecked") 
     final MultiValueMap<String, String> nativeHeaders = (MultiValueMap<String, String>) headerAccessor.getHeader(StompHeaderAccessor.NATIVE_HEADERS); 
     accessor.addNativeHeaders(nativeHeaders); 

// add custom headers 
accessor.addNativeHeader("CUSTOM01", "CUSTOM01"); 

     final Message<?> newMessage = MessageBuilder.createMessage(new byte[0], accessor.getMessageHeaders()); 
       return newMessage; 
      default: 
       break; 
      } 
     } 

     return message; 
    }