blob: b5e7b93b6851291d57d97dc6d92312648cbe809e [file] [log] [blame]
/*
* Copyright (C) 2010 Google Inc.
*
* 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 benchmarks;
import com.google.caliper.Param;
import com.google.caliper.Runner;
import com.google.caliper.SimpleBenchmark;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.FileInputStream;
import java.io.IOException;
import java.lang.reflect.Constructor;
import java.util.Arrays;
import java.util.List;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;
import org.w3c.dom.Document;
import org.w3c.dom.Node;
import org.xml.sax.Attributes;
import org.xml.sax.SAXException;
import org.xml.sax.helpers.DefaultHandler;
import org.xmlpull.v1.XmlPullParser;
public class XmlParseBenchmark extends SimpleBenchmark {
@Param String xmlFile;
ByteArrayInputStream inputStream;
static List<String> xmlFileValues = Arrays.asList(
"/etc/apns-conf.xml",
"/etc/media_profiles.xml",
"/etc/permissions/features.xml"
);
private SAXParser saxParser;
private DocumentBuilder documentBuilder;
private Constructor<? extends XmlPullParser> kxmlConstructor;
private Constructor<? extends XmlPullParser> expatConstructor;
@SuppressWarnings("unchecked")
@Override protected void setUp() throws Exception {
byte[] xmlBytes = getXmlBytes();
inputStream = new ByteArrayInputStream(xmlBytes);
inputStream.mark(xmlBytes.length);
SAXParserFactory saxParserFactory = SAXParserFactory.newInstance();
saxParser = saxParserFactory.newSAXParser();
DocumentBuilderFactory builderFactory = DocumentBuilderFactory.newInstance();
documentBuilder = builderFactory.newDocumentBuilder();
kxmlConstructor = (Constructor) Class.forName("org.kxml2.io.KXmlParser").getConstructor();
expatConstructor = (Constructor) Class.forName("org.apache.harmony.xml.ExpatPullParser")
.getConstructor();
}
private byte[] getXmlBytes() throws IOException {
FileInputStream fileIn = new FileInputStream(xmlFile);
ByteArrayOutputStream bytesOut = new ByteArrayOutputStream();
int count;
byte[] buffer = new byte[1024];
while ((count = fileIn.read(buffer)) != -1) {
bytesOut.write(buffer, 0, count);
}
fileIn.close();
return bytesOut.toByteArray();
}
public int timeSax(int reps) throws IOException, SAXException {
int elementCount = 0;
for (int i = 0; i < reps; i++) {
inputStream.reset();
ElementCounterSaxHandler elementCounterSaxHandler = new ElementCounterSaxHandler();
saxParser.parse(inputStream, elementCounterSaxHandler);
elementCount += elementCounterSaxHandler.elementCount;
}
return elementCount;
}
private static class ElementCounterSaxHandler extends DefaultHandler {
int elementCount = 0;
@Override public void startElement(String uri, String localName,
String qName, Attributes attributes) {
elementCount++;
}
}
public int timeDom(int reps) throws IOException, SAXException {
int elementCount = 0;
for (int i = 0; i < reps; i++) {
inputStream.reset();
Document document = documentBuilder.parse(inputStream);
elementCount += countDomElements(document.getDocumentElement());
}
return elementCount;
}
private int countDomElements(Node node) {
int result = 0;
for (; node != null; node = node.getNextSibling()) {
if (node.getNodeType() == Node.ELEMENT_NODE) {
result++;
}
result += countDomElements(node.getFirstChild());
}
return result;
}
public int timeExpat(int reps) throws Exception {
return testXmlPull(expatConstructor, reps);
}
public int timeKxml(int reps) throws Exception {
return testXmlPull(kxmlConstructor, reps);
}
private int testXmlPull(Constructor<? extends XmlPullParser> constructor, int reps)
throws Exception {
int elementCount = 0;
for (int i = 0; i < reps; i++) {
inputStream.reset();
XmlPullParser xmlPullParser = constructor.newInstance();
xmlPullParser.setInput(inputStream, "UTF-8");
int type;
while ((type = xmlPullParser.next()) != XmlPullParser.END_DOCUMENT) {
if (type == XmlPullParser.START_TAG) {
elementCount++;
}
}
}
return elementCount;
}
public static void main(String[] args) {
Runner.main(XmlParseBenchmark.class, args);
}
}