Examples with SessionInfo org.apache.activemq.command.SessionInfo used on opensource projects

Example 31 with SessionInfo

use of org.apache.activemq.command.SessionInfo in project activemq-artemis by apache.

the class BrokerNetworkWithStuckMessagesTest method testBrokerNetworkWithStuckMessages.

@Test(timeout = 120000)
public void testBrokerNetworkWithStuckMessages() throws Exception {
    int sendNumMessages = 10;
    int receiveNumMessages = 5;
    // Create a producer
    StubConnection connection1 = createConnection();
    ConnectionInfo connectionInfo1 = createConnectionInfo();
    SessionInfo sessionInfo1 = createSessionInfo(connectionInfo1);
    ProducerInfo producerInfo = createProducerInfo(sessionInfo1);
    connection1.send(connectionInfo1);
    connection1.send(sessionInfo1);
    connection1.send(producerInfo);
    // Create a destination on the local broker
    ActiveMQDestination destinationInfo1 = null;
    // Send a 10 messages to the local broker
    for (int i = 0; i < sendNumMessages; ++i) {
        destinationInfo1 = createDestinationInfo(connection1, connectionInfo1, ActiveMQDestination.QUEUE_TYPE);
        connection1.request(createMessage(producerInfo, destinationInfo1, DeliveryMode.NON_PERSISTENT));
    }
    // Ensure that there are 10 messages on the local broker
    Object[] messages = browseQueueWithJmx(localBroker);
    assertEquals(sendNumMessages, messages.length);
    // Create a synchronous consumer on the remote broker
    StubConnection connection2 = createRemoteConnection();
    ConnectionInfo connectionInfo2 = createConnectionInfo();
    SessionInfo sessionInfo2 = createSessionInfo(connectionInfo2);
    connection2.send(connectionInfo2);
    connection2.send(sessionInfo2);
    ActiveMQDestination destinationInfo2 = createDestinationInfo(connection2, connectionInfo2, ActiveMQDestination.QUEUE_TYPE);
    final ConsumerInfo consumerInfo2 = createConsumerInfo(sessionInfo2, destinationInfo2);
    connection2.send(consumerInfo2);
    // Consume 5 of the messages from the remote broker and ack them.
    for (int i = 0; i < receiveNumMessages; ++i) {
        Message message1 = receiveMessage(connection2, 20000);
        assertNotNull(message1);
        LOG.info("on remote, got: " + message1.getMessageId());
        connection2.send(createAck(consumerInfo2, message1, 1, MessageAck.INDIVIDUAL_ACK_TYPE));
        assertTrue("JMSActiveMQBrokerPath property present and correct", ((ActiveMQMessage) message1).getStringProperty(ActiveMQMessage.BROKER_PATH_PROPERTY).contains(localBroker.getBroker().getBrokerId().toString()));
    }
    // Ensure that there are zero messages on the local broker. This tells
    // us that those messages have been prefetched to the remote broker
    // where the demand exists.
    Wait.waitFor(new Wait.Condition() {

        @Override
        public boolean isSatisified() throws Exception {
            Object[] result = browseQueueWithJmx(localBroker);
            return 0 == result.length;
        }
    });
    messages = browseQueueWithJmx(localBroker);
    assertEquals(0, messages.length);
    // try and pull the messages from remote, should be denied b/c on networkTtl
    LOG.info("creating demand on second remote...");
    StubConnection connection3 = createSecondRemoteConnection();
    ConnectionInfo connectionInfo3 = createConnectionInfo();
    SessionInfo sessionInfo3 = createSessionInfo(connectionInfo3);
    connection3.send(connectionInfo3);
    connection3.send(sessionInfo3);
    ActiveMQDestination destinationInfo3 = createDestinationInfo(connection3, connectionInfo3, ActiveMQDestination.QUEUE_TYPE);
    final ConsumerInfo consumerInfoS3 = createConsumerInfo(sessionInfo3, destinationInfo3);
    connection3.send(consumerInfoS3);
    Message messageExceedingTtl = receiveMessage(connection3, 5000);
    if (messageExceedingTtl != null) {
        LOG.error("got message on Second remote: " + messageExceedingTtl);
        connection3.send(createAck(consumerInfoS3, messageExceedingTtl, 1, MessageAck.INDIVIDUAL_ACK_TYPE));
    }
    LOG.info("Closing consumer on remote");
    // Close the consumer on the remote broker
    connection2.send(consumerInfo2.createRemoveCommand());
    // also close connection etc.. so messages get dropped from the local consumer  q
    connection2.send(connectionInfo2.createRemoveCommand());
    // There should now be 5 messages stuck on the remote broker
    assertTrue("correct stuck message count", Wait.waitFor(new Wait.Condition() {

        @Override
        public boolean isSatisified() throws Exception {
            Object[] result = browseQueueWithJmx(remoteBroker);
            return 5 == result.length;
        }
    }));
    messages = browseQueueWithJmx(remoteBroker);
    assertEquals(5, messages.length);
    assertTrue("can see broker path property", ((String) ((CompositeData) messages[1]).get("BrokerPath")).contains(localBroker.getBroker().getBrokerId().toString()));
    LOG.info("Messages now stuck on remote");
    // receive again on the origin broker
    ConsumerInfo consumerInfo1 = createConsumerInfo(sessionInfo1, destinationInfo1);
    connection1.send(consumerInfo1);
    LOG.info("create local consumer: " + consumerInfo1);
    Message message1 = receiveMessage(connection1, 20000);
    assertNotNull("Expect to get a replay as remote consumer is gone", message1);
    connection1.send(createAck(consumerInfo1, message1, 1, MessageAck.INDIVIDUAL_ACK_TYPE));
    LOG.info("acked one message on origin, waiting for all messages to percolate back");
    Wait.waitFor(new Wait.Condition() {

        @Override
        public boolean isSatisified() throws Exception {
            Object[] result = browseQueueWithJmx(localBroker);
            return 4 == result.length;
        }
    });
    messages = browseQueueWithJmx(localBroker);
    assertEquals(4, messages.length);
    LOG.info("checking for messages on remote again");
    // messages won't migrate back again till consumer closes
    connection2 = createRemoteConnection();
    connectionInfo2 = createConnectionInfo();
    sessionInfo2 = createSessionInfo(connectionInfo2);
    connection2.send(connectionInfo2);
    connection2.send(sessionInfo2);
    ConsumerInfo consumerInfo3 = createConsumerInfo(sessionInfo2, destinationInfo2);
    connection2.send(consumerInfo3);
    message1 = receiveMessage(connection2, 20000);
    assertNull("Messages have migrated back: " + message1, message1);
    // Consume the last 4 messages from the local broker and ack them just
    // to clean up the queue.
    int counter = 1;
    for (; counter < receiveNumMessages; counter++) {
        message1 = receiveMessage(connection1);
        LOG.info("local consume of: " + (message1 != null ? message1.getMessageId() : " null"));
        connection1.send(createAck(consumerInfo1, message1, 1, MessageAck.INDIVIDUAL_ACK_TYPE));
    }
    // Ensure that 5 messages were received
    assertEquals(receiveNumMessages, counter);
    // verify all messages consumed
    Wait.waitFor(new Wait.Condition() {

        @Override
        public boolean isSatisified() throws Exception {
            Object[] result = browseQueueWithJmx(remoteBroker);
            return 0 == result.length;
        }
    });
    messages = browseQueueWithJmx(remoteBroker);
    assertEquals(0, messages.length);
    Wait.waitFor(new Wait.Condition() {

        @Override
        public boolean isSatisified() throws Exception {
            Object[] result = browseQueueWithJmx(localBroker);
            return 0 == result.length;
        }
    });
    messages = browseQueueWithJmx(localBroker);
    assertEquals(0, messages.length);
    // Close the consumer on the remote broker
    connection2.send(consumerInfo3.createRemoveCommand());
    connection1.stop();
    connection2.stop();
    connection3.stop();
}

