SpringBoot将文件夹打包成ZIP并下载

前几天发布了一篇JAVA压缩文件的文章,今天有朋友跟我说,这压缩没啥技术含量啊,你再弄个压缩完然后返回前端下载的功能呗 。其实吧我觉得下载功能比压缩更简单吧,压缩的递归我至少弄了半天才解决 。但是朋友有需求,那就搞一下呗 。
下载方法有两种:可以用response.getOutputStream().write()方法,也可以用SpringMVC的ResponseEntity<T>方法 。
下载方式也有两种,一种是先生成压缩文件放在磁盘上,然后再调用下载方法,下载完后删除磁盘中的压缩文件;另一种方法是将压缩文件以流的方式传递 。
代码比较粗糙,就是想到啥就直接写下来了,有待优化的地方还望大家多多指点
【SpringBoot将文件夹打包成ZIP并下载】import org.springframework.http.HttpHeaders;import org.springframework.http.HttpStatus;import org.springframework.http.MediaType;import org.springframework.http.ResponseEntity;import org.springframework.web.bind.annotation.GetMApping;import org.springframework.web.bind.annotation.RestController;import java.io.*;import java.util.zip.ZipEntry;import java.util.zip.ZipOutputStream;@RestControllerpublic class FileToZipDownload {/*** 文件压缩功能* @param sourcesFiles源资源,即需要被压缩的文件* @param zos压缩包* @param fileName压缩包内的文件名称*/public void compressFile(File sourcesFiles, ZipOutputStream zos,String fileName) throws IOException {//1.如果源资源是目录,先判断是否空文件夹if (sourcesFiles.isDirectory()){File[] files = sourcesFiles.listFiles();//1-1.先判断是否空文件夹if (files.length ==0){//1-2.空文件夹,则在压缩包中把目录加上zos.putNextEntry(new ZipEntry(fileName + File.separator));}else {//1-3.如果不是空文件夹,则递归列出里面的文件以及文件夹for (File file : files) {//文件名称以目录保持好compressFile(file,zos,fileName + File.separator + file.getName());}}}else {//将文件放到压缩文件中,同时保留文件名称zos.putNextEntry(new ZipEntry(fileName));//IO的固化操作,先读取文件再写入文件压缩输出流中,过程不解释FileInputStream inputStream = new FileInputStream(sourcesFiles);int len;byte[] bytes = new byte[2048];while ((len = inputStream.read(bytes)) != -1){zos.write(bytes,0,len);}zos.flush(); //刷新流zos.closeEntry();inputStream.close();}}/***ResponseEntity继承了HttpEntity类,HttpEntity代表一个http请求或者响应实体,其内部有两个成员变量:header及body,代表http请求或响应的header及body,其中的body是泛化的,具体方法可以百度一下* @returnResponseEntity<byte[]>* @throws Exception*/@GetMapping("/down")public ResponseEntity<byte[]> fileDownload(){//将要压缩的文件夹File sourceFiles = new File("E:" + File.separator + "hello");//设置响应头HttpHeaders headers = new HttpHeaders();//第一个参数name:设置响应方式,其中,attachment是通知浏览器以下载的方式打开文件;//第二个参数filename:文件下载的名称headers.setContentDispositionFormData("attachment",sourceFiles.getName() + ".zip");//通知浏览器以流的形式下载返回文件数据headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);//缓存数据输出流,可转换成byte字节ByteArrayOutputStream outputStream = new ByteArrayOutputStream();//创建压缩输出流对象,输出到缓存数据输出流中,不保存到磁盘上ZipOutputStream zos = new ZipOutputStream(outputStream);try {//调用压缩文件的方法this.compressFile(sourceFiles,zos,sourceFiles.getName());} catch (IOException e) {e.printStackTrace();}finally {try {if (zos != null){//ZipOutputStream压缩输出流必须在执行压缩操作后再关闭zos.close();}} catch (IOException e) {e.printStackTrace();}}//outputStream.toByteArray(),将缓存数据输出流转成字节,返回前端页面 。return new ResponseEntity<byte[]>(outputStream.toByteArray(),headers, HttpStatus.OK);}}





    推荐阅读


    上一篇:没有了

    下一篇:要不要走索引?MySQL 的成本分析