springboot讀resource下的csv文件

首先，有一個注意點就是，springboot項目最後打成一個jar包，tomcat容器內嵌在jar包裏，我們從代碼裏去讀resource下的文件的方式和傳統的以web容器目錄形式部署的服務器肯定是不一樣的。

 

<!--		讀寫csv-->
		<dependency>
			<groupId>org.apache.commons</groupId>
			<artifactId>commons-csv</artifactId>
			<version>1.5</version>
		</dependency>

 

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<String,String[]> readCSV(String filePath, String[] headers){
        //創建CSVFormat
        CSVFormat format = CSVFormat.DEFAULT.withHeader(headers);
        Map<String,String[]> 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());
            }
        }
    }



}