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

清疚 2023-09-30 12:02 29阅读 0赞

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();
            }
        }