一、场景
这天遇到这个这样的需求,需要导入表格中的数据,还得支持带图片;这应该难不倒咱阿里的EasyExcel,打开官网文档一看哦豁,明确表示暂时不支持解析带图片的Excel......
好了,这下只能看POI了,然后想起来项目里引入的HuTools工具类,它应该封装好了吧;于是决定用HuTools里面的ExcelUtil来实现!
二、秀个骚操作,使用EasyExcel+HuTools(POI)实现解析
说明:因为EasyExcel实在太好用了,加上看完HuTools工具类的方法发现,图片解析是单独的,所以决定使用EasyExcel来解析数据,用HuTools工具类来解析里面的图片;如果表格数据过大则谨慎使用笔者这种方式哦。
1、定义EasyExcel解析用到的与表格映射的JavaBean
/**
* <p>
* 任务表格映射
* </p>
*
* @author kobe
* @since 2023-03-04
*/
@Data
public class TaskExcelDto implements Serializable {
private static final long serialVersionUID = 1L;
//这个注解表示解析的时候忽略
@ExcelIgnore
private Long id;
@ExcelProperty("姓名")
private String name;
@ExcelProperty("归属状态")
private String status;
@ExcelProperty("计划代码")
private String planCode;
@ExcelProperty("日期")
private Date issueDate;
@ExcelProperty("时间")
private String time;
@ExcelProperty("任务分类")
private String types;
@ExcelProperty("订单号")
private String orderNum;
@ExcelProperty("金额")
private Double money;
@ExcelProperty("提成")
private Double divvy;
@ExcelProperty("总金额")
private Double totalMoney;
@ExcelProperty("计划人")
private String planer;
@ExcelProperty("所设关键词")
private String keywords;
@ExcelIgnore
private String picture;
// 图片数据,临时变量
@ExcelIgnore
private PictureData pictureData;
}
2、实现代码
@ApiOperation("excel导入")
@PostMapping("v1/excel/import")
public JsonResult importByExcel(MultipartFile file) throws IOException {
List<TaskExcelDto> list = new ArrayList<>();
// EasyExcel解析普通数据(这里为了演示简单写一下,实战建议写个实现类)
EasyExcel.read(file.getInputStream(), TaskExcelDto.class, new AnalysisEventListener<TaskExcelDto>() {
@Override
public void invoke(TaskExcelDto data, AnalysisContext context) {
list.add(data);
}
@Override
public void doAfterAllAnalysed(AnalysisContext context) {
}
}).sheet().doRead();
// 使用HuTools工具类解析表格的图片
//注意这个方法只能读取浮动的图片,嵌入单元格的读取不到!
//注意这个方法只能读取浮动的图片,嵌入单元格的读取不到!
//注意这个方法只能读取浮动的图片,嵌入单元格的读取不到!
ExcelReader reader = ExcelUtil.getReader(file.getInputStream());
//key是图片位置,row_cell的格式;value是图片数据
Map<String, PictureData> picMap = ExcelPicUtil.getPicMap(reader.getWorkbook(), 0);
//这里只关心行数,把数据装到Bean里面去,也可用map在循环中取获取
picMap.forEach((k, v) -> {
String[] split = k.split(StrUtil.UNDERLINE);
Integer index = Integer.valueOf(split[0]);
list.get(index - 1).setPictureData(v);
});
//建议使用并行流上传图片,多线程加速
list.stream().parallel().forEach(i -> {
if (null == i.getPictureData()) {
return;
}
String mimeType = i.getPictureData().getMimeType();
String[] mimeTypes = mimeType.split(StrUtil.SLASH);
try {
String fileName = UUID.randomUUID() + "." + mimeTypes[1];
FileUploadResult fileUploadResult = super.uploadFile(i.getPictureData().getData(), fileName );
i.setPicture(fileUploadResult != null ? fileUploadResult.getWebPath());
} catch (IOException e) {
log.error("file upload error, {}", JSONUtil.toJsonStr(i));
}
});
// 调用业务代码保存数据
service.importByExcel(list);
return JsonResult.success();
}
3、注意:这个读取方法只能读取浮动的图片,嵌入单元格的读取不到!如果发现没读取到,检查一下Excel中图片在单元格中的样式,右击“切换为浮动图片”即可。