Examples with Subscription org.apache.activemq.broker.region.Subscription used on opensource projects

Example 1 with Subscription

use of org.apache.activemq.broker.region.Subscription in project activemq-artemis by apache.

the class NoDuplicateOnTopicNetworkTest method testProducerConsumerTopic.

public void testProducerConsumerTopic() throws Exception {
    final CountDownLatch consumerStarted = new CountDownLatch(1);
    Thread producerThread = new Thread(new Runnable() {

        @Override
        public void run() {
            TopicWithDuplicateMessages producer = new TopicWithDuplicateMessages();
            producer.setBrokerURL(BROKER_1);
            producer.setTopicName(TOPIC_NAME);
            try {
                producer.produce();
            } catch (JMSException e) {
                fail("Unexpected " + e);
            }
        }
    });
    final TopicWithDuplicateMessages consumer = new TopicWithDuplicateMessages();
    Thread consumerThread = new Thread(new Runnable() {

        @Override
        public void run() {
            consumer.setBrokerURL(BROKER_2);
            consumer.setTopicName(TOPIC_NAME);
            try {
                consumer.consumer();
                consumerStarted.countDown();
                consumer.getLatch().await(60, TimeUnit.SECONDS);
            } catch (Exception e) {
                fail("Unexpected " + e);
            }
        }
    });
    consumerThread.start();
    LOG.info("Started Consumer");
    assertTrue("consumer started eventually", consumerStarted.await(10, TimeUnit.SECONDS));
    // ensure subscription has percolated though the network
    Thread.sleep(2000);
    // verify network consumer priority
    final RegionBroker regionBroker = (RegionBroker) broker1.getRegionBroker();
    assertTrue("Found network destination with priority as expected", Wait.waitFor(new Wait.Condition() {

        @Override
        public boolean isSatisified() throws Exception {
            Map<ActiveMQDestination, Destination> destinationMap = regionBroker.getTopicRegion().getDestinationMap();
            LOG.info("destinations: " + destinationMap.keySet());
            boolean found = false;
            for (Destination destination : destinationMap.values()) {
                List<Subscription> subscriptions = destination.getConsumers();
                LOG.info(destination + " subscriptions: " + subscriptions);
                for (Subscription subscription : subscriptions) {
                    if (subscription.getConsumerInfo().isNetworkSubscription()) {
                        LOG.info("subscription: " + subscription + ", priority: " + subscription.getConsumerInfo().getPriority());
                        assertTrue("priority is < our base: " + subscription.getConsumerInfo().getPriority(), subscription.getConsumerInfo().getPriority() <= BASE_PRIORITY);
                        found = true;
                    }
                }
            }
            return found;
        }
    }));
    producerThread.start();
    LOG.info("Started Producer");
    producerThread.join();
    consumerThread.join();
    int duplicateCount = 0;
    Map<String, String> map = new HashMap<>();
    for (String msg : consumer.getMessageStrings()) {
        if (map.containsKey(msg)) {
            LOG.info("got duplicate: " + msg);
            duplicateCount++;
        }
        map.put(msg, msg);
    }
    consumer.unSubscribe();
    if (suppressDuplicateTopicSubs || dispatchPolicy instanceof PriorityNetworkDispatchPolicy) {
        assertEquals("no duplicates", 0, duplicateCount);
        assertEquals("got all required messages: " + map.size(), consumer.getNumMessages(), map.size());
    } else {
        assertTrue("we can get some duplicates: " + duplicateCount, duplicateCount >= 0);
        if (duplicateCount == 0) {
            assertEquals("got all required messages: " + map.size(), consumer.getNumMessages(), map.size());
        }
    }
}

Example 2 with Subscription

use of org.apache.activemq.broker.region.Subscription in project activemq-artemis by apache.

the class PriorityNetworkDispatchPolicyTest method testRemoveLowerPriorityDup.

