SpringBoot + zip4j 实现多文件批量下载并压缩解析

访客 阅读:249 2021-03-31 13:19:59 评论:0

重点功能代码:

@Async("threadPoolNeo4jTaskExecutor") 
	public void batchDownload(Long engSid, UserDownload entity) { 
		// 查询指定案卷信息 
		File rootDir = null; 
		File archDir = null; 
		String tempDirStr = null; 
		ArchInfo archInfo = archInfoService.selectByPrimaryKey(entity.getArchSid()); 
		if (archInfo != null) { 
			// 电子文件打包中 
			entity.setSid(engSid); 
			entity.setDownloadState("2"); 
			userDownloadService.updateByPrimaryKeySelective(entity); 
 
			String rootDirStr = this.rootPath(); 
			tempDirStr = String.valueOf(System.currentTimeMillis());// 临时存放目录 
			rootDir = new File(rootDirStr, tempDirStr); 
			if (!rootDir.exists()) { 
				rootDir.mkdir(); 
			} 
			archDir = new File(rootDir, archInfo.getArchTitle()); 
			if (!archDir.exists()) { 
				archDir.mkdirs(); 
			} 
 
			// 查询指定案卷涉及电子文件 
			Map<String, Object> param = new HashMap<String, Object>(); 
			param.put("archSid", archInfo.getSid()); 
			List<FileInfo> fileInfos = fileInfoService.findByParams(param); 
			if (CollectionUtils.isNotEmpty(fileInfos)) { 
				for (FileInfo item : fileInfos) { 
					// 获取文件根路径 
					String fileDir = transPath(item.getPdfPath() != null ? item.getPdfPath() : ""); 
					if (fileDir != null || fileDir.length() > 0) { 
						File file = new File(fileDir + item.getPdfFilename()); 
						// 判断文件是否存在 
						if (file.exists() && archDir.exists()) { 
							// 文件拷贝 
							try { 
								FileUtils.copyFile(file, new File(archDir, item.getFileTitle() + ".pdf")); 
							} catch (Exception e) { 
								log.error("export efile task, copy pdf file error : ", e); 
							} 
						} 
					} 
				} 
				try { 
					Zip4jUtil.zip(archDir.getAbsolutePath(), 
							rootDir.getAbsolutePath() + File.separator + archInfo.getArchTitle() + ".zip", ""); 
				} finally { 
					// 电子文件打包下载完成 
					entity.setSid(engSid); 
					entity.setDownloadState("3"); 
					entity.setDownloadAddress( 
							rootDir.getAbsolutePath() + File.separator + archInfo.getArchTitle() + ".zip"); 
					userDownloadService.updateByPrimaryKeySelective(entity); 
				} 
			} 
		} 
	}

Zip4jUtil 工具类源码:

 
import java.io.File; 
import java.util.ArrayList; 
import java.util.Collections; 
import java.util.List; 
 
import org.apache.commons.lang3.StringUtils; 
 
import net.lingala.zip4j.core.ZipFile; 
import net.lingala.zip4j.exception.ZipException; 
import net.lingala.zip4j.model.FileHeader; 
import net.lingala.zip4j.model.ZipParameters; 
import net.lingala.zip4j.util.Zip4jConstants; 
 
/**  
 * ZIP压缩文件操作工具类  
 * 支持密码  
 * 依赖zip4j开源项目(http://www.lingala.net/zip4j/)  
 * 版本1.3.1  
 */   
public class Zip4jUtil {   
       
    /**  
     * 使用给定密码解压指定的ZIP压缩文件到指定目录  
     * <p>  
     * 如果指定目录不存在,可以自动创建,不合法的路径将导致异常被抛出  
     * @param zip 指定的ZIP压缩文件  
     * @param dest 解压目录  
     * @param passwd ZIP文件的密码  
     * @return 解压后文件数组  
     * @throws ZipException 压缩文件有损坏或者解压缩失败抛出  
     */   
    public static File [] unzip(String zip, String dest, String passwd) throws ZipException {   
        File zipFile = new File(zip);   
        return unzip(zipFile, dest, passwd);   
    }   
       