Example 32 with SessionInfo

use of org.apache.activemq.command.SessionInfo in project activemq-artemis by apache.

the class SessionInfoTest method createObject.

@Override
public Object createObject() throws Exception {
    SessionInfo info = new SessionInfo();
    populateObject(info);
    return info;
}

Example 33 with SessionInfo

use of org.apache.activemq.command.SessionInfo in project activemq-artemis by apache.

the class SessionInfoTest method populateObject.

@Override
protected void populateObject(Object object) throws Exception {
    super.populateObject(object);
    SessionInfo info = (SessionInfo) object;
    info.setSessionId(createSessionId("SessionId:1"));
}

Example 34 with SessionInfo

use of org.apache.activemq.command.SessionInfo in project activemq-artemis by apache.

the class SessionInfoTest method populateObject.

@Override
protected void populateObject(Object object) throws Exception {
    super.populateObject(object);
    SessionInfo info = (SessionInfo) object;
    info.setSessionId(createSessionId("SessionId:1"));
}

Example 35 with SessionInfo

use of org.apache.activemq.command.SessionInfo in project activemq-artemis by apache.

the class SessionInfoTest method createObject.

@Override
public Object createObject() throws Exception {
    SessionInfo info = new SessionInfo();
    populateObject(info);
    return info;
}