@Test
public void testRemoveLowerPriorityDup() throws Exception {
    List<Subscription> consumers = new ArrayList<>();
    for (int i = 0; i < 3; i++) {
        ConsumerInfo instance = info.copy();
        instance.setPriority((byte) i);
        consumers.add(new TopicSubscription(brokerService.getBroker(), context, instance, usageManager));
    }
    underTest.dispatch(node, null, consumers);
    long count = 0;
    for (Subscription consumer : consumers) {
        count += consumer.getEnqueueCounter();
    }
    assertEquals("only one sub got message", 1, count);
}

Example 3 with Subscription

use of org.apache.activemq.broker.region.Subscription in project activemq-artemis by apache.

the class DurableSubscriptionUnsubscribeTest method assertCount.

private void assertCount(int all, int active) throws Exception {
    int inactive = all - active;
    // broker check
    Destination destination = broker.getDestination(topic);
    List<Subscription> subs = destination.getConsumers();
    int cActive = 0, cInactive = 0;
    for (Subscription sub : subs) {
        if (sub instanceof DurableTopicSubscription) {
            DurableTopicSubscription durable = (DurableTopicSubscription) sub;
            if (durable.isActive())
                cActive++;
            else
                cInactive++;
        }
    }
    assertEquals(active, cActive);
    assertEquals(inactive, cInactive);
    // admin view
    ObjectName[] subscriptions = broker.getAdminView().getDurableTopicSubscribers();
    assertEquals(active, subscriptions.length);
    subscriptions = broker.getAdminView().getInactiveDurableTopicSubscribers();
    assertEquals(inactive, subscriptions.length);
    // check the strange false MBean
    if (all == 0)
        assertEquals(0, countMBean());
    // check if we got all advisories
    assertEquals(100, all + advisories.get());
}

Example 4 with Subscription

use of org.apache.activemq.broker.region.Subscription in project gocd by gocd.

the class ActiveMqMessagingService method removeQueue.

@Override
public void removeQueue(String queueName) {
    try {
        ActiveMQQueue destination = new ActiveMQQueue(queueName);
        ConnectionContext connectionContext = BrokerSupport.getConnectionContext(broker.getBroker());
        Destination brokerDestination = broker.getDestination(destination);
        List<Subscription> consumers = brokerDestination.getConsumers();
        for (Subscription consumer : consumers) {
            consumer.remove(connectionContext, brokerDestination);
            brokerDestination.removeSubscription(connectionContext, consumer, 0);
        }
        broker.getBroker().removeDestination(connectionContext, destination, 1000);
        broker.removeDestination(destination);
    } catch (Exception e) {
        throw bomb(e);
    }
}

Example 5 with Subscription

use of org.apache.activemq.broker.region.Subscription in project ignite by apache.

the class IgniteJmsStreamerTest method testQueueMultipleThreads.

/**
 * @throws Exception If failed.
 */
public void testQueueMultipleThreads() throws Exception {
    Destination dest = new ActiveMQQueue(QUEUE_NAME);
    // produce messages into the queue
    produceObjectMessages(dest, false);
    try (IgniteDataStreamer<String, String> dataStreamer = grid().dataStreamer(DEFAULT_CACHE_NAME)) {
        JmsStreamer<ObjectMessage, String, String> jmsStreamer = newJmsStreamer(ObjectMessage.class, dataStreamer);
        jmsStreamer.setDestination(dest);
        jmsStreamer.setThreads(5);
        // subscribe to cache PUT events and return a countdown latch starting at CACHE_ENTRY_COUNT
        CountDownLatch latch = subscribeToPutEvents(CACHE_ENTRY_COUNT);
        // start the streamer
        jmsStreamer.start();
        DestinationStatistics qStats = broker.getBroker().getDestinationMap().get(dest).getDestinationStatistics();
        assertEquals(5, qStats.getConsumers().getCount());
        // all cache PUT events received in 10 seconds
        latch.await(10, TimeUnit.SECONDS);
        // assert that all consumers received messages - given that the prefetch is 1
        for (Subscription subscription : broker.getBroker().getDestinationMap().get(dest).getConsumers()) assertTrue(subscription.getDequeueCounter() > 0);
        assertAllCacheEntriesLoaded();
        jmsStreamer.stop();
    }
}