Java读取csv文件

最新推荐文章于 2024-05-22 18:44:09 发布

IT二叔

最新推荐文章于 2024-05-22 18:44:09 发布

阅读量587

点赞数 1

分类专栏： java 文章标签： java spring log4j

本文链接：https://blog.csdn.net/weixin_44219219/article/details/129265421

版权

java 专栏收录该内容

31 篇文章 3 订阅

订阅专栏

该文章介绍了一个Java工具类，用于将CSV文件或MultipartFile对象解析为自定义的实体类。通过OpenCSV库，实现了CSV数据到对象的映射，支持GBK和UTF-8编码。实体类使用了CsvBindByName注解来指定列映射。

摘要由CSDN通过智能技术生成

1.导包

  <!-- csv文件解析依赖 -->
        <dependency>
            <groupId>com.opencsv</groupId>
            <artifactId>opencsv</artifactId>
            <version>5.4</version>
        </dependency>

csv工具类

package com.jxy.common.utils.csv;
import com.opencsv.bean.CsvToBean;
import com.opencsv.bean.CsvToBeanBuilder;
import com.opencsv.bean.HeaderColumnNameMappingStrategy;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.springframework.web.multipart.MultipartFile;
import java.io.*;
import java.nio.charset.Charset;
import java.util.*;
/**
 * @Author: 二叔
 * @Date: 2023/02/28/9:44
 */

public class CsvUtils {
    private static final Logger logger = LogManager.getLogger(CsvUtils.class);
    /**
     * 解析csv文件并转成实体类
     * @param file  csv文件
     * @param clazz 类
     * @param <T>   泛型
     * @return 泛型bean集合
     */
    public static <T> List<T> csvToEntity(MultipartFile file, Class<T> clazz) {
        InputStreamReader in = null;
        CsvToBean<T> csvToBean = null;
        try {
            in = new InputStreamReader(file.getInputStream(), Charset.forName("GBK"));
            HeaderColumnNameMappingStrategy<T> strategy = new HeaderColumnNameMappingStrategy<>();
            strategy.setType(clazz);
            csvToBean = new CsvToBeanBuilder<T>(in).withMappingStrategy(strategy).build();
        } catch (Exception e) {
            logger.error("数据转化失败");
            return null;
        }
        return csvToBean.parse();
    }

    public static <T> List<T> csvToEntity(String filePath, Class<T> clazz) throws FileNotFoundException {
        InputStreamReader in = null;
        CsvToBean<T> csvToBean = null;
        try {
            in = new InputStreamReader(new FileInputStream(filePath), Charset.forName("UTF-8"));
            HeaderColumnNameMappingStrategy<T> strategy = new HeaderColumnNameMappingStrategy<>();
            strategy.setType(clazz);
            csvToBean = new CsvToBeanBuilder<T>(in).withMappingStrategy(strategy).build();
        } catch (Exception e) {
            logger.error("数据转化失败");
            return null;
        }
        return csvToBean.parse();
    }

}

实体类

package com.jxy.entity;

import com.opencsv.bean.CsvBindByName;
import lombok.Data;

/**
 * @Author: 二叔
 * @Date: 2023/02/28/16:38
 */
@Data
public class CsvTest {
    @CsvBindByName(column = "name")
    private String name;

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

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

测试

  @GetMapping("csvparse")
    public AjaxResult download(String url) throws Exception {
        String localFile = downLoadResources.getLocalFile(url);
        List<CsvTest> csvTests = CsvUtils.csvToEntity(localFile, CsvTest.class);
        System.out.println(csvTests);
        return AjaxResult.success();
    }