Examples with AmqpReceiver org.apache.activemq.transport.amqp.client.AmqpReceiver used on opensource projects

Example 46 with AmqpReceiver

use of org.apache.activemq.transport.amqp.client.AmqpReceiver in project activemq-artemis by apache.

the class AmqpPresettledReceiverTest method testPresettledReceiverReadsAllMessages.

@Test(timeout = 60000)
public void testPresettledReceiverReadsAllMessages() throws Exception {
    final int MSG_COUNT = 100;
    sendMessages(getQueueName(), MSG_COUNT);
    AmqpClient client = createAmqpClient();
    AmqpConnection connection = addConnection(client.connect());
    AmqpSession session = connection.createSession();
    AmqpReceiver receiver = session.createReceiver(getQueueName(), null, false, true);
    final Queue queueView = getProxyToQueue(getQueueName());
    assertEquals(MSG_COUNT, queueView.getMessageCount());
    receiver.flow(MSG_COUNT);
    for (int i = 0; i < MSG_COUNT; ++i) {
        assertNotNull(receiver.receive(5, TimeUnit.SECONDS));
    }
    receiver.close();
    System.out.println("Message Count after all consumed: " + queueView.getMessageCount());
    // Open a new receiver and see if any message are left on the Queue
    receiver = session.createReceiver(getQueueName());
    receiver.flow(1);
    AmqpMessage received = receiver.receive(5, TimeUnit.SECONDS);
    if (received != null) {
        System.out.println("Message read: " + received.getMessageId());
    }
    assertNull(received);
    assertEquals(0, queueView.getMessageCount());
    connection.close();
}

Example 47 with AmqpReceiver

use of org.apache.activemq.transport.amqp.client.AmqpReceiver in project activemq-artemis by apache.

the class AmqpPresettledReceiverTest method testPresettledReceiverAndNonPresettledReceiverOnSameQueue.

@Test(timeout = 60000)
public void testPresettledReceiverAndNonPresettledReceiverOnSameQueue() throws Exception {
    final int MSG_COUNT = 2;
    sendMessages(getQueueName(), MSG_COUNT);
    AmqpClient client = createAmqpClient();
    AmqpConnection connection = addConnection(client.connect());
    AmqpSession session = connection.createSession();
    AmqpReceiver receiver1 = session.createReceiver(getQueueName(), null, false, true);
    AmqpReceiver receiver2 = session.createReceiver(getQueueName());
    final Queue queueView = getProxyToQueue(getQueueName());
    assertEquals(MSG_COUNT, queueView.getMessageCount());
    receiver1.flow(1);
    receiver2.flow(1);
    AmqpMessage message1 = receiver1.receive(5, TimeUnit.SECONDS);
    AmqpMessage message2 = receiver2.receive(5, TimeUnit.SECONDS);
    assertNotNull(message1);
    assertNotNull(message2);
    // Receiver 1 is presettled so messages are not accepted.
    assertTrue(message1.getWrappedDelivery().remotelySettled());
    // Receiver 2 is not presettled so it needs to accept.
    message2.accept();
    receiver1.close();
    receiver2.close();
    System.out.println("Message Count after all consumed: " + queueView.getMessageCount());
    // Should be nothing left on the Queue
    AmqpReceiver receiver3 = session.createReceiver(getQueueName());
    receiver3.flow(1);
    AmqpMessage received = receiver3.receive(5, TimeUnit.SECONDS);
    if (received != null) {
        System.out.println("Message read: " + received.getMessageId());
    }
    assertNull(received);
    assertEquals(0, queueView.getMessageCount());
    connection.close();
}

Example 48 with AmqpReceiver

use of org.apache.activemq.transport.amqp.client.AmqpReceiver in project activemq-artemis by apache.

the class AmqpPurgeOnNoConsumersTest method testQueueReceiverReadMessage.

