Examples with TextMessage org.springframework.web.socket.TextMessage used on opensource projects

Example 56 with TextMessage

use of org.springframework.web.socket.TextMessage in project spring-framework by spring-projects.

the class StompSubProtocolHandler method sendErrorMessage.

/**
	 * Invoked when no
	 * {@link #setErrorHandler(StompSubProtocolErrorHandler) errorHandler}
	 * is configured to send an ERROR frame to the client.
	 */
private void sendErrorMessage(WebSocketSession session, Throwable error) {
    StompHeaderAccessor headerAccessor = StompHeaderAccessor.create(StompCommand.ERROR);
    headerAccessor.setMessage(error.getMessage());
    byte[] bytes = this.stompEncoder.encode(headerAccessor.getMessageHeaders(), EMPTY_PAYLOAD);
    try {
        session.sendMessage(new TextMessage(bytes));
    } catch (Throwable ex) {
        // Could be part of normal workflow (e.g. browser tab closed)
        logger.debug("Failed to send STOMP ERROR to client", ex);
    }
}

Example 57 with TextMessage

use of org.springframework.web.socket.TextMessage in project spring-framework by spring-projects.

the class RestTemplateXhrTransportTests method connectReceiveAndCloseWithStompFrame.

@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(StandardCharsets.UTF_8), headers);
    byte[] bytes = new StompEncoder().encode(message);
    TextMessage textMessage = new TextMessage(bytes);
    SockJsFrame frame = SockJsFrame.messageFrame(new Jackson2SockJsMessageCodec(), textMessage.getPayload());
    String body = "o\n" + frame.getContent() + "\n" + "c[3000,\"Go away!\"]";
    ClientHttpResponse response = response(HttpStatus.OK, body);
    connect(response);
    verify(this.webSocketHandler).afterConnectionEstablished(any());
    verify(this.webSocketHandler).handleMessage(any(), eq(textMessage));
    verify(this.webSocketHandler).afterConnectionClosed(any(), eq(new CloseStatus(3000, "Go away!")));
    verifyNoMoreInteractions(this.webSocketHandler);
}

Example 58 with TextMessage

use of org.springframework.web.socket.TextMessage in project spring-framework by spring-projects.

the class XhrTransportTests method sendMessageError.

@Test(expected = HttpServerErrorException.class)
public void sendMessageError() throws Exception {
    TestXhrTransport transport = new TestXhrTransport();
    transport.sendMessageResponseToReturn = new ResponseEntity<>(HttpStatus.BAD_REQUEST);
    URI url = new URI("http://example.com");
    transport.executeSendRequest(url, null, new TextMessage("payload"));
}

Example 59 with TextMessage

use of org.springframework.web.socket.TextMessage in project spring-framework by spring-projects.

the class XhrTransportTests method sendMessage.

@Test
public void sendMessage() throws Exception {
    HttpHeaders requestHeaders = new HttpHeaders();
    requestHeaders.set("foo", "bar");
    requestHeaders.setContentType(MediaType.APPLICATION_JSON);
    TestXhrTransport transport = new TestXhrTransport();
    transport.sendMessageResponseToReturn = new ResponseEntity<>(HttpStatus.NO_CONTENT);
    URI url = new URI("http://example.com");
    transport.executeSendRequest(url, requestHeaders, new TextMessage("payload"));
    assertEquals(2, transport.actualSendRequestHeaders.size());
    assertEquals("bar", transport.actualSendRequestHeaders.getFirst("foo"));
    assertEquals(MediaType.APPLICATION_JSON, transport.actualSendRequestHeaders.getContentType());
}

Example 60 with TextMessage

use of org.springframework.web.socket.TextMessage in project spring-framework by spring-projects.

the class AbstractSockJsIntegrationTests method testReceiveOneMessage.

private void testReceiveOneMessage(Transport transport, WebSocketHttpHeaders headers) throws Exception {
    TestClientHandler clientHandler = new TestClientHandler();
    initSockJsClient(transport);
    this.sockJsClient.doHandshake(clientHandler, headers, new URI(this.baseUrl + "/test")).get();
    TestServerHandler serverHandler = this.wac.getBean(TestServerHandler.class);
    assertNotNull("afterConnectionEstablished should have been called", clientHandler.session);
    serverHandler.awaitSession(5000);
    TextMessage message = new TextMessage("message1");
    serverHandler.session.sendMessage(message);
    clientHandler.awaitMessage(message, 5000);
}