Examples with AmqpInterceptor org.apache.activemq.artemis.protocol.amqp.broker.AmqpInterceptor used on opensource projects

Example 1 with AmqpInterceptor

use of org.apache.activemq.artemis.protocol.amqp.broker.AmqpInterceptor in project activemq-artemis by apache.

the class AmqpSendReceiveInterceptorTest method testCreateQueueReceiver.

@Test(timeout = 60000)
public void testCreateQueueReceiver() throws Exception {
    final CountDownLatch latch = new CountDownLatch(1);
    server.getRemotingService().addIncomingInterceptor(new AmqpInterceptor() {

        @Override
        public boolean intercept(AMQPMessage message, RemotingConnection connection) throws ActiveMQException {
            latch.countDown();
            return true;
        }
    });
    AmqpClient client = createAmqpClient();
    AmqpConnection connection = addConnection(client.connect());
    AmqpSession session = connection.createSession();
    AmqpSender sender = session.createSender(getTestName());
    AmqpMessage message = new AmqpMessage();
    message.setMessageId("msg" + 1);
    message.setText("Test-Message");
    sender.send(message);
    assertTrue(latch.await(5, TimeUnit.SECONDS));
    final CountDownLatch latch2 = new CountDownLatch(1);
    server.getRemotingService().addOutgoingInterceptor(new AmqpInterceptor() {

        @Override
        public boolean intercept(AMQPMessage packet, RemotingConnection connection) throws ActiveMQException {
            latch2.countDown();
            return true;
        }
    });
    AmqpReceiver receiver = session.createReceiver(getTestName());
    receiver.flow(2);
    AmqpMessage amqpMessage = receiver.receive(5, TimeUnit.SECONDS);
    assertNotNull(amqpMessage);
    assertEquals(latch2.getCount(), 0);
    sender.close();
    receiver.close();
    connection.close();
}

Example 2 with AmqpInterceptor

use of org.apache.activemq.artemis.protocol.amqp.broker.AmqpInterceptor in project activemq-artemis by apache.

the class AmqpSendReceiveInterceptorTest method testCheckInterceptedMessageProperties.

@Test(timeout = 60000)
public void testCheckInterceptedMessageProperties() throws Exception {
    final CountDownLatch latch = new CountDownLatch(1);
    final String addressQueue = getTestName();
    final String messageId = "lala200";
    final String correlationId = "lala-corrId";
    final String msgText = "Test intercepted message";
    final boolean durableMsg = false;
    final short priority = 8;
    final long timeToLive = 10000;
    final String replyTo = "reply-to-myQueue";
    Map<String, Object> expectedProperties = new HashMap<>();
    expectedProperties.put(ADDRESS, addressQueue);
    expectedProperties.put(MESSAGE_ID, messageId);
    expectedProperties.put(CORRELATION_ID, correlationId);
    expectedProperties.put(MESSAGE_TEXT, msgText);
    expectedProperties.put(DURABLE, durableMsg);
    expectedProperties.put(PRIORITY, priority);
    expectedProperties.put(REPLY_TO, replyTo);
    expectedProperties.put(TIME_TO_LIVE, timeToLive);
    server.getRemotingService().addIncomingInterceptor(new AmqpInterceptor() {

        @Override
        public boolean intercept(AMQPMessage message, RemotingConnection connection) throws ActiveMQException {
            latch.countDown();
            return checkMessageProperties(message, expectedProperties);
        }
    });
    AmqpClient client = createAmqpClient();
    AmqpConnection connection = addConnection(client.connect());
    AmqpSession session = connection.createSession();
    AmqpSender sender = session.createSender(getTestName());
    AmqpMessage message = new AmqpMessage();
    message.setMessageId(messageId);
    message.setCorrelationId(correlationId);
    message.setText(msgText);
    message.setDurable(durableMsg);
    message.setPriority(priority);
    message.setReplyToAddress(replyTo);
    message.setTimeToLive(timeToLive);
    sender.send(message);
    assertTrue(latch.await(2, TimeUnit.SECONDS));
    final CountDownLatch latch2 = new CountDownLatch(1);
    server.getRemotingService().addOutgoingInterceptor(new AmqpInterceptor() {

        @Override
        public boolean intercept(AMQPMessage packet, RemotingConnection connection) throws ActiveMQException {
            latch2.countDown();
            return checkMessageProperties(packet, expectedProperties);
        }
    });
    AmqpReceiver receiver = session.createReceiver(getTestName());
    receiver.flow(2);
    AmqpMessage amqpMessage = receiver.receive(5, TimeUnit.SECONDS);
    assertNotNull(amqpMessage);
    assertEquals(latch2.getCount(), 0);
    sender.close();
    receiver.close();
    connection.close();
}