多个文件以zip包形式导出——亲测可用

深藏阁楼爱情的钟 2023-10-10 09:18 14阅读 0赞

工具类

package bdjk.sysmanagment.robot.util;
    
    import java.io.BufferedInputStream;
    import java.io.ByteArrayOutputStream;
    import java.io.File;
    import java.io.FileInputStream;
    import java.io.FileNotFoundException;
    import java.io.FileOutputStream;
    import java.io.IOException;
    import java.util.ArrayList;
    import java.util.List;
    import java.util.zip.ZipEntry;
    import java.util.zip.ZipOutputStream;
    
    /** * 文件上传工具类 * @author Administrator * */
    public class UploadUtil { 
    	
    	public static void uploadFile(byte[] file,String filePath,String fileName) throws IOException{ 
    		File targetFile  = new File(filePath);
    		if(!targetFile.exists()){ 
    			targetFile.createNewFile();
    			}
    		String upath =filePath+File.separator+fileName;
    		System.out.println("文件上传路径："+upath);
    		FileOutputStream out = new FileOutputStream(upath);
    		out.write(file);
    		out.flush();
    		out.close();
    	}
    	//直接传入文件路径的集合，导出的名字（如img），文件输出目录
    	public static ByteArrayOutputStream fileToZip(List<String> fileList,String zipName,String tempFilePath){ 
    		byte[] buffer = new byte[1024];
    		String filePath =tempFilePath+zipName+".zip";
    		ZipOutputStream out=null;
    		try { 
    			out =new ZipOutputStream(new FileOutputStream(filePath));
    			List<File> filedata = new ArrayList<File>();
    			for(int i=0,len=fileList.size();i<len;i++){ 
    				filedata.add(new File(fileList.get(i)));
    			}
    			for(int j=0,len=filedata.size();j<len;j++){ 
    				FileInputStream fis =new FileInputStream(filedata.get(j));
    				out.putNextEntry(new ZipEntry(filedata.get(j).getName()));
    				int dataLen;
    				while((dataLen =fis.read(buffer))>0){ 
    					out.write(buffer,0,dataLen);
    				}
    				out.closeEntry();
    				fis.close();
    			}
    			
    		} catch (Exception e) { 
    			// TODO Auto-generated catch block
    			e.printStackTrace();
    		}finally{ 
    			try { 
    				out.close();
    			} catch (IOException e) { 
    				// TODO Auto-generated catch block
    				e.printStackTrace();
    			}
    		}
    		//读取压缩包
    		File filezip = new File(filePath);
    		ByteArrayOutputStream baos =null;
    		
    		try { 
    			baos=new ByteArrayOutputStream();
    			FileInputStream inStream = new FileInputStream(filezip);
    			BufferedInputStream bis =new BufferedInputStream(inStream);
    			int c =bis.read();
    			while(c!=-1){ 
    				baos.write(c);
    				c=bis.read();
    			}
    			bis.close();
    			inStream.close();
    		} catch (Exception e) { 
    			// TODO Auto-generated catch block
    			e.printStackTrace();
    		}
    		return baos;
    		
    	}
    
    }

controller

@RequestMapping("download/downLoadImg")
    	public void downLoadImg(HttpServletRequest request,HttpServletResponse response) throws IOException { 
    		String taskId = request.getParameter("taskId");
    		if(taskId.equals("")||taskId==null){ 
    			return;
    		}
    		List<String> urls = this.infoSchService.getImgUrlByTask(taskId);
    		if(urls.size()<=0){ 
    			return;
    		}
    		ExportUtil exportUtil= new ExportUtil();
    		 List<String> imgList = exportUtil.exportImgByList(urls,filePath);
    		OutputStream out =null;
    		 try { 
    			out = response.getOutputStream();
    			ByteArrayOutputStream outPutStream = UploadUtil.fileToZip(imgList, "红外图谱文件", filePath);
    			byte[] buffer = outPutStream.toByteArray();
    			response.reset();
    			response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode("红外图谱文件.zip", "UTF-8"));
    			response.addHeader("content-Length", "" + outPutStream.size());
    			response.setContentType("application/octet-stream");
    			out.write(buffer);
    		} catch (IOException e) { 
    			e.printStackTrace();
    		} finally { 
    			 try { 
    				 out.close();
    			 } catch (IOException e) { 
    				 // TODO Auto-generated catch block
    				 e.printStackTrace();
    			 }
    			 File allFile = new File(filePath);
    			 String[] list = allFile.list();
               //如果不需要存档，最后可选择清空文件夹
    			 for (String name:list ) { 
    				 File ziFile  = new File(filePath,name);
    				 ziFile.delete();
    			 }
    			 
    		}

建议业务逻辑从controller层中摘到service里，上面只是为了更好的展示效果