En-tête STOMP sur @MessageMapping return

Question

Dans mon application Spring Boot 1.5 avec Spring Websocket, je souhaite définir un en-tête STOMP personnalisé sur la valeur de retour d'une méthode @MessageMapping, mais je ne sais pas comment procéder. Par exemple:

@Controller 
public class ChannelController { 

    @MessageMapping("/books/{id}") 
    public Book receive(@DestinationVariable("id") Long bookId) { 
     return findBook(bookId); 
    } 

    private Book findBook(Long bookId) { 
     return //... 
    } 
} 

Lorsque receive est déclenché à partir STOMP SEND d'un client, je voudrais le cadre de réponse STOMP MESSAGE avec le corps de livre pour un en-tête personnalisé: message-type:BOOK comme ceci:

MESSAGE 
message-type:BOOK 
destination:/topic/books/1 
content-type:application/json;charset=UTF-8 
subscription:sub-0 
message-id:0-7 
content-length:1868 

{ 
    "createdDate" : "2017-08-10T10:40:39.256", 
    "lastModifiedDate" : "2017-08-10T10:42:57.976", 
    "id" : 1, 
    "name" : "The big book", 
    "description" : null 
} 
^@ 

Comment faire J'ai défini un en-tête STOMP pour la valeur de réponse de réponse dans un @MessageMapping?

Answer 1

Si la signature de la valeur de retour est importante, vous pouvez utiliser SimpMessagingTemplate comme @Shchipunov noté dans les commentaires à sa réponse:

@Controller 
@AllArgsConstructor 
public class ChannelController { 

    private final SimpMessagingTemplate messagingTemplate; 

    @MessageMapping("/books/{id}") 
    public void receive(@DestinationVariable("id") Long bookId, SimpMessageHeaderAccessor accessor) { 
     accessor.setHeader("message-type", "SESSION"); 

     messagingTemplate.convertAndSend(
      "/topic/books/" + bookId, findBook(bookId), accessor.toMap() 
     ); 
    } 

    private Book findBook(Long bookId) { 
     return //... 
    } 
} 

qui ne correctement sérialiser au cadre MESSAGE dans la question.

Answer 2

Vous pouvez essayer cette solution:

@MessageMapping("/books/{id}") 
public GenericMessage<Book> receive(@DestinationVariable("id") Long bookId) { 
    Map<String, List<String>> nativeHeaders = new HashMap<>(); 
    nativeHeaders.put("message-type", Collections.singletonList("BOOK")); 

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

    return new GenericMessage<Book>(findBook(bookId), headers); 
}