1 /*
2 * (c) Copyright 2006-2020 by rapiddweller GmbH & Volker Bergmann. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, is permitted under the terms of the
6 * GNU General Public License.
7 *
8 * For redistributing this software or a derivative work under a license other
9 * than the GPL-compatible Free Software License as defined by the Free
10 * Software Foundation or approved by OSI, you must first obtain a commercial
11 * license to this software product from rapiddweller GmbH & Volker Bergmann.
12 *
13 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
14 * WITHOUT A WARRANTY OF ANY KIND. ALL EXPRESS OR IMPLIED CONDITIONS,
15 * REPRESENTATIONS AND WARRANTIES, INCLUDING ANY IMPLIED WARRANTY OF
16 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT, ARE
17 * HEREBY EXCLUDED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
18 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
19 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
20 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
21 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
22 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
23 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
24 * POSSIBILITY OF SUCH DAMAGE.
25 */
26
27 package com.rapiddweller.platform.csv;
28
29 import com.rapiddweller.benerator.engine.BeneratorContext;
30 import com.rapiddweller.benerator.factory.DataSourceProvider;
31 import com.rapiddweller.common.Converter;
32 import com.rapiddweller.common.converter.ArrayConverter;
33 import com.rapiddweller.format.DataSource;
34 import com.rapiddweller.format.csv.CSVSource;
35 import com.rapiddweller.format.util.ConvertingDataSource;
36 import com.rapiddweller.format.util.OffsetDataSource;
37
38 /**
39 * {@link DataSourceProvider} which creates array {@link Iterable}s for CSV files.<br/><br/>
40 * Created: 19.07.2011 08:23:39
41 *
42 * @author Volker Bergmann
43 * @since 0.7.0
44 */
45 public class CSVArraySourceProvider implements DataSourceProvider<Object[]> {
46
47 private final Converter<String, ?> preprocessor;
48 private final boolean rowBased;
49 private final char separator;
50 private final String encoding;
51
52 /**
53 * Instantiates a new Csv array source provider.
54 *
55 * @param type the type
56 * @param preprocessor the preprocessor
57 * @param rowBased the row based
58 * @param separator the separator
59 * @param encoding the encoding
60 */
61 public CSVArraySourceProvider(String type,
62 Converter<String, ?> preprocessor,
63 boolean rowBased, char separator,
64 String encoding) {
65 this.preprocessor = preprocessor;
66 this.rowBased = rowBased;
67 this.separator = separator;
68 this.encoding = encoding;
69 }
70
71 @Override
72 @SuppressWarnings({"unchecked", "rawtypes"})
73 public DataSource<Object[]> create(String uri, BeneratorContext context) {
74 DataSource<String[]> source;
75 source = new CSVSource(uri, separator, encoding, true, rowBased);
76 Converter<String[], Object[]> converter =
77 new ArrayConverter(String.class, Object.class, preprocessor);
78 DataSource<Object[]> result =
79 new ConvertingDataSource<>(source, converter);
80 result = new OffsetDataSource<>(result,
81 1); // offset = 1 in order to skip header row
82 return result;
83 }
84
85 }