java - 使用 Spring websockets 的自定义 Stomp header

Question

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

我希望能够向客户发送自定义 header 。 我尝试使用 SimpMessageSendingOperations.convertAndSend(destination, object, headers) 方法来执行此操作。但是，自定义 header 不包含在 stomp 消息中。

通过代码调试看起来像 StompHeaderAccessor.toStompHeaderMap() 方法调用 toNativeHeaderMap()，它使用 native header 和原始 native header 映射来构建 stomp header 。

有没有办法将自定义 header 添加到 stomp 消息中？

Answer 1

StompHeaderAccessor 扩展了 NativeMessageHeaderAccessor，它似乎是非 stomp header 所在的位置，除了它们都存储在一个名为 nativeHeaders 的 header 中——它本身就是一个映射。

@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);
}

将自定义 header 包装到 nativeHeaders header 的简单服务器端拦截器应该足以将它们暴露在客户端，在那里它们可以作为映射 message.headers.nativeHeaders 使用。同样，您可以编写客户端拦截器以将 nativeHeaders 移动到常规 header 中 - 因此在您的客户端知道该消息之前，所有预期的 header 都只是在 message.headers 中。