/**
*
* 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.activemq.protocolbuffer;
import com.google.protobuf.CodedInputStream;
import com.google.protobuf.CodedOutputStream;
import java.io.*;
/**
* @version $Revision: 1.1 $
*/
public class PerformanceTest extends TestSupport {
protected String fileName = "target/performance.openwire";
protected OpenWire.Destination destination = OpenWire.Destination.newBuilder().setName("FOO.BAR").setType(OpenWire.Destination.DestinationType.QUEUE).build();
public void testPerformance() throws Exception {
OutputStream out = new BufferedOutputStream(new FileOutputStream(fileName));
CodedOutputStream cout = CodedOutputStream.newInstance(out);
StopWatch watch = createStopWatch("writer");
for (long i = 0; i < messageCount; i++) {
watch.start();
OpenWire.Message.Builder builder = OpenWire.Message.newBuilder()
.setDestination(destination)
.setPersistent(true)
.setCorrelationId("ABCD");
if (useProducerId) {
int producerCounter = (int) i;
builder = builder.setProducerId(1234)
.setProducerCounter(producerCounter);
}
OpenWire.Message message = builder.build();
if (verbose) {
System.out.println("Writing message: " + i + " = " + message);
}
int size = message.getSerializedSize();
cout.writeRawVarint32(size);
message.writeTo(cout);
watch.stop();
}
cout.flush();
out.close();
// now lets try read them!
StopWatch watch2 = createStopWatch("reader");
InputStream in = new BufferedInputStream(new FileInputStream(fileName));
CodedInputStream cin = CodedInputStream.newInstance(in);
for (long i = 0; i < messageCount; i++) {
watch2.start();
int size = cin.readRawVarint32();
int previous = cin.pushLimit(size);
//cin.setSizeLimit(size + 4);
OpenWire.Message message = OpenWire.Message.parseFrom(cin);
cin.resetBytesReadCounter();
cin.popLimit(previous);
if (verbose) {
System.out.println("Reading message: " + i + " = " + message);
}
if (doAssertions) {
if (useProducerId) {
assertEquals("message.getProducerId()", 1234, message.getProducerId());
assertEquals("message.getProducerCounter()", i, message.getProducerCounter());
}
assertEquals("message.getPersistent()", true, message.getPersistent());
assertEquals("message.getCorrelationId()", "ABCD", message.getCorrelationId());
OpenWire.Destination actualDestination = message.getDestination();
assertNotNull("message.getDestination() is null!", actualDestination);
assertEquals("destination.getName()", destination.getName(), actualDestination.getName());
assertEquals("destination.getType()", destination.getType(), actualDestination.getType());
}
watch2.stop();
}
in.close();
}
}