导入相关依赖,依赖hutool工具;
<dependency>
<groupId>cn.hutool</groupId>
<artifactId>hutool-all</artifactId>
<version>5.7.22</version>
</dependency>
colntoller
@PostMapping("/upload")
public String csvFileUpload(@RequestParam(value = "classId") Long classId, @RequestPart("file") MultipartFile file){
return classService.uploadAndParseCsv(Long userId, MultipartFile file)
}
service
public List<UserInfo> uploadAndParseCsv(Long classId, MultipartFile file) {
// 1. 校验文件
String suffix = ".csv";
Assert.notNull(file, "请上传csv文件!");
Assert.isTrue(Objects.requireNonNull(file.getOriginalFilename()).endsWith(suffix), "未解析到相关csv文件!");
Assert.isFalse(file.getSize() >= 10L * 1024 * 1024, "文件不能超过10M, 请重新上传!");
// 2. 读取file 并转换为BufferReader
BufferedReader bufferedReader;
try {
// window word/wps下 修改csv默认使用GB2312编码
bufferedReader = new BufferedReader(new InputStreamReader(file.getInputStream(), Charset.forName("GB2312")));
} catch (IOException e) {
throw new RuntimeException(e.getMessage());
}
// 3. 读取每行内容,并解析为指定对象
// 此处使用hutool工具
CsvReader reader = CsvUtil.getReader();
// 设置读取文件内容首行为标题
reader.setContainsHeader(true);
List<UserInfo> result = new ArrayList<>();
reader.read(bufferedReader, (row) -> {
Assert.isFalse(row.getOriginalLineNumber() > 10000, "用户记录不能超过10000条!");
// 解析为指定对象
UserInfo info = row.toBean(UserInfo.class);
// 校验必填字段
boolean convertSuccess = null == info.getName() || null == info.getAge()
|| info.getPhoneNumber();
Assert.isFalse(convertSuccess, String.format("第 %s 行; 请提供必须导入字段内容!"));
info.setClassId(calssId);
result.add(info);
})
return result;
}
@Data
public class UserInfo {
// 将对应标题姓名列解析为name属性
@Alias("姓名")
private String name;
@Alias("年龄")
private Integer age;
@Alias("电话号码")
private Integer phoneNumber;
private Long classId;
}