Examples with StompSessionConnectedEvent org.springframework.integration.stomp.event.StompSessionConnectedEvent used on opensource projects

Example 1 with StompSessionConnectedEvent

use of org.springframework.integration.stomp.event.StompSessionConnectedEvent in project spring-integration by spring-projects.

the class StompServerIntegrationTests method testStompAdapters.

@Test
public void testStompAdapters() throws Exception {
    ConfigurableApplicationContext context1 = new AnnotationConfigApplicationContext(ContextConfiguration.class);
    ConfigurableApplicationContext context2 = new AnnotationConfigApplicationContext(ContextConfiguration.class);
    PollableChannel stompEvents1 = context1.getBean("stompEvents", PollableChannel.class);
    PollableChannel stompEvents2 = context2.getBean("stompEvents", PollableChannel.class);
    PollableChannel stompInputChannel1 = context1.getBean("stompInputChannel", PollableChannel.class);
    PollableChannel stompInputChannel2 = context2.getBean("stompInputChannel", PollableChannel.class);
    MessageChannel stompOutputChannel1 = context1.getBean("stompOutputChannel", MessageChannel.class);
    MessageChannel stompOutputChannel2 = context2.getBean("stompOutputChannel", MessageChannel.class);
    Message<?> eventMessage;
    do {
        eventMessage = stompEvents1.receive(10000);
    } while (eventMessage != null && !(eventMessage.getPayload() instanceof StompSessionConnectedEvent));
    assertNotNull(eventMessage);
    eventMessage = stompEvents1.receive(10000);
    assertNotNull(eventMessage);
    assertThat(eventMessage.getPayload(), instanceOf(StompReceiptEvent.class));
    StompReceiptEvent stompReceiptEvent = (StompReceiptEvent) eventMessage.getPayload();
    assertEquals(StompCommand.SUBSCRIBE, stompReceiptEvent.getStompCommand());
    assertEquals("/topic/myTopic", stompReceiptEvent.getDestination());
    eventMessage = stompEvents2.receive(10000);
    assertNotNull(eventMessage);
    assertThat(eventMessage.getPayload(), instanceOf(StompSessionConnectedEvent.class));
    eventMessage = stompEvents2.receive(10000);
    assertNotNull(eventMessage);
    assertThat(eventMessage.getPayload(), instanceOf(StompReceiptEvent.class));
    stompReceiptEvent = (StompReceiptEvent) eventMessage.getPayload();
    assertEquals(StompCommand.SUBSCRIBE, stompReceiptEvent.getStompCommand());
    assertEquals("/topic/myTopic", stompReceiptEvent.getDestination());
    stompOutputChannel1.send(new GenericMessage<byte[]>("Hello, Client#2!".getBytes()));
    Message<?> receive11 = stompInputChannel1.receive(10000);
    Message<?> receive21 = stompInputChannel2.receive(10000);
    assertNotNull(receive11);
    assertNotNull(receive21);
    assertArrayEquals("Hello, Client#2!".getBytes(), (byte[]) receive11.getPayload());
    assertArrayEquals("Hello, Client#2!".getBytes(), (byte[]) receive21.getPayload());
    stompOutputChannel2.send(new GenericMessage<byte[]>("Hello, Client#1!".getBytes()));
    Message<?> receive12 = stompInputChannel1.receive(10000);
    Message<?> receive22 = stompInputChannel2.receive(10000);
    assertNotNull(receive12);
    assertNotNull(receive22);
    assertArrayEquals("Hello, Client#1!".getBytes(), (byte[]) receive12.getPayload());
    assertArrayEquals("Hello, Client#1!".getBytes(), (byte[]) receive22.getPayload());
    eventMessage = stompEvents2.receive(10000);
    assertNotNull(eventMessage);
    assertThat(eventMessage.getPayload(), instanceOf(StompReceiptEvent.class));
    stompReceiptEvent = (StompReceiptEvent) eventMessage.getPayload();
    assertEquals(StompCommand.SEND, stompReceiptEvent.getStompCommand());
    assertEquals("/topic/myTopic", stompReceiptEvent.getDestination());
    assertArrayEquals("Hello, Client#1!".getBytes(), (byte[]) stompReceiptEvent.getMessage().getPayload());
    Lifecycle stompInboundChannelAdapter2 = context2.getBean("stompInboundChannelAdapter", Lifecycle.class);
    stompInboundChannelAdapter2.stop();
    stompOutputChannel1.send(new GenericMessage<byte[]>("How do you do?".getBytes()));
    Message<?> receive13 = stompInputChannel1.receive(10000);
    assertNotNull(receive13);
    Message<?> receive23 = stompInputChannel2.receive(100);
    assertNull(receive23);
    stompInboundChannelAdapter2.start();
    eventMessage = stompEvents2.receive(10000);
    assertNotNull(eventMessage);
    assertThat(eventMessage.getPayload(), instanceOf(StompReceiptEvent.class));
    stompReceiptEvent = (StompReceiptEvent) eventMessage.getPayload();
    assertEquals(StompCommand.SUBSCRIBE, stompReceiptEvent.getStompCommand());
    assertEquals("/topic/myTopic", stompReceiptEvent.getDestination());
    stompOutputChannel1.send(new GenericMessage<byte[]>("???".getBytes()));
    Message<?> receive24 = stompInputChannel2.receive(10000);
    assertNotNull(receive24);
    assertArrayEquals("???".getBytes(), (byte[]) receive24.getPayload());
    activeMQBroker.stop();
    do {
        eventMessage = stompEvents1.receive(10000);
        assertNotNull(eventMessage);
    } while (!(eventMessage.getPayload() instanceof StompConnectionFailedEvent));
    try {
        stompOutputChannel1.send(new GenericMessage<byte[]>("foo".getBytes()));
        fail("MessageDeliveryException is expected");
    } catch (Exception e) {
        assertThat(e, instanceOf(MessageDeliveryException.class));
        assertThat(e.getMessage(), containsString("could not deliver message"));
    }
    activeMQBroker.start(false);
    do {
        eventMessage = stompEvents1.receive(20000);
        assertNotNull(eventMessage);
    } while (!(eventMessage.getPayload() instanceof StompReceiptEvent));
    do {
        eventMessage = stompEvents2.receive(10000);
        assertNotNull(eventMessage);
    } while (!(eventMessage.getPayload() instanceof StompReceiptEvent));
    stompOutputChannel1.send(new GenericMessage<byte[]>("foo".getBytes()));
    Message<?> receive25 = stompInputChannel2.receive(10000);
    assertNotNull(receive25);
    assertArrayEquals("foo".getBytes(), (byte[]) receive25.getPayload());
    context1.close();
    context2.close();
}

