spring将文件压缩并下载

最新推荐文章于 2024-04-22 15:21:27 发布

有聊数太阳

最新推荐文章于 2024-04-22 15:21:27 发布

阅读量398

点赞数

分类专栏： java

本文链接：https://blog.csdn.net/qq_34002000/article/details/116067723

版权

java 专栏收录该内容

25 篇文章 0 订阅

订阅专栏

这段代码展示了如何使用Java进行文件压缩，将资源目录下的文件打包成ZIP，并提供HTTP下载。首先，定义了压缩源文件和目标压缩文件的路径，接着通过ZipOutputStream将源文件写入ZIP，压缩完成后删除源文件。最后，设置HTTP响应头，发送ZIP文件供用户下载，并在发送后删除ZIP文件。

摘要由CSDN通过智能技术生成

public void load(HttpServletRequest request, HttpServletResponse response) throws IOException, InterruptedException {
        //压缩源文件路径
        String zipSourcePath = ContextLoader.getCurrentWebApplicationContext().getServletContext().getRealPath("/") + "/resources/cache/";
        File sourceFile = new File(zipSourcePath);
        //生成的压缩文件路径
        String zipFilePath = ContextLoader.getCurrentWebApplicationContext().getServletContext().getRealPath("/") + "/resources/zip/";
        File zipPath = new File(zipFilePath);
        if (!zipPath.exists()) {
            zipPath.mkdirs();
        }
        File zipFile = new File(zipPath + File.separator + "资料.zip");
        try (ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(zipFile))) {
            writeZip(sourceFile, "", zos);
            //文件压缩完成后，删除被压缩文件
            boolean flag = deleteDir(sourceFile);
            File file = new File(ContextLoader.getCurrentWebApplicationContext().getServletContext().getRealPath("/") + "/resources/cache/");
            if (!file.exists()) {//如果文件夹不存在
                file.mkdir();//创建文件夹
            }
        } catch (Exception e) {
            e.printStackTrace();
            throw new RuntimeException(e.getMessage(), e.getCause());
        }
        //发送zip文件
        String readPath = ContextLoader.getCurrentWebApplicationContext().getServletContext().getRealPath("/") + "/resources/zip/资料.zip";
        InputStream in = null;
        OutputStream out = null;
        response.setContentType("multipart/form-data");
        response.addHeader("Content-Disposition", "attachment;fileName=" + URLEncoder.encode("资料.zip", "UTF-8"));

        try {
            in = new FileInputStream(readPath);
            int len = 0;
            byte buffer[] = new byte[1024];
            out = response.getOutputStream();
            while ((len = in.read(buffer)) != -1) {
                out.write(buffer, 0, len);
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (in != null) {
                in.close();
            }
            out.close();
            //删除压缩文件
            boolean flag = deleteDir(new File(readPath));
        }
    }

    private void writeZip(File file, String parentPath, ZipOutputStream zos) {
        if (file.isDirectory()) {
            //目录
            parentPath += file.getName() + File.separator;
            File[] files = file.listFiles();
            for (File f : files) {
                writeZip(f, parentPath, zos);
            }
        } else {
            //文件
            try (BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file))) {
                //指定zip文件夹
                ZipEntry zipEntry = new ZipEntry(parentPath + file.getName());
                zos.putNextEntry(zipEntry);
                int len;
                byte[] buffer = new byte[1024 * 10];
                while ((len = bis.read(buffer, 0, buffer.length)) != -1) {
                    zos.write(buffer, 0, len);
                    zos.flush();
                }
            } catch (Exception e) {
                e.printStackTrace();
                throw new RuntimeException(e.getMessage(), e.getCause());
            }
        }
    }

    public static boolean deleteDir(File dir) {
        if (dir.isDirectory()) {
            String[] children = dir.list();
            for (int i = 0; i < children.length; i++) {
                boolean success = deleteDir(new File(dir, children[i]));
                if (!success) {
                    return false;
                }
            }
        }
        //删除空文件夹
        return dir.delete();
    }