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

Example 46 with AmqpSession

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

the class AmqpExpiredMessageTest method internalSendExpiry.

public void internalSendExpiry(boolean restartServer) throws Throwable {
    AmqpClient client = createAmqpClient();
    AmqpConnection connection = client.connect();
    try {
        // Normal Session which won't create an TXN itself
        AmqpSession session = connection.createSession();
        AmqpSender sender = session.createSender(getQueueName());
        AmqpMessage message = new AmqpMessage();
        message.setDurable(true);
        message.setText("Test-Message");
        message.setDeliveryAnnotation("shouldDisappear", 1);
        message.setAbsoluteExpiryTime(System.currentTimeMillis() + 1000);
        sender.send(message);
        org.apache.activemq.artemis.core.server.Queue dlq = server.locateQueue(SimpleString.toSimpleString(getDeadLetterAddress()));
        assertTrue("Message not movied to DLQ", Wait.waitFor(() -> dlq.getMessageCount() > 0, 5000, 500));
        connection.close();
        if (restartServer) {
            server.stop();
            server.start();
        }
        connection = client.connect();
        session = connection.createSession();
        // Read all messages from the Queue
        AmqpReceiver receiver = session.createReceiver(getDeadLetterAddress());
        receiver.flow(20);
        message = receiver.receive(5, TimeUnit.SECONDS);
        Assert.assertNotNull(message);
        Assert.assertEquals(getQueueName(), message.getMessageAnnotation(org.apache.activemq.artemis.api.core.Message.HDR_ORIGINAL_ADDRESS.toString()));
        Assert.assertNull(message.getDeliveryAnnotation("shouldDisappear"));
        Assert.assertNull(receiver.receiveNoWait());
    } finally {
        connection.close();
    }
}

Example 47 with AmqpSession

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

the class AmqpFlowControlTest method testCreditsAreAllocatedOnceOnLinkCreated.

@Test(timeout = 60000)
public void testCreditsAreAllocatedOnceOnLinkCreated() throws Exception {
    AmqpClient client = createAmqpClient(new URI(singleCreditAcceptorURI));
    AmqpConnection connection = addConnection(client.connect());
    try {
        AmqpSession session = connection.createSession();
        AmqpSender sender = session.createSender(getQueueName());
        assertEquals("Should only be issued one credit", 1, sender.getSender().getCredit());
    } finally {
        connection.close();
    }
}

Example 48 with AmqpSession

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

the class AmqpFlowControlTest method fillAddress.

/*
    * Fills an address.  Careful when using this method.  Only use when rejected messages are switched on.
    */
private void fillAddress(String address) throws Exception {
    AmqpClient client = createAmqpClient();
    AmqpConnection connection = addConnection(client.connect());
    Exception exception = null;
    try {
        AmqpSession session = connection.createSession();
        AmqpSender sender = session.createSender(address);
        sendUntilFull(sender);
    } catch (Exception e) {
        exception = e;
    } finally {
        connection.close();
    }
    // Should receive a rejected error
    assertNotNull(exception);
    assertTrue(exception.getMessage().contains("amqp:resource-limit-exceeded"));
}

Example 49 with AmqpSession

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

the class AmqpFlowControlTest method testCreditsAreNotAllocatedWhenAddressIsFull.

@Test(timeout = 60000)
public void testCreditsAreNotAllocatedWhenAddressIsFull() throws Exception {
    AmqpClient client = createAmqpClient(new URI(singleCreditAcceptorURI));
    AmqpConnection connection = addConnection(client.connect());
    try {
        AmqpSession session = connection.createSession();
        AmqpSender sender = session.createSender(getQueueName());
        // Use blocking send to ensure buffered messages do not interfere with credit.
        sender.setSendTimeout(-1);
        sendUntilFull(sender);
        // This should be -1. A single message is buffered in the client, and 0 credit has been allocated.
        assertTrue(sender.getSender().getCredit() == -1);
        long addressSize = server.getPagingManager().getPageStore(new SimpleString(getQueueName())).getAddressSize();
        assertTrue(addressSize >= MAX_SIZE_BYTES && addressSize <= MAX_SIZE_BYTES_REJECT_THRESHOLD);
    } finally {
        connection.close();
    }
}

Example 50 with AmqpSession

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

the class AmqpLargeMessageTest method testSendAMQPReceiveAMQP.

@Test(timeout = 60000)
public void testSendAMQPReceiveAMQP() throws Exception {
    server.getAddressSettingsRepository().addMatch("#", new AddressSettings().setDefaultAddressRoutingType(RoutingType.ANYCAST));
    String testQueueName = "ConnectionFrameSize";
    int nMsgs = 200;
    AmqpClient client = createAmqpClient();
    AmqpConnection connection = addConnection(client.connect());
    try {
        sendMessages(nMsgs, connection);
        int count = getMessageCount(server.getPostOffice(), testQueueName);
        assertEquals(nMsgs, count);
        AmqpSession session = connection.createSession();
        AmqpReceiver receiver = session.createReceiver(testQueueName);
        receiver.flow(nMsgs);
        for (int i = 0; i < nMsgs; ++i) {
            AmqpMessage message = receiver.receive(5, TimeUnit.SECONDS);
            assertNotNull("failed at " + i, message);
            MessageImpl wrapped = (MessageImpl) message.getWrappedMessage();
            if (wrapped.getBody() instanceof Data) {
                // converters can change this to AmqValue
                Data data = (Data) wrapped.getBody();
                System.out.println("received : message: " + data.getValue().getLength());
                assertEquals(PAYLOAD, data.getValue().getLength());
            }
            message.accept();
        }
        session.close();
    } finally {
        connection.close();
    }
}