Example 2 with StompSessionConnectedEvent

use of org.springframework.integration.stomp.event.StompSessionConnectedEvent in project spring-integration by spring-projects.

the class StompInboundChannelAdapterWebSocketIntegrationTests method testWebSocketStompClient.

@Test
public void testWebSocketStompClient() throws Exception {
    Message<?> eventMessage = this.stompEvents.receive(10000);
    assertNotNull(eventMessage);
    assertThat(eventMessage.getPayload(), instanceOf(StompSessionConnectedEvent.class));
    Message<?> receive = this.stompEvents.receive(10000);
    assertNotNull(receive);
    assertThat(receive.getPayload(), instanceOf(StompReceiptEvent.class));
    StompReceiptEvent stompReceiptEvent = (StompReceiptEvent) receive.getPayload();
    assertEquals(StompCommand.SUBSCRIBE, stompReceiptEvent.getStompCommand());
    assertEquals("/topic/myTopic", stompReceiptEvent.getDestination());
    waitForSubscribe("/topic/myTopic");
    SimpMessagingTemplate messagingTemplate = this.serverContext.getBean("brokerMessagingTemplate", SimpMessagingTemplate.class);
    StompHeaderAccessor stompHeaderAccessor = StompHeaderAccessor.create(StompCommand.MESSAGE);
    stompHeaderAccessor.setContentType(MediaType.APPLICATION_JSON);
    stompHeaderAccessor.updateStompCommandAsServerMessage();
    stompHeaderAccessor.setLeaveMutable(true);
    messagingTemplate.send("/topic/myTopic", MessageBuilder.createMessage("{\"foo\": \"bar\"}".getBytes(), stompHeaderAccessor.getMessageHeaders()));
    receive = this.stompInputChannel.receive(10000);
    assertNotNull(receive);
    assertThat(receive.getPayload(), instanceOf(Map.class));
    @SuppressWarnings("unchecked") Map<String, String> payload = (Map<String, String>) receive.getPayload();
    String foo = payload.get("foo");
    assertNotNull(foo);
    assertEquals("bar", foo);
    this.stompInboundChannelAdapter.removeDestination("/topic/myTopic");
    waitForUnsubscribe("/topic/myTopic");
    messagingTemplate.convertAndSend("/topic/myTopic", "foo");
    receive = this.errorChannel.receive(100);
    assertNull(receive);
    this.stompInboundChannelAdapter.addDestination("/topic/myTopic");
    receive = this.stompEvents.receive(10000);
    assertNotNull(receive);
    waitForSubscribe("/topic/myTopic");
    messagingTemplate.convertAndSend("/topic/myTopic", "foo");
    receive = this.errorChannel.receive(10000);
    assertNotNull(receive);
    assertThat(receive, instanceOf(ErrorMessage.class));
    ErrorMessage errorMessage = (ErrorMessage) receive;
    Throwable throwable = errorMessage.getPayload();
    assertThat(throwable, instanceOf(MessageHandlingException.class));
    assertThat(throwable.getCause(), instanceOf(MessageConversionException.class));
    assertThat(throwable.getMessage(), containsString("No suitable converter for payload type [interface java.util.Map]"));
    this.serverContext.close();
    eventMessage = this.stompEvents.receive(10000);
    assertNotNull(eventMessage);
    assertThat(eventMessage.getPayload(), instanceOf(StompConnectionFailedEvent.class));
    this.serverContext.refresh();
    do {
        eventMessage = this.stompEvents.receive(10000);
        assertNotNull(eventMessage);
    } while (!(eventMessage.getPayload() instanceof StompSessionConnectedEvent));
    waitForSubscribe("/topic/myTopic");
    messagingTemplate = this.serverContext.getBean("brokerMessagingTemplate", SimpMessagingTemplate.class);
    messagingTemplate.convertAndSend("/topic/myTopic", "foo");
    receive = this.errorChannel.receive(10000);
    assertNotNull(receive);
    assertEquals(0, ((QueueChannel) this.errorChannel).getQueueSize());
}