    /**  
     * 使用给定密码解压指定的ZIP压缩文件到当前目录  
     * @param zip 指定的ZIP压缩文件  
     * @param passwd ZIP文件的密码  
     * @return  解压后文件数组  
     * @throws ZipException 压缩文件有损坏或者解压缩失败抛出  
     */   
    public static File [] unzip(String zip, String passwd) throws ZipException {   
        File zipFile = new File(zip);   
        File parentDir = zipFile.getParentFile();   
        return unzip(zipFile, parentDir.getAbsolutePath(), passwd);   
    }   
       
    /**  
     * 使用给定密码解压指定的ZIP压缩文件到指定目录  
     * <p>  
     * 如果指定目录不存在,可以自动创建,不合法的路径将导致异常被抛出  
     * @param zip 指定的ZIP压缩文件  
     * @param dest 解压目录  
     * @param passwd ZIP文件的密码  
     * @return  解压后文件数组  
     * @throws ZipException 压缩文件有损坏或者解压缩失败抛出  
     */   
    @SuppressWarnings("unchecked") 
	public static File [] unzip(File zipFile, String dest, String passwd) throws ZipException {   
        ZipFile zFile = new ZipFile(zipFile); 
//        zFile.setFileNameCharset(ApplicationPropertiesHolder.getProperty("zip4j.charset", "GBK")); 
        zFile.setFileNameCharset("GBK"); 
        if (!zFile.isValidZipFile()) {   
            throw new ZipException("压缩文件不合法,可能被损坏.");   
        }   
        File destDir = new File(dest);   
        if (destDir.isDirectory() && !destDir.exists()) {   
            destDir.mkdir();   
        }   
        if (zFile.isEncrypted()) { 
            zFile.setPassword(passwd.toCharArray());   
        }   
        zFile.extractAll(dest);   
           
        List<FileHeader> headerList = zFile.getFileHeaders();   
        List<File> extractedFileList = new ArrayList<File>();   
        for(FileHeader fileHeader : headerList) {   
            if (!fileHeader.isDirectory()) {   
                extractedFileList.add(new File(destDir,fileHeader.getFileName()));   
            }   
        }   
        File [] extractedFiles = new File[extractedFileList.size()];   
        extractedFileList.toArray(extractedFiles);   
        return extractedFiles;   
    }   
       
    /**  
     * 压缩指定文件到当前文件夹  
     * @param src 要压缩的指定文件  
     * @return 最终的压缩文件存放的绝对路径,如果为null则说明压缩失败.  
     */   
    public static String zip(String src) {   
        return zip(src,null);   
    }   
       
    /**  
     * 使用给定密码压缩指定文件或文件夹到当前目录  
     * @param src 要压缩的文件  
     * @param passwd 压缩使用的密码  
     * @return 最终的压缩文件存放的绝对路径,如果为null则说明压缩失败.  
     */   
    public static String zip(String src, String passwd) {   
        return zip(src, null, passwd);   
    }   
       
    /**  
     * 使用给定密码压缩指定文件或文件夹到当前目录  
     * @param src 要压缩的文件  
     * @param dest 压缩文件存放路径  
     * @param passwd 压缩使用的密码  
     * @return 最终的压缩文件存放的绝对路径,如果为null则说明压缩失败.  
     */   
    public static String zip(String src, String dest, String passwd) {   
        return zip(src, dest, true, passwd);   
    }   
       
