org.databene.formats.csv.ArrayCSVWriter Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of databene-formats Show documentation
Show all versions of databene-formats Show documentation
'Databene Formats' is an open source software library for supporting
data file and other formats like CSV, fixed width files, XLS, Properties and Regex.
It is designed for multithreaded use and high performance.
/*
* Copyright (C) 2011-2015 Volker Bergmann ([email protected]).
* All rights reserved.
*
* 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 org.databene.formats.csv;
import org.databene.commons.Context;
import org.databene.commons.ConversionException;
import org.databene.commons.Converter;
import org.databene.commons.SystemInfo;
import org.databene.commons.converter.ToStringConverter;
import org.databene.formats.script.AbstractScript;
import org.databene.formats.script.ConstantScript;
import org.databene.formats.script.Script;
import org.databene.formats.script.ScriptException;
import org.databene.formats.script.ScriptUtil;
import org.databene.formats.script.ScriptedDocumentWriter;
import java.io.Writer;
import java.io.IOException;
/**
* Writes arrays as CSV rows.
*
* Created: 06.06.2007 19:35:29
* @author Volker Bergmann
*/
public class ArrayCSVWriter extends ScriptedDocumentWriter {
public ArrayCSVWriter(Writer out) {
this(out, ',');
}
public ArrayCSVWriter(Writer out, char separator) {
this(out, separator, (Script)null, (Script)null);
}
public ArrayCSVWriter(Writer out, char separator, String ... columnHeads) {
this(
out,
separator,
new ConstantScript(CSVUtil.formatHeaderWithLineFeed(separator, columnHeads)),
null);
}
public ArrayCSVWriter(Writer out, char separator, String headerScriptUrl, String footerScriptUrl)
throws IOException {
this(
out,
separator,
(headerScriptUrl != null ? ScriptUtil.readFile(headerScriptUrl) : null),
(footerScriptUrl != null ? ScriptUtil.readFile(footerScriptUrl) : null)
);
}
public ArrayCSVWriter(Writer out, char separator, Script headerScript, Script footerScript) {
super(out, headerScript, new ArrayCSVScript(separator), footerScript);
}
// ArrayCSVScript ---------------------------------------------------------------------------------------------
private static class ArrayCSVScript extends AbstractScript {
private Converter converter;
private char separator;
public ArrayCSVScript(char separator) {
this.separator = separator;
this.converter = new ToStringConverter();
}
@Override
public void execute(Context context, Writer out) throws IOException, ScriptException {
try {
Object[] cellsOfCurrentRow = (Object[]) context.get("part");
String text = converter.convert(cellsOfCurrentRow[0]);
out.write(CSVUtil.renderCell(text, separator));
for (int i = 1; i < cellsOfCurrentRow.length; i++) {
out.write(separator);
out.write(converter.convert(cellsOfCurrentRow[i]));
}
out.write(SystemInfo.getLineSeparator());
} catch (ConversionException e) {
throw new ScriptException(e);
}
}
}
}