Java导出多个excel并打包压缩成.zip文件

2023-01-02 00:00:00 多个 打包 导出

1、先获取到数据,并将数据导出excel到指定位置

 public void downPoliceZip(WorksitePoliceApiInfo worksitePoliceApiInfo) throws Exception { 
        String zipName = "同步数据" + LocalDate.now() ;
        List<Map<String, Object>> workerMaps = new LinkedList<>();
        List<Map<String, Object>> worksiteMaps = new LinkedList<>();
        Map<String,Object > map = new LinkedHashMap<>();
        //获取数据
        .........
// String realPath = request.getSession().getServletContext().getContextPath();
            //创建临时文件夹保存excel
            String tempDir = zipPath + "/tempDir/" + LocalDate.now() + "/";
         
            //将导出的数据转成多个excel
            List<File> files = this.getStoreOrderExcels(tempDir, workerMaps, worksiteMaps, flag);

            //下载zip
            boolean tag = FileDownloadUtils.generateFile(tempDir, "zip", zipPath, zipName);
            //删除tempDir文件夹和其中的excel和zip文件
            boolean b = FileDownloadUtils.deleteDir(new File(zipPath + "\\tempDir"));
            if (!b) { 
                throw new RuntimeException("tempDir文件夹及其中的临时Excel和zip文件删除失败");
            }
    }

2、将导出的数据转成多个excel

 /** * 将导出的数据转成多个excel * * @param tempDir 路径 * @param workerMaps map集合 * @param worksiteMaps map集合 * @param flag 标识 * @return List<File> * @throws IOException 异常 */
    private List<File> getStoreOrderExcels(String tempDir, List<Map<String, Object>> workerMaps, List<Map<String, Object>> worksiteMaps, String[] flag) throws IOException { 
        FileDownloadUtils.createFile(tempDir);
        //存在多个文件
        List<File> files = new ArrayList<>();
        String path;
        for (int i = 0; i < flag.length; i++) { 
            if (flag[i].equals("worker")) { 
                path = this.getStoreOrderExcel(flag[i], workerMaps, tempDir);
            } else { 
                path = this.getStoreOrderExcel(flag[i], worksiteMaps, tempDir);
            }
            //excel添加到files中
            files.add(new File(path));
        }
        return files;
    }

/** * @param flag 标识 * @param maps map数组 * @param tempDir 路径 * @return String * @throws IOException 异常 */
    public String getStoreOrderExcel(String flag, List<Map<String, Object>> maps, String tempDir) throws IOException { 
        // 通过工具类创建writer,默认创建xls格式
        ExcelWriter writer = ExcelUtil.getWriter();
        if (flag.equals("worker")) { 
            //自定义标题别名
            writer.addHeaderAlias("workerName", "姓名");
            writer.addHeaderAlias("workerIdcard", "身份证号");
        } else { 
            //自定义标题别名
            writer.addHeaderAlias("workersiteName", "工地名称");
            writer.addHeaderAlias("worksiteAddress", "工地地址");
        }
        writer.write(maps, true);

        //生成一个excel
        String path = tempDir + LocalDate.now() + "_" + flag + ".xls";

        //本地测试下载
        FileOutputStream outputStream = new FileOutputStream(path);
        writer.flush(outputStream, true);
        writer.close();
        return path;
    }

3、相关工具类

import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class FileDownloadUtils { 

    /** * 创建文件夹; * * @param path 路径 */
    public static void createFile(String path) { 
        File file = new File(path);
        //判断文件是否存在;
        if (!file.exists()) { 
            //创建文件;
            file.mkdirs();
        }
    }

    /** * 删除文件夹及文件夹下所有文件 * * @param dir 文件地址 * @return boolean */
    public static boolean deleteDir(File dir) { 
        if (dir == null || !dir.exists()) { 
            return true;
        }
        if (dir.isDirectory()) { 
            String[] children = dir.list();
            //递归删除目录中的子目录下
            for (String child : children) { 
                boolean success = deleteDir(new File(dir, child));
                if (!success) { 
                    return false;
                }
            }
        }
        // 目录此时为空,可以删除
        return dir.delete();
    }

    /** * @Description 将多个文件进行压缩到指定位置 * @param path 要压缩的文件路径 * @param format 生成的格式(zip、rar) * @param zipPath zip的路径 * @param zipName zip文件名 */
    public static boolean generateFile(String path, String format,String zipPath,String zipName) throws Exception { 

        File file = new File(path);
        // 压缩文件的路径不存在
        if (!file.exists()) { 
            throw new Exception("路径 " + path + " 不存在文件,无法进行压缩...");
        }
        // 用于存放压缩文件的文件夹
        String generateFile = zipPath + File.separator ;
        File compress = new File(generateFile);
        // 如果文件夹不存在,进行创建
        if( !compress.exists() ){ 
            compress.mkdirs();
        }

        // 目的压缩文件
        String generateFileName = compress.getAbsolutePath() + File.separator  + zipName + "." + format;

        // 输出流
        FileOutputStream outputStream = new FileOutputStream(generateFileName);
        // 压缩输出流
        ZipOutputStream zipOutputStream = new ZipOutputStream(new BufferedOutputStream(outputStream));

        //压缩
        generateFile(zipOutputStream,file,"");
        System.out.println("源文件位置:" + file.getAbsolutePath() + ",目的压缩文件生成位置:" + generateFileName);
        // 关闭 输出流
        zipOutputStream.close();
        return true;
    }

    /** * @param out 输出流 * @param file 目标文件 * @param dir 文件夹 * @throws Exception */
    private static void generateFile(ZipOutputStream out, File file, String dir) throws Exception { 
        // 当前的是文件夹,则进行一步处理
        if (file.isDirectory()) { 
            //得到文件列表信息
            File[] files = file.listFiles();
            //将文件夹添加到下一级打包目录
            out.putNextEntry(new ZipEntry(dir + "/"));
            dir = dir.length() == 0 ? "" : dir + "/";
            //循环将文件夹中的文件打包
            for (int i = 0; i < files.length; i++) { 
                generateFile(out, files[i], dir + files[i].getName());
            }
        } else {  // 当前是文件
            // 输入流
            FileInputStream inputStream = new FileInputStream(file);
            // 标记要打包的条目
            out.putNextEntry(new ZipEntry(dir));
            // 进行写操作
            int len = 0;
            byte[] bytes = new byte[1024];
            while ((len = inputStream.read(bytes)) > 0) { 
                out.write(bytes, 0, len);
            }
            // 关闭输入流
            inputStream.close();
        }
    }
    原文作者:码农娟
    原文地址: https://blog.csdn.net/weixin_40750633/article/details/123864987
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。

相关文章