final int CONSUMER_COUNT=Integer.parseInt(System.getProperty("CONSUMER_COUNT","10"));
final ProgressPrinter pp = new ProgressPrinter(MESSAGE_COUNT*2, 5);
final Semaphore connectionsEstablished = new Semaphore(1-(CONSUMER_COUNT+PRODUCER_COUNT));
final Latch startTest = new Latch();
final Semaphore testsFinished = new Semaphore(1-(CONSUMER_COUNT+PRODUCER_COUNT));
final Callable producer = new Callable() {
public Object call() throws JMSException, InterruptedException {
Connection connection = connectionFactory.createConnection();
Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
MessageProducer producer = session.createProducer(dest);
producer.setDeliveryMode(DeliveryMode.PERSISTENT);
BytesMessage message = session.createBytesMessage();
message.writeBytes(new byte[1024]);
connection.start();
connectionsEstablished.release();
startTest.acquire();
final int msgs = (MESSAGE_COUNT/PRODUCER_COUNT)+1;
for (int i = 0; i < msgs; i++) {
pp.increment();
producer.send(message);
}
testsFinished.release();
connection.close();
return null;
}
};
final Callable consumer = new Callable() {
public Object call() throws JMSException, InterruptedException {
final Latch doneLatch = new Latch();
Connection connection = connectionFactory.createConnection();
Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
MessageConsumer consumer = session.createConsumer(dest);
connectionsEstablished.release();
startTest.acquire();
final int msgs = (MESSAGE_COUNT/CONSUMER_COUNT)-1;
consumer.setMessageListener(new MessageListener(){
int counter=0;
public void onMessage(Message msg) {
pp.increment();
counter++;
if( counter >= msgs ) {
doneLatch.release();
}
}
});
connection.start();
doneLatch.acquire();
testsFinished.release();
connection.close();
return null;
}