org.apache.tika.eval.io.XMLLogReader Maven / Gradle / Ivy
/*
* 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.tika.eval.io;
import javax.xml.stream.XMLInputFactory;
import javax.xml.stream.XMLStreamConstants;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.sql.SQLException;
import org.apache.log4j.Level;
import org.apache.tika.io.IOUtils;
import org.apache.tika.parser.ParseContext;
import org.apache.tika.utils.XMLReaderUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class XMLLogReader {
private static final Logger LOG = LoggerFactory.getLogger(XMLLogReader.class);
//class that wraps a logger's xml output
//into a single xml parseable input stream.
public void read(InputStream xmlLogFileIs, XMLLogMsgHandler handler) throws XMLStreamException {
InputStream is = new LogXMLWrappingInputStream(xmlLogFileIs);
XMLInputFactory factory = XMLReaderUtils.getXMLInputFactory();
XMLStreamReader reader = factory.createXMLStreamReader(is);
Level level = null;
while (reader.hasNext()) {
reader.next();
switch (reader.getEventType()) {
case XMLStreamConstants.START_ELEMENT :
if ("event".equals(reader.getLocalName())) {
level = Level.toLevel(reader.getAttributeValue("", "level"), Level.DEBUG);
} else if ("message".equals(reader.getLocalName())) {
try {
handler.handleMsg(level, reader.getElementText());
} catch (IOException e) {
LOG.warn("Error parsing: {}", reader.getElementText());
} catch (SQLException e) {
LOG.warn("SQLException: {}", e.getMessage());
}
}
break;
case XMLStreamConstants.END_ELEMENT :
if ("event".equals(reader.getLocalName())) {
level = null;
} else if ("message".equals(reader.getLocalName())) {
//do we care any more?
}
break;
};
}
}
class LogXMLWrappingInputStream extends InputStream {
//plagiarized from log4j's chainsaw
private final static String HEADER =
""
+ "";
private static final String FOOTER = " ";
private InputStream[] streams;
int currentStreamIndex = 0;
private LogXMLWrappingInputStream(InputStream xmlLogFileIs){
streams = new InputStream[3];
streams[0] = new ByteArrayInputStream(HEADER.getBytes(IOUtils.UTF_8));
streams[1] = xmlLogFileIs;
streams[2] = new ByteArrayInputStream(FOOTER.getBytes(IOUtils.UTF_8));
}
@Override
public int read() throws IOException {
int c = streams[currentStreamIndex].read();
if (c < 0) {
IOUtils.closeQuietly(streams[currentStreamIndex]);
while (currentStreamIndex < streams.length-1) {
currentStreamIndex++;
int tmpC = streams[currentStreamIndex].read();
if (tmpC < 0) {
IOUtils.closeQuietly(streams[currentStreamIndex]);
} else {
return tmpC;
}
}
return -1;
}
return c;
}
}
}