Examples of org.springframework.messaging.simp.stomp.StompHeaderAccessor

Package org.springframework.messaging.simp.stomp

Examples of org.springframework.messaging.simp.stomp.StompHeaderAccessor

org.springframework.messaging.simp.stomp.StompHeaderAccessor
A {@code MessageHeaderAccessor} to use when creating a {@code Message} from adecoded STOMP frame, or when encoding a {@code Message} to a STOMP frame.
When created from STOMP frame content, the actual STOMP headers are stored in the native header sub-map managed by the parent class {@link org.springframework.messaging.support.NativeMessageHeaderAccessor}while the parent class {@link org.springframework.messaging.simp.SimpMessageHeaderAccessor} managescommon processing headers some of which are based on STOMP headers (e.g. destination, content-type, etc).
An instance of this class can also be created by wrapping an existing {@code Message}. That message may have been created with the more generic {@link org.springframework.messaging.simp.SimpMessageHeaderAccessor} inwhich case STOMP headers are created from common processing headers. In this case it is also necessary to invoke either {@link #updateStompCommandAsClientMessage()} or{@link #updateStompCommandAsServerMessage()} if sending a message anddepending on whether a message is sent to a client or the message broker. @author Rossen Stoyanchev @since 4.0

  }


  @Test
  public void handleMessageToClientUserDestination() {


    StompHeaderAccessor headers = StompHeaderAccessor.create(StompCommand.MESSAGE);
    headers.setMessageId("mess0");
    headers.setSubscriptionId("sub0");
    headers.setDestination("/queue/foo-user123");
    headers.setNativeHeader(StompHeaderAccessor.ORIGINAL_DESTINATION, "/user/queue/foo");
    Message<byte[]> message = MessageBuilder.createMessage(EMPTY_PAYLOAD, headers.getMessageHeaders());
    this.protocolHandler.handleMessageToClient(this.session, message);


    assertEquals(1, this.session.getSentMessages().size());
    WebSocketMessage<?> textMessage = this.session.getSentMessages().get(0);
    assertTrue(((String) textMessage.getPayload()).contains("destination:/user/queue/foo\n"));

View Full Code Here


    verify(this.channel).send(this.messageCaptor.capture());
    Message<?> actual = this.messageCaptor.getValue();
    assertNotNull(actual);


    StompHeaderAccessor headers = StompHeaderAccessor.wrap(actual);
    assertEquals(StompCommand.CONNECT, headers.getCommand());
    assertEquals("s1", headers.getSessionId());
    assertNotNull(headers.getSessionAttributes());
    assertEquals("joe", headers.getUser().getName());
    assertEquals("guest", headers.getLogin());
    assertEquals("guest", headers.getPasscode());
    assertArrayEquals(new long[] {10000, 10000}, headers.getHeartbeat());
    assertEquals(new HashSet<>(Arrays.asList("1.1","1.0")), headers.getAcceptVersion());


    assertEquals(0, this.session.getSentMessages().size());
  }

View Full Code Here

      }
    };


    this.protocolHandler.afterSessionStarted(this.session, this.channel);


    StompHeaderAccessor headers = StompHeaderAccessor.create(StompCommand.CONNECT);
    Message<byte[]> message = MessageBuilder.createMessage(EMPTY_PAYLOAD, headers.getMessageHeaders());
    TextMessage textMessage = new TextMessage(new StompEncoder().encode(message));


    this.protocolHandler.handleMessageFromClient(this.session, textMessage, testChannel);
    assertEquals(Collections.emptyList(), session.getSentMessages());

