/*
*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.apache.qpid.headers;
import org.apache.qpid.client.AMQConnection;
import org.apache.qpid.client.AMQSession;
import org.apache.qpid.testutil.Config;
import javax.jms.*;
public class Publisher implements MessageListener
{
private final Object _lock = new Object();
private final AMQConnection _connection;
private final AMQSession _session;
private final Destination _exchange;
private final MessageFactory _factory;
private final MessageProducer _publisher;
private int _count;
Publisher(AMQConnection connection, Destination exchange) throws Exception
{
_connection = connection;
_exchange = exchange;
_session = (AMQSession) _connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
_factory = new MessageFactory(_session, 0, 19);
_publisher = _session.createProducer(_exchange);
}
Publisher(Config config) throws Exception
{
this((AMQConnection) config.getConnection(), config.getDestination());
}
private void test(int msgCount, int consumerCount) throws Exception
{
_count = consumerCount;
_factory.createController(_exchange).setMessageListener(this);
_connection.start();
long start = System.currentTimeMillis();
publish(msgCount);
waitForCompletion(consumerCount);
long end = System.currentTimeMillis();
System.out.println("Completed in " + (end - start) + " ms.");
//request shutdown
_publisher.send(_factory.createShutdownMessage());
_connection.stop();
_connection.close();
}
private void publish(int count) throws Exception
{
//send events
for (int i = 0; i < count; i++)
{
_publisher.send(_factory.createEventMessage());
if ((i + 1) % 100 == 0)
{
System.out.println("Sent " + (i + 1) + " messages");
}
}
//request report
_publisher.send(_factory.createReportRequestMessage());
}
private void waitForCompletion(int consumers) throws Exception
{
System.out.println("Waiting for completion...");
synchronized (_lock)
{
while (_count > 0)
{
_lock.wait();
}
}
}
public void onMessage(Message message)
{
System.out.println("Received report " + _factory.getReport(message) + " " + --_count + " remaining");
if (_count == 0)
{
synchronized (_lock)
{
_lock.notify();
}
}
}
public static void main(String[] argv) throws Exception
{
if (argv.length >= 2)
{
int msgCount = Integer.parseInt(argv[argv.length - 2]);
int consumerCount = Integer.parseInt(argv[argv.length - 1]);
Config config = new Config();
config.setType(Config.HEADERS);
config.setName("test_headers_exchange");
String[] options = new String[argv.length - 2];
System.arraycopy(argv, 0, options, 0, options.length);
config.setOptions(options);
new Publisher(config).test(msgCount, consumerCount);
}
}
}