orcl-zung commented 5 months ago

建议先去看文档

异常代码

  @Override
    @PostMapping("/test")
    public HttpResult<String> test(@RequestParam("count") Integer count) throws Exception {
        HttpServletResponse res = getResponse();
        res.setContentType("application/force-download");
        String fileName = URLEncoder.encode("订单列表.xlsx", StandardCharsets.UTF_8.toString());
        res.setHeader("Content-Disposition", "attachment;filename=" + fileName);
        ExcelWriter excelWriter = EasyExcelFactory.write(res.getOutputStream()).build();
        try {
            List<TestVo> list = new ArrayList<>();

            File file = new File("D:\\wallpaper\\1347724.png");
            URL url = file.toURL();
            for (int i = 1; i <= count; i++) {
                TestVo testVo = new TestVo();

                testVo.setTestString("row " + i);
                testVo.setInchImage(url);
                list.add(testVo);
            }

            WriteSheet writeSheet = EasyExcelFactory.writerSheet(0, "订单").head(TestVo.class).build();
            excelWriter.write(list, writeSheet);

            return HttpResult.success();
        } catch (Exception e) {
            logger.error("列表导出 Excel 异常!", e);
            return HttpResult.failure("导出异常");
        } finally {
            excelWriter.finish();
        }
    }

    @Getter
    @Setter
    @ColumnWidth(20)
    @ContentRowHeight(120)
    @ToString
    static class TestVo {

        @ExcelProperty(value = "测试列")
        private String testString;

        @ExcelProperty(value = "图片", converter = CustomUrlImageConverter.class)
        private URL inchImage;
    }

public class CustomUrlImageConverter implements Converter<URL> {

    @Override
    public Class<?> supportJavaTypeKey() {
        return URL.class;
    }

    @Override
    public CellDataTypeEnum supportExcelTypeKey() {
        return CellDataTypeEnum.IMAGE;
    }

    @Override
    public URL convertToJavaData(CellData cellData, ExcelContentProperty contentProperty,
            GlobalConfiguration globalConfiguration) {
        throw new UnsupportedOperationException("Cannot convert images to url.");
    }

    @Override
    public CellData<?>  convertToExcelData(URL value, ExcelContentProperty contentProperty,
            GlobalConfiguration globalConfiguration) throws IOException {
        InputStream inputStream = null;
        try {
            if (null == value) {
                return new CellData<>();
            }
            inputStream = value.openStream();
            byte[] bytes = IoUtils.toByteArray(inputStream);
            return new CellData<>(bytes);
        } finally {
            if (inputStream != null) {
                inputStream.close();
            }
        }
    }
}