private boolean closeFile() {
boolean retval = false;
if (data.OpenedNewFile) {
try {
// Close the parent element
data.writer.write(
("</" + meta.getMainElement() + ">" + Const.CR)
.toCharArray()); //$NON-NLS-1$ //$NON-NLS-2$
if (meta.isZipped()) {
// System.out.println("close zip entry ");
data.zip.closeEntry();
// System.out.println("finish file...");
data.zip.finish();
data.zip.close();
} else {
data.writer.close();
}
// System.out.println("Closed file...");
retval = true;
} catch (Exception e) {
}
}
return retval;
}
public boolean init(StepMetaInterface smi, StepDataInterface sdi) {
meta = (XMLOutputMeta) smi;
data = (XMLOutputData) sdi;
if (super.init(smi, sdi)) {
data.splitnr = 0;
if (!meta.isDoNotOpenNewFileInit()) {
if (openNewFile()) {
data.OpenedNewFile = true;
return true;
} else {
logError("Couldn't open file " + meta.getFileName()); // $NON-NLS-1$
setErrors(1L);
stopAll();
}
} else return true;
}
return false;
}
public boolean processRow(StepMetaInterface smi, StepDataInterface sdi) throws KettleException {
meta = (XMLOutputMeta) smi;
data = (XMLOutputData) sdi;
Object[] r;
boolean result = true;
r = getRow(); // This also waits for a row to be finished.
if (first && meta.isDoNotOpenNewFileInit()) {
// no more input to be expected...
// In this case, no file was opened.
if (r == null) {
setOutputDone();
return false;
}
if (openNewFile()) {
data.OpenedNewFile = true;
} else {
logError("Couldn't open file " + meta.getFileName()); // $NON-NLS-1$
setErrors(1L);
return false;
}
}
if ((r != null
&& getLinesOutput() > 0
&& meta.getSplitEvery() > 0
&& (getLinesOutput() % meta.getSplitEvery()) == 0)) {
// Done with this part or with everything.
closeFile();
// Not finished: open another file...
if (r != null) {
if (!openNewFile()) {
logError(
"Unable to open new file (split #"
+ data.splitnr
+ "..."); //$NON-NLS-1$ //$NON-NLS-2$
setErrors(1);
return false;
}
}
}
if (r == null) // no more input to be expected...
{
setOutputDone();
return false;
}
writeRowToFile(getInputRowMeta(), r);
data.outputRowMeta = getInputRowMeta().clone();
meta.getFields(data.outputRowMeta, getStepname(), null, null, this);
putRow(data.outputRowMeta, r); // in case we want it to go further...
if (checkFeedback(getLinesOutput())) logBasic("linenr " + getLinesOutput()); // $NON-NLS-1$
return result;
}
public boolean openNewFile() {
boolean retval = false;
data.writer = null;
try {
FileObject file = KettleVFS.getFileObject(buildFilename(true), getTransMeta());
if (meta.isAddToResultFiles()) {
// Add this to the result file names...
ResultFile resultFile =
new ResultFile(
ResultFile.FILE_TYPE_GENERAL, file, getTransMeta().getName(), getStepname());
resultFile.setComment("This file was created with a xml output step"); // $NON-NLS-1$
addResultFile(resultFile);
}
OutputStream outputStream;
if (meta.isZipped()) {
OutputStream fos = KettleVFS.getOutputStream(file, false);
data.zip = new ZipOutputStream(fos);
File entry = new File(buildFilename(false));
ZipEntry zipentry = new ZipEntry(entry.getName());
zipentry.setComment("Compressed by Kettle"); // $NON-NLS-1$
data.zip.putNextEntry(zipentry);
outputStream = data.zip;
} else {
OutputStream fos = KettleVFS.getOutputStream(file, false);
outputStream = fos;
}
if (meta.getEncoding() != null && meta.getEncoding().length() > 0) {
logBasic("Opening output stream in encoding: " + meta.getEncoding()); // $NON-NLS-1$
data.writer = new OutputStreamWriter(outputStream, meta.getEncoding());
data.writer.write(XMLHandler.getXMLHeader(meta.getEncoding()).toCharArray());
} else {
logBasic(
"Opening output stream in default encoding : " + Const.XML_ENCODING); // $NON-NLS-1$
data.writer = new OutputStreamWriter(outputStream);
data.writer.write(XMLHandler.getXMLHeader(Const.XML_ENCODING).toCharArray());
}
// Add the name space if defined
StringBuffer nameSpace = new StringBuffer();
if ((meta.getNameSpace() != null) && (!"".equals(meta.getNameSpace()))) { // $NON-NLS-1$
nameSpace.append(" xmlns=\""); // $NON-NLS-1$
nameSpace.append(meta.getNameSpace());
nameSpace.append("\""); // $NON-NLS-1$
}
// OK, write the header & the parent element:
data.writer.write(
("<" + meta.getMainElement() + nameSpace.toString() + ">" + Const.CR)
.toCharArray()); //$NON-NLS-1$//$NON-NLS-2$
retval = true;
} catch (Exception e) {
logError("Error opening new file : " + e.toString()); // $NON-NLS-1$
}
// System.out.println("end of newFile(), splitnr="+splitnr);
data.splitnr++;
return retval;
}
public String buildFilename(boolean ziparchive) {
return meta.buildFilename(this, getCopy(), data.splitnr, ziparchive);
}
private void writeRowToFile(RowMetaInterface rowMeta, Object[] r) throws KettleException {
try {
if (first) {
data.formatRowMeta = rowMeta.clone();
first = false;
data.fieldnrs = new int[meta.getOutputFields().length];
for (int i = 0; i < meta.getOutputFields().length; i++) {
data.fieldnrs[i] =
data.formatRowMeta.indexOfValue(meta.getOutputFields()[i].getFieldName());
if (data.fieldnrs[i] < 0) {
throw new KettleException(
"Field ["
+ meta.getOutputFields()[i].getFieldName()
+ "] couldn't be found in the input stream!"); //$NON-NLS-1$ //$NON-NLS-2$
}
// Apply the formatting settings to the valueMeta object...
//
ValueMetaInterface valueMeta = data.formatRowMeta.getValueMeta(data.fieldnrs[i]);
XMLField field = meta.getOutputFields()[i];
valueMeta.setConversionMask(field.getFormat());
valueMeta.setLength(field.getLength(), field.getPrecision());
valueMeta.setDecimalSymbol(field.getDecimalSymbol());
valueMeta.setGroupingSymbol(field.getGroupingSymbol());
valueMeta.setCurrencySymbol(field.getCurrencySymbol());
}
}
if (meta.getOutputFields() == null || meta.getOutputFields().length == 0) {
/*
* Write all values in stream to text file.
*/
// OK, write a new row to the XML file:
data.writer.write(
(" <" + meta.getRepeatElement() + ">").toCharArray()); // $NON-NLS-1$ //$NON-NLS-2$
for (int i = 0; i < data.formatRowMeta.size(); i++) {
// Put a space between the XML elements of the row
//
if (i > 0) data.writer.write(' ');
ValueMetaInterface valueMeta = data.formatRowMeta.getValueMeta(i);
Object valueData = r[i];
writeField(valueMeta, valueData, valueMeta.getName());
}
} else {
/*
* Only write the fields specified!
*/
// Write a new row to the XML file:
data.writer.write(
(" <" + meta.getRepeatElement() + ">").toCharArray()); // $NON-NLS-1$ //$NON-NLS-2$
for (int i = 0; i < meta.getOutputFields().length; i++) {
XMLField outputField = meta.getOutputFields()[i];
if (i > 0) data.writer.write(' '); // a space between
// elements
ValueMetaInterface valueMeta = data.formatRowMeta.getValueMeta(data.fieldnrs[i]);
Object valueData = r[data.fieldnrs[i]];
String elementName = outputField.getElementName();
if (Const.isEmpty(elementName)) {
elementName = outputField.getFieldName();
}
if (!(valueMeta.isNull(valueData) && meta.isOmitNullValues())) {
writeField(valueMeta, valueData, elementName);
}
}
}
data.writer.write(
(" </" + meta.getRepeatElement() + ">").toCharArray()); // $NON-NLS-1$ //$NON-NLS-2$
data.writer.write(Const.CR.toCharArray());
} catch (Exception e) {
throw new KettleException(
"Error writing XML row :"
+ e.toString()
+ Const.CR
+ "Row: "
+ getInputRowMeta().getString(r),
e); //$NON-NLS-1$ //$NON-NLS-2$
}
incrementLinesOutput();
}