Examples with ClientMessage org.apache.activemq.artemis.api.core.client.ClientMessage used on opensource projects

Example 46 with ClientMessage

use of org.apache.activemq.artemis.api.core.client.ClientMessage in project activemq-artemis by apache.

the class ConsumerRoundRobinTest method testConsumersRoundRobinCorrectly.

@Test
public void testConsumersRoundRobinCorrectly() throws Exception {
    ActiveMQServer server = createServer(false);
    server.start();
    ServerLocator locator = createInVMNonHALocator();
    ClientSessionFactory cf = createSessionFactory(locator);
    ClientSession session = addClientSession(cf.createSession(false, true, true));
    session.createQueue(addressA, queueA, false);
    ClientConsumer[] consumers = new ClientConsumer[5];
    // start the session before we create the consumers, this is because start is non blocking and we have to
    // guarantee
    // all consumers have been started before sending messages
    session.start();
    consumers[0] = session.createConsumer(queueA);
    consumers[1] = session.createConsumer(queueA);
    consumers[2] = session.createConsumer(queueA);
    consumers[3] = session.createConsumer(queueA);
    consumers[4] = session.createConsumer(queueA);
    ClientProducer cp = session.createProducer(addressA);
    int numMessage = 10;
    for (int i = 0; i < numMessage; i++) {
        ClientMessage cm = session.createMessage(false);
        cm.getBodyBuffer().writeInt(i);
        cp.send(cm);
    }
    int currMessage = 0;
    for (int i = 0; i < numMessage / 5; i++) {
        log.info("i is " + i);
        for (int j = 0; j < 5; j++) {
            log.info("j is " + j);
            ClientMessage cm = consumers[j].receive(5000);
            Assert.assertNotNull(cm);
            Assert.assertEquals(currMessage++, cm.getBodyBuffer().readInt());
            cm.acknowledge();
        }
    }
}

Example 47 with ClientMessage

use of org.apache.activemq.artemis.api.core.client.ClientMessage in project activemq-artemis by apache.

the class ConsumerTest method testReceiveAndResend.

@Test
public void testReceiveAndResend() throws Exception {
    final Set<Object> sessions = new ConcurrentHashSet<>();
    final AtomicInteger errors = new AtomicInteger(0);
    final SimpleString QUEUE_RESPONSE = SimpleString.toSimpleString("QUEUE_RESPONSE");
    final int numberOfSessions = 50;
    final int numberOfMessages = 10;
    final CountDownLatch latchReceive = new CountDownLatch(numberOfSessions * numberOfMessages);
    ClientSessionFactory sf = locator.createSessionFactory();
    for (int i = 0; i < numberOfSessions; i++) {
        ClientSession session = sf.createSession(false, true, true);
        sessions.add(session);
        session.createQueue(QUEUE, QUEUE.concat("" + i), null, true);
        if (i == 0) {
            session.createQueue(QUEUE_RESPONSE, QUEUE_RESPONSE);
        }
        ClientConsumer consumer = session.createConsumer(QUEUE.concat("" + i));
        sessions.add(consumer);
        {
            consumer.setMessageHandler(new MessageHandler() {

                @Override
                public void onMessage(final ClientMessage msg) {
                    try {
                        ServerLocator locatorSendx = createFactory(isNetty()).setReconnectAttempts(15);
                        ClientSessionFactory factoryx = locatorSendx.createSessionFactory();
                        ClientSession sessionSend = factoryx.createSession(true, true);
                        sessions.add(sessionSend);
                        sessions.add(locatorSendx);
                        sessions.add(factoryx);
                        final ClientProducer prod = sessionSend.createProducer(QUEUE_RESPONSE);
                        sessionSend.start();
                        sessions.add(prod);
                        msg.acknowledge();
                        prod.send(sessionSend.createMessage(true));
                        prod.close();
                        sessionSend.commit();
                        sessionSend.close();
                        factoryx.close();
                        if (Thread.currentThread().isInterrupted()) {
                            System.err.println("Netty has interrupted a thread!!!");
                            errors.incrementAndGet();
                        }
                    } catch (Throwable e) {
                        e.printStackTrace();
                        errors.incrementAndGet();
                    } finally {
                        latchReceive.countDown();
                    }
                }
            });
        }
        session.start();
    }
    Thread tCons = new Thread() {

        @Override
        public void run() {
            try {
                final ServerLocator locatorSend = createFactory(isNetty());
                final ClientSessionFactory factory = locatorSend.createSessionFactory();
                final ClientSession sessionSend = factory.createSession(true, true);
                ClientConsumer cons = sessionSend.createConsumer(QUEUE_RESPONSE);
                sessionSend.start();
                for (int i = 0; i < numberOfMessages * numberOfSessions; i++) {
                    ClientMessage msg = cons.receive(5000);
                    if (msg == null) {
                        break;
                    }
                    msg.acknowledge();
                }
                if (cons.receiveImmediate() != null) {
                    System.out.println("ERROR: Received an extra message");
                    errors.incrementAndGet();
                }
                sessionSend.close();
                factory.close();
                locatorSend.close();
            } catch (Exception e) {
                e.printStackTrace();
                errors.incrementAndGet();
            }
        }
    };
    tCons.start();
    ClientSession mainSessionSend = sf.createSession(true, true);
    ClientProducer mainProd = mainSessionSend.createProducer(QUEUE);
    for (int i = 0; i < numberOfMessages; i++) {
        mainProd.send(mainSessionSend.createMessage(true));
    }
    latchReceive.await(2, TimeUnit.MINUTES);
    tCons.join();
    sf.close();
    assertEquals("Had errors along the execution", 0, errors.get());
}

