Examples with ChannelInterceptorAdapter org.springframework.messaging.support.ChannelInterceptorAdapter used on opensource projects

Example 1 with ChannelInterceptorAdapter

use of org.springframework.messaging.support.ChannelInterceptorAdapter in project spring-integration by spring-projects.

the class GatewayParserTests method testAsyncCompletableNoAsyncMessage.

@Test
public void testAsyncCompletableNoAsyncMessage() throws Exception {
    QueueChannel requestChannel = (QueueChannel) context.getBean("requestChannel");
    final AtomicReference<Thread> thread = new AtomicReference<>();
    requestChannel.addInterceptor(new ChannelInterceptorAdapter() {

        @Override
        public Message<?> preSend(Message<?> message, MessageChannel channel) {
            thread.set(Thread.currentThread());
            return super.preSend(message, channel);
        }
    });
    MessageChannel replyChannel = (MessageChannel) context.getBean("replyChannel");
    this.startResponder(requestChannel, replyChannel);
    TestService service = context.getBean("completableNoAsync", TestService.class);
    CompletableFuture<Message<?>> result = service.completableReturnsMessage("flowCompletableM");
    Message<?> reply = result.get(10, TimeUnit.SECONDS);
    assertEquals("flowCompletableM", reply.getPayload());
    assertEquals(Thread.currentThread(), thread.get());
    assertNull(TestUtils.getPropertyValue(context.getBean("&completableNoAsync"), "asyncExecutor"));
}

Example 2 with ChannelInterceptorAdapter

use of org.springframework.messaging.support.ChannelInterceptorAdapter in project spring-integration by spring-projects.

the class GatewayParserTests method testCustomCompletableNoAsync.

@Test
public void testCustomCompletableNoAsync() throws Exception {
    QueueChannel requestChannel = (QueueChannel) context.getBean("requestChannel");
    final AtomicReference<Thread> thread = new AtomicReference<>();
    requestChannel.addInterceptor(new ChannelInterceptorAdapter() {

        @Override
        public Message<?> preSend(Message<?> message, MessageChannel channel) {
            thread.set(Thread.currentThread());
            return super.preSend(message, channel);
        }
    });
    MessageChannel replyChannel = (MessageChannel) context.getBean("replyChannel");
    this.startResponder(requestChannel, replyChannel);
    TestService service = context.getBean("completableNoAsync", TestService.class);
    MyCompletableFuture result = service.customCompletable("flowCustomCompletable");
    String reply = result.get(10, TimeUnit.SECONDS);
    assertEquals("SYNC_CUSTOM_COMPLETABLE", reply);
    assertEquals(Thread.currentThread(), thread.get());
    assertNull(TestUtils.getPropertyValue(context.getBean("&completableNoAsync"), "asyncExecutor"));
}

Example 3 with ChannelInterceptorAdapter

use of org.springframework.messaging.support.ChannelInterceptorAdapter in project spring-integration by spring-projects.

the class GatewayParserTests method testAsyncCompletableNoAsync.

@Test
public void testAsyncCompletableNoAsync() throws Exception {
    QueueChannel requestChannel = (QueueChannel) context.getBean("requestChannel");
    final AtomicReference<Thread> thread = new AtomicReference<>();
    requestChannel.addInterceptor(new ChannelInterceptorAdapter() {

        @Override
        public Message<?> preSend(Message<?> message, MessageChannel channel) {
            thread.set(Thread.currentThread());
            return super.preSend(message, channel);
        }
    });
    MessageChannel replyChannel = (MessageChannel) context.getBean("replyChannel");
    this.startResponder(requestChannel, replyChannel);
    TestService service = context.getBean("completableNoAsync", TestService.class);
    CompletableFuture<String> result = service.completable("flowCompletable");
    String reply = result.get(10, TimeUnit.SECONDS);
    assertEquals("SYNC_COMPLETABLE", reply);
    assertEquals(Thread.currentThread(), thread.get());
    assertNull(TestUtils.getPropertyValue(context.getBean("&completableNoAsync"), "asyncExecutor"));
}

Example 4 with ChannelInterceptorAdapter

use of org.springframework.messaging.support.ChannelInterceptorAdapter in project spring-integration by spring-projects.

the class ChannelInterceptorTests method testPostSendInterceptorWithSentMessage.

@Test
public void testPostSendInterceptorWithSentMessage() {
    final AtomicBoolean invoked = new AtomicBoolean(false);
    channel.addInterceptor(new ChannelInterceptorAdapter() {

        @Override
        public void postSend(Message<?> message, MessageChannel channel, boolean sent) {
            assertNotNull(message);
            assertNotNull(channel);
            assertSame(ChannelInterceptorTests.this.channel, channel);
            assertTrue(sent);
            invoked.set(true);
        }
    });
    channel.send(new GenericMessage<String>("test"));
    assertTrue(invoked.get());
}

Example 5 with ChannelInterceptorAdapter

use of org.springframework.messaging.support.ChannelInterceptorAdapter in project spring-integration by spring-projects.

the class ChannelInterceptorTests method testPostReceiveInterceptor.

@Test
public void testPostReceiveInterceptor() {
    final AtomicInteger invokedCount = new AtomicInteger();
    final AtomicInteger messageCount = new AtomicInteger();
    channel.addInterceptor(new ChannelInterceptorAdapter() {

        @Override
        public Message<?> postReceive(Message<?> message, MessageChannel channel) {
            assertNotNull(channel);
            assertSame(ChannelInterceptorTests.this.channel, channel);
            if (message != null) {
                messageCount.incrementAndGet();
            }
            invokedCount.incrementAndGet();
            return message;
        }
    });
    channel.receive(0);
    assertEquals(1, invokedCount.get());
    assertEquals(0, messageCount.get());
    channel.send(new GenericMessage<String>("test"));
    Message<?> result = channel.receive(0);
    assertNotNull(result);
    assertEquals(2, invokedCount.get());
    assertEquals(1, messageCount.get());
}