@Test(timeout = 60000)
public void testQueueReceiverReadMessage() throws Exception {
    AmqpConnection connection = null;
    String queue = "purgeQueue";
    SimpleString ssQueue = new SimpleString(queue);
    server.addAddressInfo(new AddressInfo(ssQueue, RoutingType.ANYCAST));
    server.createQueue(ssQueue, RoutingType.ANYCAST, ssQueue, null, true, false, 1, true, false);
    AmqpClient client = createAmqpClient();
    connection = addConnection(client.connect());
    AmqpSession session = connection.createSession();
    final AmqpReceiver receiver = session.createReceiver(queue);
    QueueImpl queueView = (QueueImpl) getProxyToQueue(queue);
    assertEquals(0L, queueView.getPageSubscription().getPagingStore().getAddressSize());
    assertEquals(0, queueView.getMessageCount());
    sendMessages(queue, 5, null, true);
    Wait.assertEquals(5, queueView::getMessageCount);
    receiver.flow(5);
    for (int i = 0; i < 4; i++) {
        try {
            AmqpMessage receive = receiver.receive(5, TimeUnit.SECONDS);
            receive.accept();
            assertNotNull(receive);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    try {
        receiver.close();
    } catch (IOException e) {
        e.printStackTrace();
    }
    Wait.assertEquals(0, queueView::getMessageCount);
    assertEquals(0L, queueView.getPageSubscription().getPagingStore().getAddressSize());
    connection.close();
    server.stop();
    server.start();
    queueView = (QueueImpl) getProxyToQueue(queue);
    assertEquals(0, queueView.getMessageCount());
    assertEquals(0L, queueView.getPageSubscription().getPagingStore().getAddressSize());
}

Example 49 with AmqpReceiver

use of org.apache.activemq.transport.amqp.client.AmqpReceiver in project activemq-artemis by apache.

the class AmqpReceiverDispositionTest method testRejectedDisposition.

@Test(timeout = 30000)
public void testRejectedDisposition() throws Exception {
    sendMessages(getQueueName(), 1);
    AmqpClient client = createAmqpClient();
    AmqpConnection connection = addConnection(client.connect());
    AmqpSession session = connection.createSession();
    AmqpReceiver receiver1 = session.createReceiver(getQueueName());
    receiver1.flow(1);
    AmqpMessage message = receiver1.receive(5, TimeUnit.SECONDS);
    assertNotNull("did not receive message first time", message);
    assertEquals("MessageID:0", message.getMessageId());
    Message protonMessage = message.getWrappedMessage();
    assertNotNull(protonMessage);
    assertEquals("Unexpected initial value for AMQP delivery-count", 0, protonMessage.getDeliveryCount());
    message.reject();
    // Reject is a terminal outcome and should not be redelivered to the rejecting receiver
    // or any other as it should move to the archived state.
    receiver1.flow(1);
    message = receiver1.receiveNoWait();
    assertNull("Should not receive message again", message);
    // Attempt to Read the message again with another receiver to validate it is archived.
    AmqpReceiver receiver2 = session.createReceiver(getQueueName());
    receiver2.flow(1);
    assertNull(receiver2.receiveNoWait());
    connection.close();
}

Example 50 with AmqpReceiver

use of org.apache.activemq.transport.amqp.client.AmqpReceiver in project activemq-artemis by apache.

the class AmqpReceiverTest method testReceiverCloseSendsRemoteClose.

@Test(timeout = 60000)
public void testReceiverCloseSendsRemoteClose() throws Exception {
    AmqpClient client = createAmqpClient();
    assertNotNull(client);
    final AtomicBoolean closed = new AtomicBoolean();
    client.setValidator(new AmqpValidator() {

        @Override
        public void inspectClosedResource(Session session) {
            IntegrationTestLogger.LOGGER.info("Session closed: " + session.getContext());
        }

        @Override
        public void inspectDetachedResource(Receiver receiver) {
            markAsInvalid("Broker should not detach receiver linked to closed session.");
        }

        @Override
        public void inspectClosedResource(Receiver receiver) {
            IntegrationTestLogger.LOGGER.info("Receiver closed: " + receiver.getContext());
            closed.set(true);
        }
    });
    AmqpConnection connection = addConnection(client.connect());
    assertNotNull(connection);
    AmqpSession session = connection.createSession();
    assertNotNull(session);
    AmqpReceiver receiver = session.createReceiver(getQueueName());
    assertNotNull(receiver);
    receiver.close();
    assertTrue("Did not process remote close as expected", closed.get());
    connection.getStateInspector().assertValid();
    connection.close();
}