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

Example 26 with ClientProducer

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

the class ConsumerCloseTest method testCloseWithManyMessagesInBufferAndSlowConsumer.

// https://jira.jboss.org/jira/browse/JBMESSAGING-1526
@Test
public void testCloseWithManyMessagesInBufferAndSlowConsumer() throws Exception {
    ClientConsumer consumer = session.createConsumer(queue);
    ClientProducer producer = session.createProducer(address);
    final int numMessages = 100;
    for (int i = 0; i < numMessages; i++) {
        ClientMessage message = session.createMessage(false);
        producer.send(message);
    }
    final CountDownLatch received = new CountDownLatch(1);
    final CountDownLatch waitingToProceed = new CountDownLatch(1);
    class MyHandler implements MessageHandler {

        @Override
        public void onMessage(final ClientMessage message) {
            try {
                received.countDown();
                waitingToProceed.await();
            } catch (Exception e) {
            }
        }
    }
    consumer.setMessageHandler(new MyHandler());
    session.start();
    assertTrue(received.await(5, TimeUnit.SECONDS));
    long timeout = System.currentTimeMillis() + 1000;
    // Instead of waiting a long time (like 1 second) we just make sure the buffer is full on the client
    while (((ClientConsumerImpl) consumer).getBufferSize() < 2 && System.currentTimeMillis() > timeout) {
        Thread.sleep(10);
    }
    waitingToProceed.countDown();
    // Close shouldn't wait for all messages to be processed before closing
    long start = System.currentTimeMillis();
    consumer.close();
    long end = System.currentTimeMillis();
    Assert.assertTrue(end - start <= 1500);
}

Example 27 with ClientProducer

use of org.apache.activemq.artemis.api.core.client.ClientProducer 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 28 with ClientProducer

use of org.apache.activemq.artemis.api.core.client.ClientProducer 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 29 with ClientProducer

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

the class ConsumerTest method testInVMURI.

// https://jira.jboss.org/jira/browse/HORNETQ-111
// Test that, on rollback credits are released for messages cleared in the buffer
@Test
public void testInVMURI() throws Exception {
    locator.close();
    ServerLocator locator = addServerLocator(ServerLocatorImpl.newLocator("vm:/1"));
    ClientSessionFactory factory = locator.createSessionFactory();
    ClientSession session = factory.createSession();
    ClientProducer producer = session.createProducer(QUEUE);
    producer.send(session.createMessage(true));
    ClientConsumer consumer = session.createConsumer(QUEUE);
    session.start();
    Assert.assertNotNull(consumer.receiveImmediate());
    session.close();
    factory.close();
}

Example 30 with ClientProducer

use of org.apache.activemq.artemis.api.core.client.ClientProducer 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();
}