/**
*
* Copyright 2004 Protique Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
**/
package org.codehaus.activemq.usecases;
import org.codehaus.activemq.ActiveMQConnectionFactory;
import org.codehaus.activemq.JmsSendReceiveTestSupport;
import org.codehaus.activemq.message.ActiveMQTopic;
import javax.jms.Connection;
import javax.jms.Destination;
import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.MessageConsumer;
import javax.jms.MessageListener;
import javax.jms.Session;
import java.util.List;
/**
* @version $Revision: 1.4 $
*/
public class CompositePublishTest extends JmsSendReceiveTestSupport {
protected Connection sendConnection;
protected Connection receiveConnection;
protected Session receiveSession;
protected MessageConsumer[] consumers;
protected List[] messageLists;
protected void setUp() throws Exception {
super.setUp();
connectionFactory = createConnectionFactory();
sendConnection = createConnection();
sendConnection.start();
receiveConnection = createConnection();
receiveConnection.start();
System.out.println("Created sendConnection: " + sendConnection);
System.out.println("Created receiveConnection: " + receiveConnection);
session = sendConnection.createSession(false, Session.AUTO_ACKNOWLEDGE);
receiveSession = receiveConnection.createSession(false, Session.AUTO_ACKNOWLEDGE);
System.out.println("Created sendSession: " + session);
System.out.println("Created receiveSession: " + receiveSession);
producer = session.createProducer(null);
System.out.println("Created producer: " + producer);
if (topic) {
consumerDestination = session.createTopic(getConsumerSubject());
producerDestination = session.createTopic(getProducerSubject());
}
else {
consumerDestination = session.createQueue(getConsumerSubject());
producerDestination = session.createQueue(getProducerSubject());
}
System.out.println("Created consumer destination: " + consumerDestination + " of type: " + consumerDestination.getClass());
System.out.println("Created producer destination: " + producerDestination + " of type: " + producerDestination.getClass());
Destination[] destinations = getDestinations();
consumers = new MessageConsumer[destinations.length];
messageLists = new List[destinations.length];
for (int i = 0; i < destinations.length; i++) {
Destination dest = destinations[i];
messageLists[i] = createConcurrentList();
consumers[i] = receiveSession.createConsumer(dest);
consumers[i].setMessageListener(createMessageListener(i, messageLists[i]));
}
System.out.println("Started connections");
}
protected MessageListener createMessageListener(int i, final List messageList) {
return new MessageListener() {
public void onMessage(Message message) {
consumeMessage(message, messageList);
}
};
}
/**
* Returns the subject on which we publish
*/
protected String getSubject() {
return getPrefix() + "FOO.BAR," + getPrefix() + "FOO.X.Y";
}
/**
* Returns the destinations to which we consume
*/
protected Destination[] getDestinations() {
return new Destination[]{new ActiveMQTopic(getPrefix() + "FOO.BAR"), new ActiveMQTopic(getPrefix() + "FOO.*"), new ActiveMQTopic(getPrefix() + "FOO.X.Y")};
}
protected String getPrefix() {
return super.getSubject() + ".";
}
protected void assertMessagesAreReceived() throws JMSException {
waitForMessagesToBeDelivered();
for (int i = 0, size = messageLists.length; i < size; i++) {
System.out.println("Message list: " + i + " contains: " + messageLists[i].size() + " message(s)");
}
for (int i = 0, size = messageLists.length; i < size; i++) {
assertMessagesReceivedAreValid(messageLists[i]);
}
}
protected ActiveMQConnectionFactory createConnectionFactory() {
return new ActiveMQConnectionFactory("vm://localhost");
}
protected void tearDown() throws Exception {
session.close();
receiveSession.close();
sendConnection.close();
receiveConnection.close();
}
}