/*
* (c) Copyright 2011 by Volker Bergmann. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, is permitted under the terms of the
* GNU General Public License (GPL).
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* WITHOUT A WARRANTY OF ANY KIND. ALL EXPRESS OR IMPLIED CONDITIONS,
* REPRESENTATIONS AND WARRANTIES, INCLUDING ANY IMPLIED WARRANTY OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT, ARE
* HEREBY EXCLUDED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
package org.databene.platform.csv;
import org.databene.benerator.engine.BeneratorContext;
import org.databene.benerator.factory.DataSourceProvider;
import org.databene.commons.converter.ArrayConverter;
import org.databene.commons.Converter;
import org.databene.document.csv.CSVSource;
import org.databene.webdecs.DataSource;
import org.databene.webdecs.util.ConvertingDataSource;
import org.databene.webdecs.util.OffsetDataSource;
/**
* {@link DataSourceProvider} which creates array {@link Iterable}s for CSV files.<br/><br/>
* Created: 19.07.2011 08:23:39
* @since 0.7.0
* @author Volker Bergmann
*/
public class CSVArraySourceProvider implements DataSourceProvider<Object[]> {
private Converter<String, ?> preprocessor;
private boolean rowBased;
private char separator;
private String encoding;
public CSVArraySourceProvider(String type, Converter<String, ?> preprocessor, boolean rowBased, char separator, String encoding) {
this.preprocessor = preprocessor;
this.rowBased = rowBased;
this.separator = separator;
this.encoding = encoding;
}
@SuppressWarnings({ "unchecked", "rawtypes" })
public DataSource<Object[]> create(String uri, BeneratorContext context) {
DataSource<String[]> source;
source = new CSVSource(uri, separator, encoding, true, rowBased);
Converter<String[], Object[]> converter = new ArrayConverter(String.class, Object.class, preprocessor);
DataSource<Object[]> result = new ConvertingDataSource<String[], Object[]>(source, converter);
result = new OffsetDataSource<Object[]>(result, 1); // offset = 1 in order to skip header row
return result;
}
}