com.mockrunner.mock.jms.MockTopic Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mockrunner-jms Show documentation
Mock classes for Java Messaging System
The newest version!
package com.mockrunner.mock.jms;

import java.util.Iterator;
import java.util.List;
import java.util.Map;

import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.MessageListener;
import javax.jms.Topic;

/**
 * Mock implementation of JMS Topic.
 */
public class MockTopic extends MockDestination implements Topic
{
    private String name;
    
    public MockTopic(String name)
    {
        this.name = name;
    }
        
    public String getTopicName() throws JMSException
    {
        return name;
    }
    
    /**
     * Adds a message to this Topic that will
     * be propagated to the corresponding receivers.
     * @param message the message
     */
    public void addMessage(Message message) throws JMSException
    {
        addReceivedMessage(message);    
        boolean isConsumed = false;
        for (Object o1 : sessionSet()) {
            MockSession session = (MockSession) o1;
            MessageListener globalListener = session.getMessageListener();
            if (null != globalListener) {
                globalListener.onMessage(message);
                isConsumed = true;
                acknowledgeMessage(message, session);
            } else {
                List subscribers = session.getTopicTransmissionManager().getTopicSubscriberList(name);
                for (Object subscriber1 : subscribers) {
                    MockTopicSubscriber subscriber = (MockTopicSubscriber) subscriber1;
                    if (subscriber.canConsume(message)) {
                        subscriber.receiveMessage(message);
                        isConsumed = true;
                        acknowledgeMessage(message, session);
                    }
                }
                Map durableSubscribers = session.getTopicTransmissionManager().getDurableTopicSubscriberMap(name);
                for (Object o : durableSubscribers.keySet()) {
                    MockTopicSubscriber subscriber = (MockTopicSubscriber) durableSubscribers.get(o);
                    if (subscriber.canConsume(message)) {
                        subscriber.receiveMessage(message);
                        isConsumed = true;
                        acknowledgeMessage(message, session);
                    }
                }
            }
        }
        if(!isConsumed)
        {
            addCurrentMessage(message);
        }
    }
}