private String pullNextXmlChunkFromTopElementOnStack(XMLChunkerState data)
throws KettleException {
Stack<String> elementStack = data.getElementStack();
XMLStreamReader xmlStreamReader = data.getXmlStreamReader();
int elementStackDepthOnEntry = elementStack.size();
StringWriter stringWriter = new StringWriter();
try {
XMLStreamWriter xmlStreamWriter =
data.getXmlOutputFactory().createXMLStreamWriter(stringWriter);
xmlStreamWriter.writeStartDocument(CharEncoding.UTF_8, "1.0");
// put the current element on because presumably it's the open element for the one
// that is being looked for.
XmlReaderToWriter.write(xmlStreamReader, xmlStreamWriter);
while (xmlStreamReader.hasNext() & elementStack.size() >= elementStackDepthOnEntry) {
switch (xmlStreamReader.next()) {
case XMLStreamConstants.END_DOCUMENT:
break; // handled below explicitly.
case XMLStreamConstants.END_ELEMENT:
elementStack.pop();
XmlReaderToWriter.write(xmlStreamReader, xmlStreamWriter);
break;
case XMLStreamConstants.START_ELEMENT:
elementStack.push(xmlStreamReader.getLocalName());
XmlReaderToWriter.write(xmlStreamReader, xmlStreamWriter);
break;
default:
XmlReaderToWriter.write(xmlStreamReader, xmlStreamWriter);
break;
}
}
xmlStreamWriter.writeEndDocument();
xmlStreamWriter.close();
} catch (Exception e) {
throw new KettleException("unable to process a chunk of the xero xml stream", e);
}
return stringWriter.toString();
}
// public static void write(OutputStream stream, Element element, int hashLength) throws
// XMLStreamException {
// XMLOutputFactory xof = XMLOutputFactory.newInstance();
// IndentingXMLStreamWriter xtw = new
// IndentingXMLStreamWriter(xof.createXMLStreamWriter(stream));
// writeXMLStreamWriter(xtw, element, hashLength);
// }
//
private static void writeXMLStreamWriter(XMLStreamWriter xtw, Element element, int hashLength)
throws XMLStreamException {
if (hashLength > 0) {
String hash = element.getHash(hashLength);
Element hashElement = new Element(CRC_ELEMENT);
hashElement.setText(hash);
element.addChild(hashElement);
}
writeElement(xtw, element);
xtw.writeEndDocument();
xtw.flush();
xtw.close();
if (hashLength > 0) {
element.removeChildren(CRC_ELEMENT);
}
}
@Override
public int writeData(OutputStream out, DbData data) throws Exception {
XMLStreamWriter sw = _staxOutFactory.createXMLStreamWriter(out, "UTF-8");
sw.writeStartDocument();
sw.writeStartElement(FIELD_TABLE);
Iterator<DbRow> it = data.rows();
while (it.hasNext()) {
DbRow row = it.next();
sw.writeStartElement(FIELD_ROW); // <row>
sw.writeStartElement(DbRow.Field.id.name());
sw.writeCharacters(String.valueOf(row.getId()));
sw.writeEndElement();
sw.writeStartElement(DbRow.Field.firstname.name());
sw.writeCharacters(row.getFirstname());
sw.writeEndElement();
sw.writeStartElement(DbRow.Field.lastname.name());
sw.writeCharacters(row.getLastname());
sw.writeEndElement();
sw.writeStartElement(DbRow.Field.zip.name());
sw.writeCharacters(String.valueOf(row.getZip()));
sw.writeEndElement();
sw.writeStartElement(DbRow.Field.street.name());
sw.writeCharacters(row.getStreet());
sw.writeEndElement();
sw.writeStartElement(DbRow.Field.city.name());
sw.writeCharacters(row.getCity());
sw.writeEndElement();
sw.writeStartElement(DbRow.Field.state.name());
sw.writeCharacters(row.getState());
sw.writeEndElement();
sw.writeEndElement(); // </row>
}
sw.writeEndElement();
sw.writeEndDocument();
sw.close();
return -1;
}