    /**  
     * 使用给定密码压缩指定文件或文件夹到指定位置.  
     * <p>  
     * dest可传最终压缩文件存放的绝对路径,也可以传存放目录,也可以传null或者"".<br />  
     * 如果传null或者""则将压缩文件存放在当前目录,即跟源文件同目录,压缩文件名取源文件名,以.zip为后缀;<br />  
     * 如果以路径分隔符(File.separator)结尾,则视为目录,压缩文件名取源文件名,以.zip为后缀,否则视为文件名.  
     * @param src 要压缩的文件或文件夹路径  
     * @param dest 压缩文件存放路径  
     * @param isCreateDir 是否在压缩文件里创建目录,仅在压缩文件为目录时有效.<br />  
     * 如果为false,将直接压缩目录下文件到压缩文件.  
     * @param passwd 压缩使用的密码  
     * @return 最终的压缩文件存放的绝对路径,如果为null则说明压缩失败.  
     */   
    public static String zip(String src, String dest, boolean isCreateDir, String passwd) {   
        File srcFile = new File(src);   
        dest = buildDestinationZipFilePath(srcFile, dest);   
        ZipParameters parameters = new ZipParameters();   
        parameters.setCompressionMethod(Zip4jConstants.COMP_DEFLATE);           // 压缩方式   
        parameters.setCompressionLevel(Zip4jConstants.DEFLATE_LEVEL_NORMAL);    // 压缩级别   
        if (!StringUtils.isBlank(passwd)) { 
            parameters.setEncryptFiles(true);   
            parameters.setEncryptionMethod(Zip4jConstants.ENC_METHOD_STANDARD); // 加密方式   
            parameters.setPassword(passwd.toCharArray());   
        }   
        try {   
            ZipFile zipFile = new ZipFile(dest);   
            if (srcFile.isDirectory()) {   
                // 如果不创建目录的话,将直接把给定目录下的文件压缩到压缩文件,即没有目录结构   
                if (!isCreateDir) {   
                    File [] subFiles = srcFile.listFiles();   
                    ArrayList<File> temp = new ArrayList<File>();   
                    Collections.addAll(temp, subFiles);   
                    zipFile.addFiles(temp, parameters);   
                    return dest;   
                }   
                zipFile.addFolder(srcFile, parameters);   
            } else {   
                zipFile.addFile(srcFile, parameters);   
            }   
            return dest;   
        } catch (ZipException e) {   
            e.printStackTrace();   
        }   
        return null;   
    }   
       
    /**  
     * 构建压缩文件存放路径,如果不存在将会创建  
     * 传入的可能是文件名或者目录,也可能不传,此方法用以转换最终压缩文件的存放路径  
     * @param srcFile 源文件  
     * @param destParam 压缩目标路径  
     * @return 正确的压缩文件存放路径  
     */   
    private static String buildDestinationZipFilePath(File srcFile,String destParam) {   
        if (StringUtils.isBlank(destParam)) {   
            if (srcFile.isDirectory()) {   
                destParam = srcFile.getParent() + File.separator + srcFile.getName() + ".zip";   
            } else {   
                String fileName = srcFile.getName().substring(0, srcFile.getName().lastIndexOf("."));   
                destParam = srcFile.getParent() + File.separator + fileName + ".zip";   
            }   
        } else {   
            createDestDirectoryIfNecessary(destParam);  // 在指定路径不存在的情况下将其创建出来   
            if (destParam.endsWith(File.separator)) {   
                String fileName = "";   
                if (srcFile.isDirectory()) {   
                    fileName = srcFile.getName();   
                } else {   
                    fileName = srcFile.getName().substring(0, srcFile.getName().lastIndexOf("."));   
                }   
                destParam += fileName + ".zip";   
            }   
        }   
        return destParam;   
    }   
       
    /**  
     * 在必要的情况下创建压缩文件存放目录,比如指定的存放路径并没有被创建  
     * @param destParam 指定的存放路径,有可能该路径并没有被创建  
     */   
    private static void createDestDirectoryIfNecessary(String destParam) {   
        File destDir = null;   
        if (destParam.endsWith(File.separator)) {   
            destDir = new File(destParam);   
        } else {   
            destDir = new File(destParam.substring(0, destParam.lastIndexOf(File.separator)));   
        }   
        if (!destDir.exists()) {   
            destDir.mkdirs();   
        }   
    } 
     
	/* 
	 * 文件操作 返回不带扩展名的文件名 
	 */ 
	public static String getFileNameNoEx(String filename) { 
		if ((filename != null) && (filename.length() > 0)) { 
			int dot = filename.lastIndexOf('.'); 
			if ((dot > -1) && (dot < (filename.length()))) { 
				return filename.substring(0, dot); 
			} 
		} 
		return filename; 
	} 
  
}   

pom.xml 依赖:

<!-- zip处理工具包 --> 
		<dependency> 
			<groupId>net.lingala.zip4j</groupId> 
			<artifactId>zip4j</artifactId> 
			<version>1.3.2</version> 
		</dependency>

 

声明

1.本站遵循行业规范,任何转载的稿件都会明确标注作者和来源;2.本站的原创文章,请转载时务必注明文章作者和来源,不尊重原创的行为我们将追究责任;3.作者投稿可能会经我们编辑修改或补充。

发表评论
搜索
排行榜
KIKK导航

KIKK导航

关注我们