Examples with Queue org.apache.activemq.artemis.core.server.Queue used on opensource projects

Example 86 with Queue

use of org.apache.activemq.artemis.core.server.Queue in project activemq-artemis by apache.

the class AmqpSendReceiveTest method doTestBrokerRestartAndDurability.

private void doTestBrokerRestartAndDurability(boolean durable, boolean enforceHeader) throws Exception {
    AmqpClient client = createAmqpClient();
    AmqpConnection connection = addConnection(client.connect());
    AmqpSession session = connection.createSession();
    AmqpSender sender = session.createSender(getQueueName());
    final Queue queueView1 = getProxyToQueue(getQueueName());
    // Create default message that should be sent as non-durable
    AmqpMessage message = new AmqpMessage();
    message.setText("Test-Message -> non-durable");
    message.setMessageId("ID:Message:1");
    if (durable) {
        message.setDurable(true);
    } else {
        if (enforceHeader) {
            message.setDurable(false);
            assertNotNull(message.getWrappedMessage().getHeader());
        } else {
            assertNull(message.getWrappedMessage().getHeader());
        }
    }
    sender.send(message);
    connection.close();
    Wait.assertEquals(1, queueView1::getMessageCount);
    // Restart the server and the Queue should be empty
    server.stop();
    server.start();
    // Reconnect now
    connection = addConnection(client.connect());
    session = connection.createSession();
    AmqpReceiver receiver = session.createReceiver(getQueueName());
    final Queue queueView2 = getProxyToQueue(getQueueName());
    if (durable) {
        Wait.assertTrue("Message should not have returned", () -> queueView2.getMessageCount() == 1);
    } else {
        Wait.assertTrue("Message should have been restored", () -> queueView2.getMessageCount() == 0);
    }
    receiver.flow(1);
    message = receiver.receive(1, TimeUnit.SECONDS);
    if (durable) {
        assertNotNull("Should have read a message", message);
    } else {
        assertNull("Should not have read a message", message);
    }
    connection.close();
}

Example 87 with Queue

use of org.apache.activemq.artemis.core.server.Queue in project activemq-artemis by apache.

the class AmqpSendReceiveTest method testReceiveMessageBeyondAckedAmountQueue.

@Test(timeout = 60000)
public void testReceiveMessageBeyondAckedAmountQueue() throws Exception {
    final int MSG_COUNT = 50;
    AmqpClient client = createAmqpClient();
    AmqpConnection connection = addConnection(client.connect());
    AmqpSession session = connection.createSession();
    final String address = getQueueName();
    AmqpReceiver receiver = session.createReceiver(address);
    AmqpSender sender = session.createSender(address);
    final Queue destinationView = getProxyToQueue(address);
    for (int i = 0; i < MSG_COUNT; i++) {
        AmqpMessage message = new AmqpMessage();
        message.setMessageId("msg" + i);
        sender.send(message);
    }
    List<AmqpMessage> pendingAcks = new ArrayList<>();
    for (int i = 0; i < MSG_COUNT; i++) {
        receiver.flow(1);
        AmqpMessage received = receiver.receive(5, TimeUnit.SECONDS);
        assertNotNull(received);
        pendingAcks.add(received);
    }
    // Send one more to check in-flight stays at zero with no credit and all
    // pending messages settled.
    AmqpMessage message = new AmqpMessage();
    message.setMessageId("msg-final");
    sender.send(message);
    for (AmqpMessage pendingAck : pendingAcks) {
        pendingAck.accept();
    }
    Wait.assertEquals(0, destinationView::getDeliveringCount);
    sender.close();
    receiver.close();
    connection.close();
}

Example 88 with Queue

use of org.apache.activemq.artemis.core.server.Queue in project activemq-artemis by apache.

the class AmqpSendReceiveTest method testSecondReceiverOnQueueGetsAllUnconsumedMessages.

