src/main/java/com/benburwell/planes/data/CSVObjectStore.java


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;
    }
}