将多张图片打成一个zip压缩包

淩亂°似流年 2023-05-29 02:44 29阅读 0赞

/**
         * 文件压缩
         * @param newVo 图片地址集合
         * @param filePath 文件生成地址
         * @param zipPath 压缩包生成地址
         * @return
         * @throws Exception
         */
        public List<String> fileZipSave(List<String> newVo, String filePath, String zipPath)  throws Exception{
            List<String> savePathList = new ArrayList<>();
            try {
                // 创建一个文件夹
                File file = new File(filePath);
                //如果文件夹存在
                if(file.exists()){
                    // 先删除文件
                    FileUtil.delFolder(filePath);
                }
                // 再创建文件
                file.mkdirs();
                logger.info("成交确认单文件名称=======>>>>  "+file.getName()+": "+file.getPath());
    
                List<String> fileList = new ArrayList<>();
                fileList.add(filePath);
    
                long picLength = 0;
                //多个图片下载地址
                for(int i=0; i<newVo.size() ; i++) {
                    String dealPicUrl = newVo.get(i);
                    //new一个URL对象
                    StringBuffer sb = new StringBuffer();
                    URL url = new URL(sb.toString());
                    //打开链接
                    HttpURLConnection conn = (HttpURLConnection)url.openConnection();
                    //设置请求方式为"GET"
                    conn.setRequestMethod("GET");
                    //超时响应时间为5秒
                    conn.setConnectTimeout(5 * 1000);
                    //通过输入流获取图片数据
                    InputStream inStream = conn.getInputStream();
                    //得到图片的二进制数据，以二进制封装得到数据，具有通用性
                    byte[] data = FileUtil.readInputStream(inStream);
                    picLength+= data.length;
                    // 如果图片大于10M,就再创建一个文件夹
                    if (picLength > 10*1024*1024){
                        picLength = data.length;
                        filePath = filePath+i;
                        File files = new File(filePath);
                        //如果文件夹存在
                        if(files.exists()){
                            // 先删除文件
                            FileUtil.delFolder(filePath);
                        }
                        files.mkdirs();
                        fileList.add(filePath);
                    }
    
                    //new一个文件对象用来保存图片，默认保存当前工程根目录
    
                    String picName = filePath+"/"+"图片名称".png;
                    logger.info("成交确认单写入文件的图片名称：=======>>>>  "+picName);
                    File imageFile = new File(picName);
                    //创建输出流
                    FileOutputStream outStream = new FileOutputStream(imageFile);
                    //写入数据
                    outStream.write(data);
                    //关闭输出流
                    outStream.close();
                }
    
                // 压缩法
                for (int i = 0 ; i<fileList.size() ; i++){
                    zipPath = zipPath+i;
                    FileOutputStream fos1 = new FileOutputStream(new File(zipPath+".zip"));
                    FileUtil.toZip1(fileList.get(i), fos1,false);
    
                    File zipFile = new File(zipPath+".zip");
                    logger.info("成交确认单压缩文件名称=======>>>>  "+zipFile.getName()+": "+zipFile.getPath());
                    FileInputStream fileInputStream = new FileInputStream(zipFile);
                    // 注意：这个是我保存到自己服务器代码
                    String savePath = mediaStorageService.storageMedia(fileInputStream, "zip");
                    if(org.apache.commons.lang.StringUtils.isNotBlank(savePath)){
                        savePathList.add(savePath);
                    }
                    logger.info("成交确认单zip=======>>>>  "+Configure.getConfigFromZookeeper("uc.pai.file.real.url.prefix")+savePath);
                    if (org.apache.commons.lang.StringUtils.isNotEmpty(savePath)){
                        // 删除文件和压缩文件
                        FileUtil.delFolder(fileList.get(i));
                        FileUtil.delFolder(zipPath+".zip");
                    }
                }
            } catch (MalformedURLException e) {
                e.printStackTrace();
            } catch (IOException e) {
                e.printStackTrace();
                logger.error("方法fileZipSave发生IO异常："+e.getMessage());
            }
    
            return savePathList;
        }
    
    
    /**
    	 * 得到图片的二进制数据，以二进制封装得到数据，具有通用性
    	 * @param inStream
    	 * @return
    	 * @throws Exception
    	 */
    	public static byte[] readInputStream(InputStream inStream) throws Exception{
    		ByteArrayOutputStream outStream = new ByteArrayOutputStream();
    		//创建一个Buffer字符串
    		byte[] buffer = new byte[1024];
    		//每次读取的字符串长度，如果为-1，代表全部读取完毕
    		int len = 0;
    		//使用一个输入流从buffer里把数据读取出来
    		while( (len=inStream.read(buffer)) != -1 ){
    			//用输出流往buffer里写入数据，中间参数代表从哪个位置开始读，len代表读取的长度
    			outStream.write(buffer, 0, len);
    		}
    		//关闭输入流
    		inStream.close();
    		//把outStream里的数据写入内存
    		return outStream.toByteArray();
    	}
    
    /**
    	 * 压缩成ZIP 方法1
    	 * @param srcDir 压缩文件夹路径
    	 * @param out    压缩文件输出流
    	 * @param KeepDirStructure  是否保留原来的目录结构,true:保留目录结构;
    	 *                          false:所有文件跑到压缩包根目录下(注意：不保留目录结构可能会出现同名文件,会压缩失败)
    	 * @throws RuntimeException 压缩失败会抛出运行时异常
    	 */
    	public static void toZip1(String srcDir, OutputStream out, boolean KeepDirStructure)
    			throws RuntimeException{
    		long start = System.currentTimeMillis();
    		ZipOutputStream zos = null ;
    		try {
    			zos = new ZipOutputStream(out);
    			File sourceFile = new File(srcDir);
    			compress(sourceFile,zos,sourceFile.getName(),KeepDirStructure);
    			long end = System.currentTimeMillis();
    			System.out.println("压缩完成，耗时：" + (end - start) +" ms");
    		} catch (Exception e) {
    			throw new RuntimeException("zip error from ZipUtils",e);
    		}finally{
    			if(zos != null){
    				try {
    					zos.close();
    				} catch (IOException e) {
    					e.printStackTrace();
    				}
    			}
    		}
    	}
    
    	/**
    	 * 压缩成ZIP 方法2
    	 * @param srcFiles 需要压缩的文件列表
    	 * @param out           压缩文件输出流
    	 * @throws RuntimeException 压缩失败会抛出运行时异常
    	 */
    	public static void toZip(List<File> srcFiles , OutputStream out)throws RuntimeException {
    		long start = System.currentTimeMillis();
    		ZipOutputStream zos = null ;
    		try {
    			zos = new ZipOutputStream(out);
    			for (File srcFile : srcFiles) {
    				byte[] buf = new byte[2 * 1024];
    				zos.putNextEntry(new ZipEntry(srcFile.getName()));
    				int len;
    				FileInputStream in = new FileInputStream(srcFile);
    				while ((len = in.read(buf)) != -1){
    					zos.write(buf, 0, len);
    				}
    				zos.closeEntry();
    				in.close();
    			}
    			long end = System.currentTimeMillis();
    			System.out.println("压缩完成，耗时：" + (end - start) +" ms");
    		} catch (Exception e) {
    			throw new RuntimeException("zip error from ZipUtils",e);
    		}finally{
    			if(zos != null){
    				try {
    					zos.close();
    				} catch (IOException e) {
    					e.printStackTrace();
    				}
    			}
    		}
    	}
    
    /**
    	 * 递归压缩方法
    	 * @param sourceFile 源文件
    	 * @param zos        zip输出流
    	 * @param name       压缩后的名称
    	 * @param KeepDirStructure  是否保留原来的目录结构,true:保留目录结构;
    	 *                          false:所有文件跑到压缩包根目录下(注意：不保留目录结构可能会出现同名文件,会压缩失败)
    	 * @throws Exception
    	 */
    	private static void compress(File sourceFile, ZipOutputStream zos, String name,
    								 boolean KeepDirStructure) throws Exception{
    		byte[] buf = new byte[2 * 1024];
    		if(sourceFile.isFile()){
    			// 向zip输出流中添加一个zip实体，构造器中name为zip实体的文件的名字
    			zos.putNextEntry(new ZipEntry(name));
    			// copy文件到zip输出流中
    			int len;
    			FileInputStream in = new FileInputStream(sourceFile);
    			while ((len = in.read(buf)) != -1){
    				zos.write(buf, 0, len);
    			}
    			// Complete the entry
    			zos.closeEntry();
    			in.close();
    		} else {
    			File[] listFiles = sourceFile.listFiles();
    			if(listFiles == null || listFiles.length == 0){
    				// 需要保留原来的文件结构时,需要对空文件夹进行处理
    				if(KeepDirStructure){
    					// 空文件夹的处理
    					zos.putNextEntry(new ZipEntry(name + "/"));
    					// 没有文件，不需要文件的copy
    					zos.closeEntry();
    				}
    			}else {
    				for (File file : listFiles) {
    					// 判断是否需要保留原来的文件结构
    					if (KeepDirStructure) {
    						// 注意：file.getName()前面需要带上父文件夹的名字加一斜杠,
    						// 不然最后压缩包中就不能保留原来的文件结构,即：所有文件都跑到压缩包根目录下了
    						compress(file, zos, name + "/" + file.getName(),KeepDirStructure);
    					} else {
    						compress(file, zos, file.getName(),KeepDirStructure);
    					}
    				}
    			}
    		}
    	}

以上是接口，大家调用fileZipSave这个方法，然后传入相关参数即可