blob: d04cfe8f064d1decfebf32af822141ca4c60efa1 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
package com.benburwell.planes.data;
import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVParser;
import org.apache.commons.csv.CSVRecord;
import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;
/**
* @author ben
*/
public class CSVObjectStore<T extends AbstractCSVReader> {
private List<T> objects = new ArrayList<>();
public void readFromFile(String fileName, Class<T> klass) throws IOException, IllegalAccessException, InstantiationException {
File csvData = new File(fileName);
CSVParser parser = CSVParser.parse(csvData, Charset.defaultCharset(), CSVFormat.RFC4180.withFirstRecordAsHeader());
for (CSVRecord record : parser) {
T obj = klass.newInstance();
obj.readRecord(record);
this.objects.add(obj);
}
}
public void readFromResource(String resourceName, Class<T> klass) throws IOException, IllegalAccessException, InstantiationException {
InputStream stream = this.getClass().getResourceAsStream(resourceName);
Reader csvData = new BufferedReader(new InputStreamReader(stream));
CSVParser parser = new CSVParser(csvData, CSVFormat.RFC4180.withFirstRecordAsHeader());
for (CSVRecord record : parser) {
T obj = klass.newInstance();
obj.readRecord(record);
this.objects.add(obj);
}
}
public List<T> getObjects() {
return this.objects;
}
}
|