@Test(timeout = 60000)
public void testSecondReceiverOnQueueGetsAllUnconsumedMessages() throws Exception {
    int MSG_COUNT = 20;
    sendMessages(getQueueName(), MSG_COUNT);
    AmqpClient client = createAmqpClient();
    AmqpConnection connection = addConnection(client.connect());
    AmqpSession session = connection.createSession();
    AmqpReceiver receiver1 = session.createReceiver(getQueueName());
    final Queue queueView = getProxyToQueue(getQueueName());
    assertEquals(MSG_COUNT, queueView.getMessageCount());
    receiver1.flow(20);
    assertTrue("Should have dispatch to prefetch", Wait.waitFor(new Wait.Condition() {

        @Override
        public boolean isSatisfied() throws Exception {
            return queueView.getDeliveringCount() >= 2;
        }
    }, TimeUnit.SECONDS.toMillis(5), TimeUnit.MILLISECONDS.toMillis(50)));
    receiver1.close();
    AmqpReceiver receiver2 = session.createReceiver(getQueueName());
    assertEquals(1, server.getTotalConsumerCount());
    receiver2.flow(MSG_COUNT * 2);
    AmqpMessage message = receiver2.receive(5, TimeUnit.SECONDS);
    assertNotNull(message);
    message.accept();
    message = receiver2.receive(5, TimeUnit.SECONDS);
    assertNotNull(message);
    message.accept();
    assertTrue("Should have ack'd two", Wait.waitFor(new Wait.Condition() {

        @Override
        public boolean isSatisfied() throws Exception {
            return queueView.getMessagesAcknowledged() == 2;
        }
    }, TimeUnit.SECONDS.toMillis(5), TimeUnit.MILLISECONDS.toMillis(50)));
    receiver2.close();
    assertEquals(MSG_COUNT - 2, queueView.getMessageCount());
    connection.close();
}

Example 89 with Queue

use of org.apache.activemq.artemis.core.server.Queue in project activemq-artemis by apache.

the class AmqpSendReceiveTest method testMessageDurableFalse.

@Test(timeout = 60000)
public void testMessageDurableFalse() throws Exception {
    sendMessages(getQueueName(), 1, false);
    AmqpClient client = createAmqpClient();
    AmqpConnection connection = addConnection(client.connect());
    AmqpSession session = connection.createSession();
    AmqpReceiver receiver = session.createReceiver(getQueueName());
    Queue queueView = getProxyToQueue(getQueueName());
    assertEquals(1, queueView.getMessageCount());
    receiver.flow(1);
    AmqpMessage receive = receiver.receive(5, TimeUnit.SECONDS);
    assertNotNull(receive);
    assertFalse(receive.isDurable());
    receiver.close();
    assertEquals(1, queueView.getMessageCount());
    connection.close();
}

Example 90 with Queue

use of org.apache.activemq.artemis.core.server.Queue in project activemq-artemis by apache.

the class AmqpSendReceiveTest method doTestSendReceiveLotsOfDurableMessages.

private void doTestSendReceiveLotsOfDurableMessages(Class<?> destType) throws Exception {
    final int MSG_COUNT = 1000;
    AmqpClient client = createAmqpClient();
    AmqpConnection connection = addConnection(client.connect());
    AmqpSession session = connection.createSession();
    final CountDownLatch done = new CountDownLatch(MSG_COUNT);
    final AtomicBoolean error = new AtomicBoolean(false);
    final ExecutorService executor = Executors.newSingleThreadExecutor();
    final String address;
    if (Queue.class.equals(destType)) {
        address = getQueueName();
    } else {
        address = getTopicName();
    }
    final AmqpReceiver receiver = session.createReceiver(address);
    receiver.flow(MSG_COUNT);
    AmqpSender sender = session.createSender(address);
    Queue queueView = getProxyToQueue(address);
    executor.execute(new Runnable() {

        @Override
        public void run() {
            for (int i = 0; i < MSG_COUNT; i++) {
                try {
                    AmqpMessage received = receiver.receive(5, TimeUnit.SECONDS);
                    received.accept();
                    done.countDown();
                } catch (Exception ex) {
                    LOG.info("Caught error: {}", ex.getClass().getSimpleName());
                    error.set(true);
                }
            }
        }
    });
    for (int i = 0; i < MSG_COUNT; i++) {
        AmqpMessage message = new AmqpMessage();
        message.setMessageId("msg" + i);
        sender.send(message);
    }
    assertTrue("did not read all messages, waiting on: " + done.getCount(), done.await(10, TimeUnit.SECONDS));
    assertFalse("should not be any errors on receive", error.get());
    Wait.assertEquals(0, queueView::getDeliveringCount);
    sender.close();
    receiver.close();
    connection.close();
}