View Full Code Here

  public void handleMessageToClientConnected() {


    UserSessionRegistry registry = new DefaultUserSessionRegistry();
    this.protocolHandler.setUserSessionRegistry(registry);


    StompHeaderAccessor headers = StompHeaderAccessor.create(StompCommand.CONNECTED);
    Message<byte[]> message = MessageBuilder.createMessage(EMPTY_PAYLOAD, headers.getMessageHeaders());
    this.protocolHandler.handleMessageToClient(this.session, message);


    assertEquals(1, this.session.getSentMessages().size());
    WebSocketMessage<?> textMessage = this.session.getSentMessages().get(0);
    assertEquals("CONNECTED\n" + "user-name:joe\n" + "\n" + "\u0000", textMessage.getPayload());

View Full Code Here

    this.session.setPrincipal(new UniqueUser("joe"));


    UserSessionRegistry registry = new DefaultUserSessionRegistry();
    this.protocolHandler.setUserSessionRegistry(registry);


    StompHeaderAccessor headers = StompHeaderAccessor.create(StompCommand.CONNECTED);
    Message<byte[]> message = MessageBuilder.createMessage(EMPTY_PAYLOAD, headers.getMessageHeaders());
    this.protocolHandler.handleMessageToClient(this.session, message);


    assertEquals(1, this.session.getSentMessages().size());
    WebSocketMessage<?> textMessage = this.session.getSentMessages().get(0);
    assertEquals("CONNECTED\n" + "user-name:joe\n" + "\n" + "\u0000", textMessage.getPayload());

View Full Code Here

  @Test
  public void handleMessageToClientConnectedWithHeartbeats() {


    SockJsSession sockJsSession = Mockito.mock(SockJsSession.class);


    StompHeaderAccessor headers = StompHeaderAccessor.create(StompCommand.CONNECTED);
    headers.setHeartbeat(0,10);
    Message<byte[]> message = MessageBuilder.createMessage(EMPTY_PAYLOAD, headers.getMessageHeaders());
    this.protocolHandler.handleMessageToClient(sockJsSession, message);


    verify(sockJsSession).disableHeartbeat();
  }

View Full Code Here

  }


  @Test
  public void handleMessageToClientConnectAck() {


    StompHeaderAccessor connectHeaders = StompHeaderAccessor.create(StompCommand.CONNECT);
    connectHeaders.setHeartbeat(10000, 10000);
    connectHeaders.setAcceptVersion("1.0,1.1");
    Message<?> connectMessage = MessageBuilder.createMessage(EMPTY_PAYLOAD, connectHeaders.getMessageHeaders());


    SimpMessageHeaderAccessor connectAckHeaders = SimpMessageHeaderAccessor.create(SimpMessageType.CONNECT_ACK);
    connectAckHeaders.setHeader(SimpMessageHeaderAccessor.CONNECT_MESSAGE_HEADER, connectMessage);
    Message<byte[]> connectAckMessage = MessageBuilder.createMessage(EMPTY_PAYLOAD, connectAckHeaders.getMessageHeaders());


    this.protocolHandler.handleMessageToClient(this.session, connectAckMessage);


    verifyNoMoreInteractions(this.channel);


    // Check CONNECTED reply


    assertEquals(1, this.session.getSentMessages().size());
    TextMessage textMessage = (TextMessage) this.session.getSentMessages().get(0);


    List<Message<byte[]>> messages = new StompDecoder().decode(ByteBuffer.wrap(textMessage.getPayload().getBytes()));
    assertEquals(1, messages.size());
    StompHeaderAccessor replyHeaders = StompHeaderAccessor.wrap(messages.get(0));


    assertEquals(StompCommand.CONNECTED, replyHeaders.getCommand());
    assertEquals("1.1", replyHeaders.getVersion());
    assertArrayEquals(new long[] {0, 0}, replyHeaders.getHeartbeat());
    assertEquals("joe", replyHeaders.getNativeHeader("user-name").get(0));
  }

View Full Code Here

    UserSessionRegistry registry = new DefaultUserSessionRegistry();
    this.protocolHandler.setUserSessionRegistry(registry);
    this.protocolHandler.setApplicationEventPublisher(publisher);
    this.protocolHandler.afterSessionStarted(this.session, this.channel);


    StompHeaderAccessor headers = StompHeaderAccessor.create(StompCommand.CONNECT);
    Message<byte[]> message = MessageBuilder.createMessage(EMPTY_PAYLOAD, headers.getMessageHeaders());
    TextMessage textMessage = new TextMessage(new StompEncoder().encode(message));
    this.protocolHandler.handleMessageFromClient(this.session, textMessage, this.channel);


    headers = StompHeaderAccessor.create(StompCommand.CONNECTED);
    message = MessageBuilder.createMessage(EMPTY_PAYLOAD, headers.getMessageHeaders());
    this.protocolHandler.handleMessageToClient(this.session, message);


    headers = StompHeaderAccessor.create(StompCommand.SUBSCRIBE);
    message = MessageBuilder.createMessage(EMPTY_PAYLOAD, headers.getMessageHeaders());
    textMessage = new TextMessage(new StompEncoder().encode(message));
    this.protocolHandler.handleMessageFromClient(this.session, textMessage, this.channel);


    headers = StompHeaderAccessor.create(StompCommand.UNSUBSCRIBE);
    message = MessageBuilder.createMessage(EMPTY_PAYLOAD, headers.getMessageHeaders());
    textMessage = new TextMessage(new StompEncoder().encode(message));
    this.protocolHandler.handleMessageFromClient(this.session, textMessage, this.channel);


    this.protocolHandler.afterSessionEnded(this.session, CloseStatus.BAD_DATA, this.channel);

