将CSV文件数据转换为List的工具类

Jason-Cheung

已于 2022-08-10 11:35:31 修改

阅读量1.2k

点赞数

分类专栏： Java开发工具类文章标签： java 开发语言

于 2022-08-10 11:12:24 首次发布

本文链接：https://blog.csdn.net/z19970223q/article/details/126261401

版权

Java开发工具类专栏收录该内容

1 篇文章 0 订阅

订阅专栏

一、导入OpenCSV的jar包

<dependency>
	<groupId>com.opencsv</groupId>
	<artifactId>opencsv</artifactId>
	<version>5.6</version>
</dependency>

二、创建csv文件解析工具类

public class CsvUtils {
    /**
     * 获取CSV文件数据
     * @param filePath csv文件所在的路径
     * @param clazz 按文件标题映射的类，属性使用注解@CsvBindByName(column = "xxx")
     * @return 返回解析后的数据集合
     * @param <T>
     */
    public static <T> List<T> getCsvData(String filePath, Class<T> clazz) {
        InputStreamReader in;
        CsvToBean<T> csvToBean;
        try {
            in = new InputStreamReader(Files.newInputStream(Paths.get(filePath)), StandardCharsets.UTF_8);
            HeaderColumnNameMappingStrategy<T> strategy = new HeaderColumnNameMappingStrategy<>();
            strategy.setType(clazz);
            csvToBean = new CsvToBeanBuilder<T>(in).withMappingStrategy(strategy).build();
        } catch (Exception e) {
            return null;
        }
        return csvToBean.parse();
    }
}

三、编写csv文件列标题的映射对象

1. csv文件示例

user name,userAge,user-sex
张三,18,男
田七,19,女

2. 映射类

/**
 * @author xxx
 * @date xxxx-xx-xx
 * @description CSV文件标题映射对象
 * 这是一个示例
 */
public class UserInfoCsvBO {

    @CsvBindByName(column = "user name")
    private String name;

    @CsvBindByName(column = "userAge")
    private String age;

    @CsvBindByName(column = "user-sex")
    private String sex;
    
}

四、使用工具类

private static final String CSV_FILE_PATH = "/home/data/csv/csv_test.csv";

/**
*获取用户信息CSV文件的数据
*/
public List<UserInfoCsvBO> getUserInfo(){
	return CsvUtils.getCsvData(CSV_FILE_PATH, UserInfoCsvBO.class);
}