springboot读resource下的csv文件

首先，有一个注意点就是，springboot项目最后打成一个jar包，tomcat容器内嵌在jar包里，我们从代码里去读resource下的文件的方式和传统的以web容器目录形式部署的服务器肯定是不一样的。

 

		
			org.apache.commons
			commons-csv
			1.5
		

 

import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVParser;
import org.apache.commons.csv.CSVRecord;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.io.ClassPathResource;
import java.io.*;
import java.util.*;

/**
 * csv文件读写工具类
 *
 * @author by YangLD
 * @date 2018/7/10
 */
public class CsvUtils {

    private static final Logger logger = LoggerFactory.getLogger("CsvUtils.class");




    /**
     * 读取csv文件  (一次性读取文件不宜过大)
     *
     * @param filePath 文件路径 pinyin/xxx.csv
     * @param headers  csv列头
     * @return CSVRecord 列表
     **/
    public static Map readCSV(String filePath, String[] headers){
        //创建CSVFormat
        CSVFormat format = CSVFormat.DEFAULT.withHeader(headers);
        Map resultMap = new HashMap<>();
        InputStreamReader inputStreamReader = null;
        CSVParser parser = null;
        try {
            ClassPathResource classPathResource = new ClassPathResource(filePath);
            //获取文件流
            InputStream fileStream = classPathResource.getInputStream();
            inputStreamReader = new InputStreamReader(fileStream,"UTF-8");
            //创建CSVParser对象
            parser = new CSVParser(inputStreamReader, format);
            int rowIndex = 0;
            for (CSVRecord record : parser) {
                if(rowIndex > 0){
                    resultMap.put(record.get(0),record.get(1).split(","));
                }
                rowIndex++;
            }
            parser.close();
            inputStreamReader.close();
        } catch (Exception e) {
            logger.error("CsvUtils.readCSV,读取csv文件,message:{}", e.getMessage(), e);

        }finally {
            closeStream(inputStreamReader, parser);
        }
        return resultMap;
    }

    private static void closeStream(InputStreamReader inputStreamReader, CSVParser parser) {
        if(inputStreamReader != null){
            try {
                inputStreamReader.close();
            } catch (IOException e) {
                logger.error("文件打开异常",e.getMessage());
            }
        }
        if(parser != null){
            try {
                parser.close();
            } catch (IOException e) {
                logger.error("文件打开异常",e.getMessage());
            }
        }
    }



}