首先,有一个注意点就是,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());
}
}
}
}