Example 48 with ClientMessage

use of org.apache.activemq.artemis.api.core.client.ClientMessage in project activemq-artemis by apache.

the class ConsumerTest method testConsumerAckImmediateAutoCommitTrue.

@Test
public void testConsumerAckImmediateAutoCommitTrue() throws Exception {
    ClientSessionFactory sf = createSessionFactory(locator);
    ClientSession session = sf.createSession(false, true, true, true);
    ClientProducer producer = session.createProducer(QUEUE);
    final int numMessages = 100;
    for (int i = 0; i < numMessages; i++) {
        ClientMessage message = createTextMessage(session, "m" + i);
        producer.send(message);
    }
    ClientConsumer consumer = session.createConsumer(QUEUE);
    session.start();
    for (int i = 0; i < numMessages; i++) {
        ClientMessage message2 = consumer.receive(1000);
        Assert.assertEquals("m" + i, message2.getBodyBuffer().readString());
    }
    // assert that all the messages are there and none have been acked
    Assert.assertEquals(0, ((Queue) server.getPostOffice().getBinding(QUEUE).getBindable()).getDeliveringCount());
    Assert.assertEquals(0, getMessageCount(((Queue) server.getPostOffice().getBinding(QUEUE).getBindable())));
    session.close();
}

Example 49 with ClientMessage

use of org.apache.activemq.artemis.api.core.client.ClientMessage in project activemq-artemis by apache.

the class AcknowledgeTest method testAsyncConsumerAck.

@Test
public void testAsyncConsumerAck() throws Exception {
    ActiveMQServer server = createServer(false);
    server.start();
    ServerLocator locator = createInVMNonHALocator().setBlockOnAcknowledge(true).setAckBatchSize(0);
    ClientSessionFactory cf = createSessionFactory(locator);
    ClientSession sendSession = cf.createSession(false, true, true);
    final ClientSession session = cf.createSession(false, true, true);
    sendSession.createQueue(addressA, queueA, false);
    ClientProducer cp = sendSession.createProducer(addressA);
    ClientConsumer cc = session.createConsumer(queueA);
    int numMessages = 100;
    for (int i = 0; i < numMessages; i++) {
        cp.send(sendSession.createMessage(false));
    }
    final CountDownLatch latch = new CountDownLatch(numMessages);
    session.start();
    cc.setMessageHandler(new MessageHandler() {

        @Override
        public void onMessage(final ClientMessage message) {
            try {
                message.acknowledge();
            } catch (ActiveMQException e) {
                try {
                    session.close();
                } catch (ActiveMQException e1) {
                    e1.printStackTrace();
                }
            }
            latch.countDown();
        }
    });
    Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
    Queue q = (Queue) server.getPostOffice().getBinding(queueA).getBindable();
    Assert.assertEquals(0, q.getDeliveringCount());
    sendSession.close();
    session.close();
}

Example 50 with ClientMessage

use of org.apache.activemq.artemis.api.core.client.ClientMessage in project activemq-artemis by apache.

the class AcknowledgeTest method testReceiveAckLastMessageOnly.

@Test
public void testReceiveAckLastMessageOnly() throws Exception {
    ActiveMQServer server = createServer(false);
    server.start();
    ServerLocator locator = createInVMNonHALocator().setAckBatchSize(0).setBlockOnAcknowledge(true);
    ClientSessionFactory cf = createSessionFactory(locator);
    ClientSession sendSession = cf.createSession(false, true, true);
    ClientSession session = cf.createSession(false, true, true);
    sendSession.createQueue(addressA, queueA, false);
    ClientProducer cp = sendSession.createProducer(addressA);
    ClientConsumer cc = session.createConsumer(queueA);
    int numMessages = 100;
    for (int i = 0; i < numMessages; i++) {
        cp.send(sendSession.createMessage(false));
    }
    session.start();
    ClientMessage cm = null;
    for (int i = 0; i < numMessages; i++) {
        cm = cc.receive(5000);
        Assert.assertNotNull(cm);
    }
    cm.acknowledge();
    Queue q = (Queue) server.getPostOffice().getBinding(queueA).getBindable();
    Assert.assertEquals(0, q.getDeliveringCount());
    session.close();
    sendSession.close();
}