springboot+vue实现页面下载文件

本文实例为大家分享了springboot+vue页面下载文件的具体代码，供大家参考，具体内容如下

1.前端代码：

<template v-slot:operate='{ row }'> <vxe-button circle @click='downloadFile(row)'></vxe-button></template>downloadFile(row) { window.location = 'http://localhost:8001/file/downloadFile?taskId=' + row.id;}

2.后端代码：

package com.gridknow.analyse.controller;import com.alibaba.fastjson.JSON;import com.gridknow.analyse.entity.DataInfo;import com.gridknow.analyse.service.FileService;import com.gridknow.analyse.utils.Download;import com.gridknow.analyse.utils.Result;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.beans.factory.annotation.Value;import org.springframework.core.io.InputStreamResource;import org.springframework.http.HttpHeaders;import org.springframework.http.MediaType;import org.springframework.http.ResponseEntity;import org.springframework.stereotype.Controller;import org.springframework.web.bind.annotation.*;import org.springframework.web.multipart.MultipartFile;import org.springframework.web.multipart.MultipartHttpServletRequest;import javax.servlet.http.HttpServletResponse;import java.io.*;import java.util.List;import java.util.Map;/** * @ClassName FileController * @Description: TODO * @Author Administrator * @Date 2020/8/20 14:02 * @Version TODO **/@Controller@RequestMapping('/file')public class FileController { @Value('${gridknow.mltc.imgurl}') private String imgUrl; @Autowired private FileService fileService; @CrossOrigin @RequestMapping(value = '/upload', method = RequestMethod.POST) @ResponseBody public Result upload(MultipartHttpServletRequest request) { List<MultipartFile> multipartFiles = request.getFiles('file'); Map<String, Object> map = (Map<String, Object>) JSON.parse(request.getParameter('body')); String companyId = request.getParameter('companyId'); String companyName = request.getParameter('companyName'); boolean bool = fileService.uploadAndInsert(multipartFiles, map, companyId, companyName); if (bool) { return new Result(200); } else { return new Result(500); } } @GetMapping('/downloadFile') public ResponseEntity<Object> downloadFile(@RequestParam('taskId') String taskId, HttpServletResponse response) { DataInfo dataInfo = fileService.queryTaskById(taskId); if (dataInfo == null) { return null; } File file = new File(dataInfo.getResponseUrl()); // 文件下载 if (file.isFile()) { return downloadFile(taskId); } // 文件夹压缩成zip下载 if (file.isDirectory()) { String parent = file.getParent(); // 创建临时存放文件夹 File temDir = new File(parent + '/' + taskId); if (!temDir.exists()) { temDir.mkdirs(); } // 将需要下载的文件夹和内容拷贝到临时文件夹中 try { Download.copyDir(dataInfo.getResponseUrl(), parent + '/' + taskId); } catch (IOException e) { e.printStackTrace(); } // 设置头部格式 response.setContentType('application/zip'); response.setHeader('Content-Disposition', 'attachment; filename='+taskId+'.zip'); // 调用工具类，下载zip压缩包 try { Download.toZip(temDir.getPath(), response.getOutputStream(), true); } catch (IOException e) { e.printStackTrace(); } // 删除临时文件夹和内容 Download.delAllFile(new File(parent + '/' + taskId)); } return null; } public ResponseEntity<Object> downloadFile(String taskId) { DataInfo dataInfo = fileService.queryTaskById(taskId); if (dataInfo == null) { return null; } File file = new File(dataInfo.getResponseUrl()); String fileName = file.getName(); InputStreamResource resource = null; try { resource = new InputStreamResource(new FileInputStream(file)); } catch (Exception e) { e.printStackTrace(); } HttpHeaders headers = new HttpHeaders(); headers.add('Content-Disposition', String.format('attachment;filename='%s', fileName)); headers.add('Cache-Control', 'no-cache,no-store,must-revalidate'); headers.add('Pragma', 'no-cache'); headers.add('Expires', '0'); ResponseEntity<Object> responseEntity = ResponseEntity.ok() .headers(headers) .contentLength(file.length()) .contentType(MediaType.parseMediaType('application/octet-stream')) .body(resource); return responseEntity; }}

工具类

package com.gridknow.analyse.utils;import lombok.extern.slf4j.Slf4j;import java.io.*;import java.util.zip.ZipEntry;import java.util.zip.ZipOutputStream;/** * @ClassName Download * @Description: TODO * @Author Administrator * @Date 2020/9/2 9:54 * @Version TODO **/@Slf4jpublic class Download { private static final int BUFFER_SIZE = 2 * 1024; public static void toZip(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(); log.info('压缩完成，耗时：' + (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[BUFFER_SIZE]; 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); } } } } } /** * 拷贝文件夹 * * @param oldPath 原文件夹 * @param newPath 指定文件夹 */ public static void copyDir(String oldPath, String newPath) throws IOException { File file = new File(oldPath); //文件名称列表 String[] filePath = file.list(); if (!(new File(newPath)).exists()) { (new File(newPath)).mkdir(); } for (int i = 0; i < filePath.length; i++) { if ((new File(oldPath + File.separator + filePath[i])).isDirectory()) { copyDir(oldPath + File.separator + filePath[i], newPath + File.separator + filePath[i]); } if (new File(oldPath + File.separator + filePath[i]).isFile()) { copyFile(oldPath + File.separator + filePath[i], newPath + File.separator + filePath[i]); } } } /** * 拷贝文件 * * @param oldPath 资源文件 * @param newPath 指定文件 */ public static void copyFile(String oldPath, String newPath) throws IOException { File oldFile = new File(oldPath); File file = new File(newPath); FileInputStream in = new FileInputStream(oldFile); FileOutputStream out = new FileOutputStream(file);; byte[] buffer=new byte[2097152]; while((in.read(buffer)) != -1){ out.write(buffer); } in.close(); out.close(); } /** * 删除文件或文件夹 * @param directory */ public static void delAllFile(File directory){ if (!directory.isDirectory()){ directory.delete(); } else{ File [] files = directory.listFiles(); // 空文件夹 if (files.length == 0){ directory.delete(); System.out.println('删除' + directory.getAbsolutePath()); return; } // 删除子文件夹和子文件 for (File file : files){ if (file.isDirectory()){ delAllFile(file); } else { file.delete(); System.out.println('删除' + file.getAbsolutePath()); } } // 删除文件夹本身 directory.delete(); System.out.println('删除' + directory.getAbsolutePath()); } }}

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持好吧啦网。