Example 3 with StompSessionConnectedEvent

use of org.springframework.integration.stomp.event.StompSessionConnectedEvent in project spring-integration by spring-projects.

the class AbstractStompSessionManager method connect.

private synchronized void connect() {
    if (this.connecting || this.connected) {
        if (this.logger.isDebugEnabled()) {
            this.logger.debug("Aborting connect; another thread is connecting.");
        }
        return;
    }
    final int epoch = this.epoch.get();
    this.connecting = true;
    if (this.logger.isDebugEnabled()) {
        this.logger.debug("Connecting " + this);
    }
    try {
        this.stompSessionListenableFuture = doConnect(this.compositeStompSessionHandler);
    } catch (Exception e) {
        if (epoch == this.epoch.get()) {
            scheduleReconnect(e);
        } else {
            this.logger.error("STOMP doConnect() error for " + this, e);
        }
        return;
    }
    final CountDownLatch connectLatch = new CountDownLatch(1);
    this.stompSessionListenableFuture.addCallback(stompSession -> {
        if (AbstractStompSessionManager.this.logger.isDebugEnabled()) {
            AbstractStompSessionManager.this.logger.debug("onSuccess");
        }
        AbstractStompSessionManager.this.connected = true;
        AbstractStompSessionManager.this.connecting = false;
        stompSession.setAutoReceipt(isAutoReceiptEnabled());
        if (AbstractStompSessionManager.this.applicationEventPublisher != null) {
            AbstractStompSessionManager.this.applicationEventPublisher.publishEvent(new StompSessionConnectedEvent(this));
        }
        AbstractStompSessionManager.this.reconnectFuture = null;
        connectLatch.countDown();
    }, e -> {
        if (AbstractStompSessionManager.this.logger.isDebugEnabled()) {
            AbstractStompSessionManager.this.logger.debug("onFailure", e);
        }
        connectLatch.countDown();
        if (epoch == AbstractStompSessionManager.this.epoch.get()) {
            scheduleReconnect(e);
        }
    });
    try {
        if (!connectLatch.await(30, TimeUnit.SECONDS)) {
            this.logger.error("No response to connection attempt");
            if (epoch == this.epoch.get()) {
                scheduleReconnect(null);
            }
        }
    } catch (InterruptedException e1) {
        this.logger.error("Interrupted while waiting for connection attempt");
        Thread.currentThread().interrupt();
    }
}