使用Spring websockets定制Stomp Headers

Question

我有一个基本的spring websocket应用程序,它目前向订阅者发送基本数据.目前,系统使用SimpMessageSendingOperations该类作为消息处理程序.如果我调用,SimpMessageSendingOperations.convertAndSend(destination, object)那么订阅的客户端将转换和接收该对象.

我希望能够向客户端发送自定义标头.我尝试过使用这种SimpMessageSendingOperations.convertAndSend(destination, object, headers)方法来做到这一点.但是,自定义标头不包含在stomp消息中.

通过代码调试它看起来像StompHeaderAccessor.toStompHeaderMap()方法调用 toNativeHeaderMap(),它使用本机头和原始本机头映射来构建stomp头.

有没有办法将自定义标头添加到stomp消息？

Answer 1

StompHeaderAccessor扩展了NativeMessageHeaderAccessor,它似乎是非stomp标头所在的位置,除了它们都存储在名为nativeHeaders的单个标头中 - 它本身就是一个映射.

@MessageMapping("/hello")
@SendTo("/topic/greetings")
public  GenericMessage<Greeting> greeting(HelloMessage message) throws Exception {      
    Map<String, List<String>> nativeHeaders = new HashMap<>();
    nativeHeaders.put("hello", Collections.singletonList("world"));

    Map<String,Object> headers = new HashMap<>();
    headers.put(NativeMessageHeaderAccessor.NATIVE_HEADERS, nativeHeaders);

    return new GenericMessage<Greeting>(new Greeting("Hello, " + message.getName() + "!"), headers);
}

用于将自定义标头包装到nativeHeaders标头的简单拦截器服务器端应足以在客户端公开它们作为映射message.headers.nativeHeaders的位置.同样地,您可以编写一个客户端拦截器来将nativeHeaders移动到常规头中 - 因此在您的客户端知道该消息之前,所有预期的头文件都只是在message.headers中.