SuperCSV从多个文件读取并解析为一个bean对象

时间:2015-01-05 09:33:46

标签: java csv supercsv

我目前正尝试使用beanReader读取多个CSV文件,然后从每个文件中取出几列并将它们解析为一个bean。

到目前为止,我似乎无法将不同文件中的列解析为一个bean对象。这对ICsvBeanReader来说是否可行?

1 个答案:

答案 0 :(得分:0)

是的,这是可能的:)从Super CSV 2.2.0开始,您可以读入现有的bean(请参阅javadoc)。

以下示例同时使用3个读取器(在3个不同的文件上运行) - 第一个读取器用于创建bean,另外两个读取器只更新现有bean。此方法假定每个文件具有相同的行数(并且每个行号代表同一个人)。如果他们不这样做,但他们共享一些唯一标识符,则必须首先将第一个文件中的所有记录读入内存,然后从标识符的第二个/第三个匹配更新。

我试图让它变得有点聪明,所以你不必对名称映射进行硬编码 - 它只是将它不知道的标题置空(这样Super CSV就不会尝试映射bean中不存在的字段 - 请参阅网站上的partial reading examples。当然,只有当你的文件有标题时才会有效 - 否则你只需要在适当的位置用空值对映射数组进行硬编码。

人豆

public class Person {
    private String firstName;
    private String sex;
    private String country;
    // getters/setters
}

示例代码

public class Example {

    private static final String FILE1 = "firstName,lastName\nJohn,Smith\nSally,Jones";
    private static final String FILE2 = "age,sex\n21,male\n24,female";
    private static final String FILE3 = "city,country\nBrisbane,Australia\nBerlin,Germany";
    private static final List<String> DESIRED_HEADERS = Arrays.asList("firstName", "sex", "country");

    @Test
    public void testMultipleFiles() throws Exception {

        try (
            ICsvBeanReader reader1 = new CsvBeanReader(new StringReader(FILE1), CsvPreference.STANDARD_PREFERENCE);
            ICsvBeanReader reader2 = new CsvBeanReader(new StringReader(FILE2), CsvPreference.STANDARD_PREFERENCE);
            ICsvBeanReader reader3 = new CsvBeanReader(new StringReader(FILE3), CsvPreference.STANDARD_PREFERENCE);){

            String[] mapping1 = getNameMappingFromHeader(reader1);
            String[] mapping2 = getNameMappingFromHeader(reader2);
            String[] mapping3 = getNameMappingFromHeader(reader3);

            Person person;
            while((person = reader1.read(Person.class, mapping1)) != null){
                reader2.read(person, mapping2);
                reader3.read(person, mapping3);
                System.out.println(person);
            }
        } 

    }

    private String[] getNameMappingFromHeader(ICsvBeanReader reader) throws IOException{
        String[] header = reader.getHeader(true);

        // only read in the desired fields (set unknown headers to null to ignore)
        for (int i = 0; i < header.length; i++){
            if (!DESIRED_HEADERS.contains(header[i])){
                header[i] = null;
            }
        }

        return header;
    }
}

<强>输出

Person [firstName=John, sex=male, country=Australia]
Person [firstName=Sally, sex=female, country=Germany]