前段时间做项目时,有个需要下载excel模板的需求,但是由于用的是K8S部署,不能直接把模板文件放到指定的服务器,于是就把模板文件放到了项目resources目录下,然后本地测试一切正常,顺利提到了测试。

结果测试环境除了问题,下载下来是个空的excel,返回的response数据也是空。于是开始找问题,本地postman测试,response返回是乱码(其实文件是正常的,应该该是编码问题),文件下载出来没问题。postman连接测试接口,response是空,下载的文件还是空excel。然后网上各种找解决方案,发现百度出来的帖子,基本上千篇一律,就是一个帖子被N多人CV的,连格式都一模一样,找小伙伴帮忙看,也没有找到解决方案,问题就这么搁置了两天,一直没解决。

后来偶尔看到一个帖子,用了不一样的方法读取,我就试了一下,问题就解决了。直接贴代码,注意resource别导错包。

import org.springframework.core.io.Resource;
  
/**
     * 模板下载
     */
    public void downloadTemplate(HttpServletResponse response){
        OutputStream out = null;
        InputStream in = null;
        ByteArrayOutputStream bos = null;
        String fileName = "导入模版";

        try {
            // 读取模板
            Resource res = new ClassPathResource("Template.xlsx");
            XSSFWorkbook workbook = new XSSFWorkbook(res.getInputStream());

            // 转换为字节流
            bos = new ByteArrayOutputStream();
            workbook.write(bos);
            byte[] barray = bos.toByteArray();
            in = new ByteArrayInputStream(barray);

            response.reset();
            response.setContentType("application/octet-stream");
            response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName, "UTF-8") + ".xlsx");
            out = response.getOutputStream();
            byte[] b = new byte[1024];
            int len;
            while ((len = in.read(b)) > 0) {
                out.write(b, 0, len);
            }
            out.flush();
        } catch (Exception e) {
            log.error("下载模板失败",e);
        } finally {
            if (null != in) {
                try {
                    in.close();
                } catch (IOException e) {
                    log.error("关闭资源异常",e);
                }
                in = null;
            }
            if (null != out) {
                try {
                    out.close();
                } catch (IOException e) {
                    log.error("关闭资源异常",e);
                }
                out = null;
            }
            if (null != bos) {
                try {
                    bos.flush();
                    bos.close();
                } catch (IOException e) {
                    log.error("关闭资源异常",e);
                }
                out = null;
            }
        }
    }

困扰我两天的问题,Resource res = new ClassPathResource("whiteListTemplate.xlsx"); 这一行代码就解决了。

总结:本地测试时,可以直接读取到项目中的文件,所以能把文件下载下来。但是线上部署用的Docker和K8S,是jar包运行的,在读取文件的时候,因为之前的方法问题,一直读取不到文件内容,然后用Resource res = new ClassPathResource("Template.xlsx"); 是通过流的形式读取的,resource继承了InputStreamSource,所以才能以流的方式读取到文件内容。

因为对K8S、Docker和Spring源码并不了解,以上只是我个人的猜想,希望知道原理或者正确答案的小伙伴不吝赐教,万分感谢!

 

Logo

K8S/Kubernetes社区为您提供最前沿的新闻资讯和知识内容

更多推荐