View Full Code Here

    UserSessionRegistry registry = new DefaultUserSessionRegistry();
    this.protocolHandler.setUserSessionRegistry(registry);
    this.protocolHandler.setApplicationEventPublisher(publisher);
    this.protocolHandler.afterSessionStarted(this.session, this.channel);


    StompHeaderAccessor headers = StompHeaderAccessor.create(StompCommand.CONNECT);
    Message<byte[]> message = MessageBuilder.createMessage(EMPTY_PAYLOAD, headers.getMessageHeaders());
    TextMessage textMessage = new TextMessage(new StompEncoder().encode(message));
    this.protocolHandler.handleMessageFromClient(this.session, textMessage, this.channel);


    verify(this.channel).send(this.messageCaptor.capture());
    Message<?> actual = this.messageCaptor.getValue();
    assertNotNull(actual);
    assertEquals(StompCommand.CONNECT, StompHeaderAccessor.wrap(actual).getCommand());
    reset(this.channel);


    headers = StompHeaderAccessor.create(StompCommand.CONNECTED);
    message = MessageBuilder.createMessage(EMPTY_PAYLOAD, headers.getMessageHeaders());
    this.protocolHandler.handleMessageToClient(this.session, message);


    assertEquals(1, this.session.getSentMessages().size());
    textMessage = (TextMessage) this.session.getSentMessages().get(0);
    assertEquals("CONNECTED\n" + "user-name:joe\n" + "\n" + "\u0000", textMessage.getPayload());


    this.protocolHandler.afterSessionEnded(this.session, CloseStatus.BAD_DATA, this.channel);


    verify(this.channel).send(this.messageCaptor.capture());
    actual = this.messageCaptor.getValue();
    assertNotNull(actual);
    StompHeaderAccessor accessor = StompHeaderAccessor.wrap(actual);
    assertEquals(StompCommand.DISCONNECT, accessor.getCommand());
    assertEquals("s1", accessor.getSessionId());
    assertEquals("joe", accessor.getUser().getName());
  }

View Full Code Here

    verifyNoMoreInteractions(this.webSocketHandler);
  }


  @Test
  public void connectReceiveAndCloseWithStompFrame() throws Exception {
    StompHeaderAccessor accessor = StompHeaderAccessor.create(StompCommand.SEND);
    accessor.setDestination("/destination");
    MessageHeaders headers = accessor.getMessageHeaders();
    Message<byte[]> message = MessageBuilder.createMessage("body".getBytes(Charset.forName("UTF-8")), headers);
    byte[] bytes = new StompEncoder().encode(message);
    TextMessage textMessage = new TextMessage(bytes);
    SockJsFrame frame = SockJsFrame.messageFrame(new Jackson2SockJsMessageCodec(), textMessage.getPayload());

View Full Code Here

0 1 2

TOP

Related Classes of org.springframework.messaging.simp.stomp.StompHeaderAccessor

org.springframework.messaging.simp.config.MessageBrokerConfigurationTests

org.springframework.messaging.simp.SimpMessageType

org.springframework.messaging.simp.SimpMessagingTemplateTests

org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurationSupportTests

org.springframework.web.socket.messaging.StompSubProtocolHandler

org.springframework.web.socket.messaging.StompSubProtocolHandlerTests

org.springframework.web.socket.sockjs.client.RestTemplateXhrTransportTests

java.nio.charset.Charset

All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.